server/mylog/MyLog.cpp

#include "stdafx.h"
#include "MyLog.h"

MyLog::MyLog(FILE_TYPE type)
{
	flag = false;

	fileType = type;

	curFileCount = 0;
	strExeDir = GetExeDir();
	if (fileType >= 0 && fileType < FILE_TYPE_TOTAL) {
		status = false;
		InitLogConfig();
		flag = true;
	}		
}

MyLog::MyLog(FILE_TYPE type,bool status)
{
	flag = false;

	fileType = type;

	curFileCount = 0;
	this->status = status;
	strExeDir = GetExeDir();

	if (fileType >= 0 && fileType < FILE_TYPE_TOTAL) {
		InitLogConfig();
		flag = true;
	}	
}

MyLog::~MyLog()
{
	//pCategory->shutdown();
}

int MyLog::WriteLog(const std::string msg)
{
	pCategory->info(msg.c_str());

	//如果日志文件超过大小，则开启新文件名
	if (getCurTime() != m_Lasttime)
	{
		m_Lasttime = getCurTime();
		std::string filename = GetFileName();
		SetAppender(filename);
	}

	return 0;
}

int MyLog::WriteLog(const char * pMsg,...)
{
	va_list args;
	char log[1024];
	va_start(args, pMsg);
	vsnprintf_s(log,1024-1,pMsg,args);

	std::string strLog = "";
	strLog = log;
	WriteLog(strLog);

	return 0;
}

int MyLog::test()
{
	bool bRet = false;

	std::string dir = "log/20170603";

	dir = GetFileName();

	for (int i = 0;i < 10;i++) {
		curFileCount++;
		dir = GetFileName();
	}

	dir = "";
	dir = GetExeDir();

	bRet = IsDirExist(dir);
	if (!bRet) {
		CreateDir(dir.c_str());
	}

	return 0;
}

int MyLog::InitLogConfig()
{
	std::string name = my_log_file_name[fileType];
	
	std::string str_category = my_log_file_name[fileType] + "_category";
	m_Lasttime = getCurTime();
	std::string filepath = strExeDir + "log\\" + m_Lasttime + "\\";
	if (!IsDirExist(filepath))
	{
		CreateDir(filepath.c_str());
	}
	std::string filename = filepath + GetFileName();
	pLayout = new log4cpp::PatternLayout();
	/*
	%c category；
	%d 日期；日期可以进一步的设置格式，用花括号包围，例如%d{ %H:%M : %S,%l } 或者 %d{ %d %m %Y %H:%M : %S,%l }。如果不设置具体日期格式，则如下默认格式被使用“Wed Jan 02 02:03 : 55 1980”。日期的格式符号与ANSI C函数strftime中的一致。但增加了一个格式符号%l，表示毫秒，占三个十进制位。
	%m 消息；
	%n 换行符，会根据平台的不同而不同，但对于用户透明；
	%p 优先级；
	%r 自从layout被创建后的毫秒数；
	%R 从1970年1月1日0时开始到目前为止的秒数；
	%u 进程开始到目前为止的时钟周期数；
	%x NDC。
	*/

	//如果status是false，表示不输出时间信息
	//如果status是true，则RAW_S文件不输出时间信息，其他文件类型输出时间信息
	pLayout->setConversionPattern(log_file_pattern[status]);

	//appender = new log4cpp::FileAppender(name.c_str(), filename.c_str());
	//RollingFileAppender参数说明
	//第三个参数表示每个文件的最大阈值，这里为200M
	//第四个参数表示备份文件最大数
	//第五个参数表示缓冲立即刷新
	pAppender = new log4cpp::RollingFileAppender(name.c_str(), filename.c_str(),MAX_FILE_SIZE,MAX_BACKUP_COUNTS,true);

	pAppender->setLayout(pLayout);

	pCategory = &(log4cpp::Category::getInstance(str_category.c_str()));
	pCategory->addAppender(pAppender);
	pCategory->setPriority(log4cpp::Priority::INFO);

	return 0;
}

std::string MyLog::GetExeDir()
{
	std::string dir = "";

	TCHAR chpath[MAX_PATH];
	::GetModuleFileName(NULL,chpath, MAX_PATH);
	(_tcsrchr(chpath, _T('\\')))[1] = 0;
	dir = TCHAR2string(chpath);

	return dir;
}

bool MyLog::IsDirExist(std::string path)
{
	bool bRet = false;

	struct _stat fileStat;
	if ((_stat(path.c_str(), &fileStat) == 0) && (fileStat.st_mode & _S_IFDIR))
	{
		bRet = true;
	}

	return bRet;
}

bool MyLog::CreateDir(const char * pDir)
{
	//std::string cmd = "md " + path;
	//system(cmd.c_str());//创建一个文件夹 
	int i = 0;  
	int iRet;  
	int iLen;  
	char* pszDir;  

	if(NULL == pDir)  
	{  
		return 0;  
	}  

	pszDir = strdup(pDir);  
	iLen = strlen(pszDir);  

	// 创建中间目录  
	for (i = 0;i < iLen;i ++)  
	{  
		if (pszDir[i] == '\\' || pszDir[i] == '/')  
		{   
			pszDir[i] = '\0';  

			//如果不存在,创建  
			iRet = ACCESS(pszDir,0);  
			if (iRet != 0)  
			{  
				iRet = MKDIR(pszDir);  
				if (iRet != 0)  
				{  
					return false;  
				}   
			}  
			//支持linux,将所有\换成/  
			pszDir[i] = '/';  
		}   
	}  

	iRet = MKDIR(pszDir);  
	free(pszDir);  
	return true;
}

std::string MyLog::TCHAR2string(TCHAR * tch)
{
	int iLen = WideCharToMultiByte(CP_ACP, 0, tch, -1, NULL, 0, NULL, NULL);

	char* chRtn = new char[iLen * sizeof(char)];

	WideCharToMultiByte(CP_ACP, 0, tch, -1, chRtn, iLen, NULL, NULL);

	std::string str(chRtn);

	return str;
}

std::string MyLog::int2string(int value)
{
	return std::to_string(value);
}

/*
 * 当日期变化后，在新目录内重新生成日志 
*/
int MyLog::SetAppender(const char* filename)
{
	std::string name = my_log_file_name[fileType];
	//std::string strFileName = strExeDir + "log/" + filename;
	std::string filepath = strExeDir + "log/" + m_Lasttime + "/";
	if (!IsDirExist(filepath))
	{
		CreateDir(filepath.c_str());
	}
	std::string strFileName = filepath + filename;

	pCategory->removeAppender(pAppender);

	pAppender = new log4cpp::RollingFileAppender(name.c_str(), strFileName.c_str(),MAX_FILE_SIZE,MAX_BACKUP_COUNTS,true);
	//need malloc again 
	pLayout = new log4cpp::PatternLayout();
	pLayout->setConversionPattern(log_file_pattern[status]);

	pAppender->setLayout(pLayout);

	pCategory->setAppender(pAppender);
	return 0;
}

int MyLog::SetAppender(std::string filename)
{
	SetAppender(filename.c_str());
	return 0;
}

int MyLog::SetLayout(bool status)
{
	if (status)
	{
		pLayout->setConversionPattern(log_file_pattern[1]);
	} 
	else
	{
		pLayout->setConversionPattern(log_file_pattern[0]);
	}
	
	return 0;
}

long MyLog::GetFileSize()
{
	if (curFileName == "") {
		return 0;
	}

	struct _stat info;
	_stat(curFileName.c_str(), &info);
	long size = info.st_size;

	return size;
}
std::string MyLog::getCurTime()
{
	std::string stringDate;
	char chDate[255] = {0};
	time_t today;
	tm* t = new tm;

	time(&today);
	localtime_s(t, &today);

	strftime(chDate, 255, "%Y%m%d", t);
	stringDate = chDate;
	if (t) {
		delete t;
	}
	return stringDate;
}
std::string MyLog::GetFileName()
{
	//日志文件格式为日期 + 计数 + 后缀（.log）
	std::string stringDate = "";
	stringDate = getCurTime();
	//curFileName = my_log_file_name[fileType] + "_" + curDate + "_" + int2string(curFileCount) + ".log";
	curFileName = my_log_file_name[fileType] + "_" + stringDate + ".log";

	return curFileName;
}