C++的日志类

前一个项目中，当时遇到日志的问题，因为时间紧迫，用了开源社区的glog。但本人对开源的东西又或者说不是自己写的东西的总有点不放心，也可以说是心理有点疾病吧。于是今天就趁着闲下来，自己写了个C++的日志类，机制是仿照自己之前写的C#log类。

       日志类所要解决的问题：

       1. 自检查目录以及文件的存在性，如果不存在，则创建

       2. 多线程冲突问题

       3. 写日志的功能（最根本的需求）

       本文中采用的思路为增加自定义缓冲区与冲突锁以解决线程间冲突问题，采用后台线程负责单独写操作，采用单例以解决文件访问冲突问题。至于目录以及文件的存在性问题，这些为基本API函数调用而已。

        故而定义的类结构如下： 

View Code

  1 #include "MyFile.h"
  2 #include <queue>
  3 #include <process.h>
  4 #include "WTypes.h"
  5 #include <stdio.h>
  6 #include <tchar.h>
  7 #include "DType.h"
  8 #include "fstream"
  9 #include "iostream"
10 #include <map>
11 #include <string>
12 #include <io.h>
13 #include <atlconv.h>
14
15 using namespace std;
16
17 // 日志类
18 // 功能：
19 //        写日志，面对多线程
20 // 思路：
21 //      创建自身缓冲区，缓冲所有待写数据
22 //        单独线程负责实际写入操作
23 //      对上提供统一的写入接口
24 // 注意：
25 //        单例模型
26 class CMyLog
27 {
28 private:
29     static CMyLog* m_instance;
30
31 private:
32     // 私有构造函数
33     CMyLog()
34     {
35         InitializeCriticalSection(&this->m_csLock);
36         this->m_hThread = INVALID_HANDLE_VALUE;
37         this->m_hThreadEventExit = INVALID_HANDLE_VALUE;
38     }
39 public:
40     ~CMyLog()
41     {
42         DeleteCriticalSection(&m_csLock);
43     }
44 public:
45      static CMyLog* GetInstance();
46
47
48 private:
49     HANDLE                    m_hThread;                       // 写线程
50     HANDLE                    m_hThreadEventExit;               // 线程退出事件，用于主线程通知子线程退出
51     UINT                    m_hThreadID;                   // 写线程ID
52     CRITICAL_SECTION        m_csLock;                       // 互斥锁，用于多线程写入冲突解决
53     queue<string>            m_strText;                       // 写入缓冲区
54     ofstream                m_streamWriter;                   // 写入流
55     string                    m_fileFullName;                    // 文件全路径名
56 protected:
57     // 创建工作线程
58     void CreateWorkThread();
59
60     // 工作线程
61     static UINT    WINAPI WriteThreadProc (LPVOID pParam);
62
63     // 工作线执行体
64     void RunWritting();
65
66     // 检查目录是否存在，如果不存在则创建其
67     void FolderCheck();
68
69 public:
70
71     // 作用:
72     //        准备工作，负责打开文件，准备好输出流，启动写入线程
73     // 参数:
74     // 返回:
75     void GetReady();
76
77     // 作用:
78     //        写入数据
79     // 参数:
80     //        __in string        strMessage        信息
81     // 返回:
82     void Write(string strMessage);
83
84     // 作用:
85     //    关闭写入流，释放相关资源
86     // 参数:
87     // 返回:
88     void Close();
89
90     // 作用:
91     //     释放实例
92     // 参数:
93     // 返回:
94     static void Dispose()
95     {
96         delete m_instance;
97     }
98
99     // 作用：
100     //        设置文件全路径名
101     // 参数:
102     //        __in  string            pstrName        文件全路径名  
103     // 返回:
104    void SetFullName(string pstrName);
105
106     // 作用：
107     //        获取文件全路径名
108     // 参数: 
109     // 返回:
110     //      string 文件全路径名
111     // 异常:
112    string GetFullName();
113
114     // 作用：
115     //        获取文件路径    
116     // 返回:
117     //     int 文件全路径名长度 ERROR_FUNC_EXECU表示函数执行失败，通过GetLastError获取详细错误代码
118    string GetFilePath();
119
120     // 作用：
121     //        设置文件全路径名
122     // 参数:   
123     // 返回:
124     //     int 文件全路径名长度
125    string GetFileName();
126
127 };

       日志类的方法实现如下：

View Code

  1 #include "MyLog.h"
  2 #include <time.h>
  3 #include <direct.h>
  4 #include <shlobj.h>
  5
  6 CMyLog* CMyLog::m_instance;
  7
  8 // 公共访问点
  9 CMyLog* CMyLog::GetInstance()
10 { 
11     if (m_instance == NULL)
12         m_instance = new CMyLog;
13     return m_instance;
14 }
15
16  //创建工作线程
17 void CMyLog::CreateWorkThread()
18  {
19       m_hThreadEventExit = CreateEvent(NULL, TRUE, FALSE, NULL);
20       m_hThread = (HANDLE)_beginthreadex(NULL,0, WriteThreadProc, this,0, &m_hThreadID);
21  }
22
23 // 工作线程
24 UINT WINAPI CMyLog::WriteThreadProc(LPVOID pParam)
25  {
26      CMyLog* pThis = reinterpret_cast<CMyLog*>( pParam );
27      pThis->RunWritting();
28      return1L;
29  }
30
31 // 工作线执行体
32 void CMyLog::RunWritting()
33  {
34      while(true)
35      {
36          if (m_strText.size() ==0)
37              Sleep(5000);
38          else
39          {
40             time_t t = time(0);
41             char tmp[64];
42             ZeroMemory(tmp, 0);
43             strftime( tmp, sizeof(tmp), "%Y/%m/%d %X",localtime(&t));
44
45             m_streamWriter<<tmp<<" "<<m_strText.back()<<"\n";
46              m_strText.pop();
47          }
48          UINT eventIdx = WaitForSingleObject(m_hThreadEventExit,200);
49          if (eventIdx -WAIT_OBJECT_0 ==0)
50          {
51              // 线程退出收尾工作，将手头任务全部写完
52             while (m_strText.size() > 0)
53              {
54                 time_t t = time(0);
55                 char tmp[64];
56                 ZeroMemory(tmp, 0);
57                 strftime( tmp, sizeof(tmp), "%Y/%m/%d %X",localtime(&t));
58
59                 m_streamWriter<<tmp<<" "<<m_strText.back()<<"\n";
60                  m_strText.pop();
61              }
62              m_streamWriter.flush();
63              return;
64          }
65      }
66 }
67
68 // 就绪工作
69 void CMyLog::GetReady()
70 {
71     FolderCheck();
72      m_streamWriter.open(m_fileFullName, fstream::out | fstream::app);
73      CreateWorkThread();
74 }
75
76 // 检查目录，如果不存在则创建
77 void CMyLog::FolderCheck()
78 {
79     string dir = GetFilePath();
80     if(_access(dir.c_str(),0) != -1)
81         return;
82     dir = dir + "\\";
83     UINT ilen = MultiByteToWideChar (CP_ACP,0, dir.c_str(), -1, NULL,0);
84     wstring wdir(dir.begin(), dir.end());
85     int re = SHCreateDirectoryEx(NULL,wdir.c_str(),NULL);
86 }
87
88 // 外部接口，写入数据
89 void CMyLog::Write(string strMessage)
90 {
91      EnterCriticalSection(&m_csLock);
92      m_strText.push(strMessage);
93      LeaveCriticalSection(&m_csLock);
94 }
95
96 // 关闭写入流，释放相关资源
97 void CMyLog::Close()
98 {
99      SetEvent(m_hThreadEventExit);               // 向子线程发出退出信号
100      WaitForSingleObject(m_hThread, INFINITE);   // 等待子线程关闭
101      m_streamWriter.flush();
102      m_streamWriter.close();
103 }
104
105 // 设置文件全路径名
106 void CMyLog::SetFullName(string pstrName)
107 {
108     m_fileFullName =  pstrName;
109 }
110
111 // 获取文件全路径名
112 string CMyLog::GetFullName()
113 {
114     return m_fileFullName;
115 }
116
117 // 获取文件路径
118 string CMyLog::GetFilePath()
119 {
120     int idx = m_fileFullName.find_last_of("\\");
121     return m_fileFullName.substr(0, idx);
122 }
123
124 // 设置文件名称
125 string CMyLog::GetFileName()
126 {
127     int idx = m_fileFullName.find_last_of("\\");
128     return m_fileFullName.substr(idx+1, m_fileFullName.length());
129 }

       测试用例如下：

View Code

1 #include "iostream"
2 #include "MyLog.h"
3
4 using namespace std;
5
6 int _tmain(int argc, _TCHAR* argv[])
7 {
8     CMyLog* myLog = CMyLog::GetInstance();
9     myLog->SetFullName("E:\\Dlog\\DMMM\\TNN\\TMM\\12345\\MM\\log.txt");
10     myLog->GetReady();
11
12     int i =10;
13     while (i--)
14     {
15         Sleep(1000);
16         myLog->Write("i am a student");
17     }
18     myLog->Close();
19
20     CMyFileStatics myFile;
21     myFile.SetFullName("E:\\Dlog\\11.txt");
22     myFile.Statics();
23
24     system("PAUSE");
25     return 0;
26 }

      对于此解决方案可能存在的问题，还希望阅读此文的你试用或者核对源码以及思路，提出你的修改意见，以促进该日志类的壮大