C++打印日记代码

参考自: https://www.cnblogs.com/thefirstfeeling/p/5896562.html

main.cpp

#include <iostream>
#include "NetDataLog.h"
#define _AFXDLL
#include <afx.h>
using namespace std;

void main(void)
{
	printf("**************************\n");

	NetDataLog netlog;

	for (int i=0; ; i++)
	{
		char buf[2048] = {0};
		sprintf(buf, "cnt = %d\n", i);
		netlog.addLog(buf);
		Sleep(1000);
	}

	system("pause");
}

NetDataLog.h

#ifndef NETDATALOG_H
#define NETDATALOG_H
#include <iostream>
#include <fstream>
#include <stdio.h>
#include <direct.h>
#include <string>
#include <io.h>
#include <stdlib.h>
#include <time.h>
#include <fcntl.h>

using namespace std;
enum TIMEFORMAT
{
	NETLOG = 0,    //    [yyyy\mm\dd hh.MM.ss]
	LOGINLOG=1,    //    mm-dd hh:MM:ss
};

class NetDataLog
{
public:
	NetDataLog(string strDir = "log",string filename = "record",int maxfilesize=0,int filecount=0,int timeformat=0);
	~NetDataLog();

	void addLog(string log);    //添加日志记录到日志文件
	void fileSizeLimit();        //判断文件大小是否达到限定值
	int getCurrentLogFileSize();//获取当前日志文件的大小
	string getLogFileName();    //获取日志文件名称
	void setMaxFileSize(int);//设置文件最大大小
	void setFileName(string); //设置日志文件名
	void setFileCount(int);    //设置日志文件的个数
	void setLogDir(string strDir);        //设置日志文件目录
private:
	void fileOffset();        //文件名称进行偏移
	bool checkFolderExist(const string &strPath);
	string getCurrentTime();


private:
	string m_LogFileName;    //文件名
	int m_MaxFileSize;        //文件大小
	int m_FileCount;        //文件个数
	fstream *m_outputFile;    //输出文件流
	string m_strDir;        //目录
	int m_timeFormat;


};
#endif

NetDataLog.cpp

//#include "stdafx.h"
#include "NetDataLog.h"
NetDataLog::NetDataLog(string strDir,string filename,int maxsize,int filecount,int timeFormat)
{
	m_strDir = strDir;
	m_LogFileName = m_strDir+string("\\")+filename;
	m_MaxFileSize = maxsize;
	m_FileCount = filecount;
	m_timeFormat = timeFormat;

	//判断日志目录是否存在
	if(!checkFolderExist(m_strDir.c_str()))    
	{
		_mkdir(m_strDir.c_str());
	}
	m_outputFile = new fstream;
	string strname = m_LogFileName+".txt";
	m_outputFile->open(strname.c_str(),ofstream::out|ofstream::app);    //打开日志文件
	bool b=m_outputFile->is_open();


}
NetDataLog::~NetDataLog()
{
	if(m_outputFile)
		delete m_outputFile;
}
//********************************
//函数名：NetDataLog::checkFolderExist
//描  述：测试目录是否存在
//参  数：strPath 目录名
//返回值：存在返回真
//*************************************
bool NetDataLog::checkFolderExist( const string  & strPath)  
{  
	if(_access(strPath.data(),0) == 0)
		return true;
	else
		return false;
}  
//********************************
//函数名：NetDataLog::addLog
//描  述：向文件中添加日志信息
//参  数 log 为信息内容
//返回值：void
//*************************************
void NetDataLog::addLog(string log)
{
	string currentTime = getCurrentTime(); //获取本地时间
	if(m_timeFormat == NETLOG)
		*m_outputFile<<"["<<currentTime<<"] "<<log<<endl;
	else
		*m_outputFile<<currentTime<<" "<<log<<endl;
	//判断文件大小
	fileSizeLimit();
}
//********************************
//函数名：NetDataLog::fileSizeLimit
//描  述：判断文件大小是否达到最大值
//参  数：无
//返回值：void
//*************************************
void NetDataLog::fileSizeLimit() 
{
	int filesize = getCurrentLogFileSize();
	if(filesize>=m_MaxFileSize*1024)
		fileOffset();

} 
//********************************
//函数名：NetDataLog::fileOffset
//描  述：实现文件名的偏移
//参  数：无
//返回值：void
//*************************************
void NetDataLog::fileOffset()
{
	m_outputFile->close();    //关闭当前文件
	char filename[100]={0};
	char newfilename[100] = {0};
	for(int i = m_FileCount-1;i > 0;i--)
	{
		memset(filename,0,100);
		sprintf(filename,"%s%d.txt",m_LogFileName.data(),i);
		if(checkFolderExist(filename))  //存在
		{
			if(i == m_FileCount-1)
			{
				remove(filename);//删除文件
				continue;
			}
			//文件名序号向后偏移
			memset(newfilename,0,100);
			sprintf(newfilename,"%s%d.txt",m_LogFileName.data(),i+1);
			rename(filename,newfilename);
		}
	}
	memset(filename,0,100);
	sprintf(filename,"%s.txt",m_LogFileName.data());
	sprintf(newfilename,"%s%d.txt",m_LogFileName.data(),1);
	rename(filename,newfilename);
	m_outputFile->open(filename,ofstream::out|ofstream::app);    //打开日志文件
}

//********************************
//函数名：NetDataLog::getCurrentLogFileSize
//描  述：计算当前日记文件的大小
//参  数：无
//返回值：文件大小（KB）
//*************************************
int NetDataLog::getCurrentLogFileSize()
{
	long long filepos = m_outputFile->tellp(); //保存当前文件位置
	m_outputFile->seekp(0,ios_base::end);            //移动到文件尾
	long long filesize = m_outputFile->tellp();    
	m_outputFile->seekp(filepos,ios_base::beg);        //恢复文件位置
	return filesize/1024;

}
//获取文件名
string NetDataLog::getLogFileName()
{
	return m_LogFileName+".txt";
}
//设置文件个数
void NetDataLog::setFileCount(int count)
{
	m_FileCount = count;
}
//设置文件名
void NetDataLog::setFileName(string filename)
{
	m_LogFileName = m_strDir+string("\\")+filename;
}
//设置文件大小
void NetDataLog::setMaxFileSize(int maxsize)
{
	m_MaxFileSize = maxsize;
}

//********************************
//函数名：NetDataLog::getCurrentTime
//描  述：获取本地时间
//返回值：时间字符串
//*************************************
string NetDataLog::getCurrentTime()
{
	time_t seconds = time(NULL);    //获取时间
	struct tm *p;
	p = localtime(&seconds);//获取本地时间
	char strTime[100] = {0};
	if(m_timeFormat == NETLOG)
		sprintf(strTime,"%d\\%d\\%d %d.%d.%d",1900+p->tm_year,1+p->tm_mon,p->tm_mday,p->tm_hour,p->tm_min,p->tm_sec);
	else
		sprintf(strTime,"%02d-%02d %02d:%02d:%02d",1+p->tm_mon,p->tm_mday,p->tm_hour,p->tm_min,p->tm_sec);
	return string(strTime);
}