工具类库系列(十四)-FileData

第十四个工具类：FileData

用于文本文件读取/保存，以及在内存中修改文本文件内容。

读取：ifstream

保存：ofstream

修改内容：以string类型，获取/设置，文本文件内容

最大支持文件大小1M

上代码：

FileData.h

#ifndef __FileData_h__
#define __FileData_h__

#include <string>

namespace common{
	namespace tool{

		class FileData
		{
		public:
			FileData();
			FileData(const FileData& other);
			~FileData();

			FileData& operator=(const FileData& other);

			// 提取文本文件内容，将文本中内容下标为[begin, end]之间（包括）的字符，提取到data中
			void GetData(std::string& data, unsigned int begin, unsigned int end);
			// 设置文本文件内容，用data中内容，替换掉文本中内容下标为[begin, end]之间（包括）的字符
			void SetData(const std::string& data, unsigned int begin, unsigned int end);

			// 获取文本文件内容长度
			unsigned int GetLen();
			// 设置文本文件内容长度
			void SetLen(unsigned int len);

			// 读取/保存文本文件
			void Load(const std::string& fileName);
			void Save(const std::string& fileName);

			// 文本数据最大支持1M
			static const unsigned int MaxDataLen = 1024 * 1024;

		private:
			unsigned int m_DataLen;
			char m_Data[MaxDataLen];
		};

	}
}

#endif

#include "FileData.h"

#include <string.h>
#include <fstream>

namespace common{
	namespace tool{

		FileData::FileData()
		{
			m_DataLen = 0;
		}
		FileData::FileData(const FileData& other)
		{
			*this = other;
		}
		FileData::~FileData()
		{

		}

		FileData& FileData::operator=(const FileData& other)
		{
			m_DataLen = other.m_DataLen;
			memcpy(m_Data, other.m_Data, other.m_DataLen);

			return *this;
		}

		void FileData::GetData(std::string& data, unsigned int begin, unsigned int end)
		{
			if (begin <= end && end < MaxDataLen)
			{
				data.assign(m_Data + begin, end - begin + 1);
			}
		}

		void FileData::SetData(const std::string& data, unsigned int begin, unsigned int end)
		{
			if (begin <= end && end < MaxDataLen && (end - begin + 1) <= data.length())
			{
				memcpy(m_Data + begin, data.c_str(), end - begin + 1);

				if (m_DataLen < end + 1)
				{
					m_DataLen = end + 1;
				}
			}
		}

		unsigned int FileData::GetLen()
		{
			return m_DataLen;
		}

		void FileData::SetLen(unsigned int len)
		{
			m_DataLen = len;
		}

		void FileData::Load(const std::string& fileName)
		{
			std::ifstream file;
			file.open(fileName.c_str(), std::ios::in | std::ios::binary | std::ios::ate);
			if (file)
			{
				m_DataLen = static_cast<unsigned int>(file.tellg());
				if (m_DataLen <= MaxDataLen)
				{
					file.seekg(0, std::ios::beg);
					file.read(m_Data, m_DataLen);
				}
				file.close();
			}
		}

		void FileData::Save(const std::string& fileName)
		{
			std::ofstream file;
			file.open(fileName.c_str(), std::ios::out | std::ios::trunc | std::ios::binary);
			if (file)
			{
				file.write(m_Data, m_DataLen);
				file.flush();
				file.close();
			}
		}

	}
}