log mylog

1. /*
   
2.  * 支持linux，windows平台，不支持UNICODE
   
3.  * 用例：
   
4. #include "mylog.h"
   
5. #ifdef _WIN32
   
6. int tmain(int argc, _TCHAR* argv[])
   
7. #else
   
8. int main(int argc, char* argv[])
   
9. #endif
   
10. {
    
11.     log_init("1.txt");
    
12.     //log_init(NULL);
    
13.     L_ENTER();
    
14.     L_FATAL(_TEXT("fatal %d\n"), 1);
    
15.     L_ERROR(_TEXT("fatal %d\n"), 1);
    
16.     L_WARNING(_TEXT("fatal %d\n"), 1);
    
17.     L_INFO(_TEXT("fatal %d\n"), 1);
    
18.     L_DEBUG(_TEXT("fatal %d\n"), 1);
    
19.     L_TRACE(_TEXT("fatal %d\n"), 1);
    
20.     L_LEAVE();
    
21. 
    
22.     return 0;
    
23. }
    
24.  */
    
25. #ifndef __MYLOG_H__
    
26. #define __MYLOG_H__
    
27. 
    
28. // 日志格式
    
29. //1812::INFO::2014-04-15 14:48:54::log_version::0.1.0.0
    
30. 
    
31. #ifdef __cplusplus
    
32. extern "C" {
    
33. #endif
    
34. 
    
35. 
    
36. #include <stdio.h>
    
37. #include <stdint.h>
    
38. #include <time.h>
    
39. #include <wchar.h>
    
40. 
    
41. #ifdef _WIN32
    
42. #include <windows.h>
    
43. #else
    
44. #include <sys/types.h>
    
45. #include <unistd.h>
    
46. #endif
    
47. 
    
48. 
    
49. enum LOG_LEVEL_ {
    
50.     LOG_LEVEL__TRACE = 0,
    
51.     LOG_LEVEL__DEBUG,
    
52.     LOG_LEVEL__INFO,
    
53.     LOG_LEVEL__WARNING,
    
54.     LOG_LEVEL__ERROR,
    
55.     LOG_LEVEL__FATAL
    
56. };
    
57. 
    
58. /*
    
59.  *    配置
    
60.  * MY_DEFAULT_LOG_FILE: 默认日志文件名
    
61.  * MY_DEFAULT_LOG_LEVEL: 默认输出的日志级别
    
62.  */
    
63. #define MY_DEFAULT_LOG_FILE    "log.txt"
    
64. #define MY_DEFAULT_LOG_LEVEL LOG_LEVEL__DEBUG
    
65. 
    
66. 
    
67. /**************************************************************************/
    
68. /**************************************************************************/
    
69. 
    
70. 
    
71. #ifndef TRUE
    
72. #define TRUE 1
    
73. #endif
    
74. 
    
75. #ifndef FALSE
    
76. #define FALSE 0
    
77. #endif
    
78. 
    
79. #ifndef _WIN32
    
80. #define _TEXT(x)    x
    
81. #define _tfopen fopen
    
82. #define _ftprintf fprintf
    
83. #define GetCurrentThreadId getpid
    
84. #define TCHAR char
    
85. #endif
    
86. 
    
87. #undef LOG_FILE_NAME
    
88. #define LOG_FILE_NAME    _TEXT(MY_DEFAULT_LOG_FILE)
    
89. 
    
90. static uint32_t log_level = MY_DEFAULT_LOG_LEVEL;
    
91. static uint32_t log_enable = FALSE;
    
92. static FILE *fp = NULL;
    
93. 
    
94. #ifdef _WIN32
    
95. # pragma warning(push)
    
96. /* C4996: _CRT_SECURE_NO_WARNINGS. */
    
97. # pragma warning(disable: 4996)
    
98. #endif
    
99. 
    
100. uint32_t log_init(const TCHAR* filename)
     
101. {
     
102.     if ( filename == NULL ) {
     
103.         filename = LOG_FILE_NAME;
     
104.     }
     
105. 
     
106.     if ( fp != NULL ) {
     
107.         fclose(fp);
     
108.     }
     
109. 
     
110.     fp = _tfopen(filename, _TEXT("a+"));
     
111.     if ( fp == NULL ) {
     
112.         _ftprintf(stderr, _TEXT("open %s failed, log disable\n"), filename);
     
113.         return FALSE;
     
114.     }
     
115. 
     
116.     log_enable = TRUE;
     
117.     return TRUE;
     
118. }
     
119. 
     
120. #ifdef _WIN32
     
121. # pragma warning(pop)
     
122. #endif
     
123. 
     
124. void current_time(TCHAR *out, int out_len)
     
125. {
     
126.     time_t timer;                            
     
127.     timer = time(NULL);                        
     
128. #ifdef _WIN32
     
129.     struct tm today;                        
     
130.     localtime_s(&today, &timer);                
     
131.     strftime(out, out_len, "%Y-%m-%d %H:%M:%S", &today);                        
     
132. #else
     
133.     struct tm *today;                        
     
134.     today = localtime(&timer);                
     
135.     strftime(out, out_len, "%Y-%m-%d %H:%M:%S", today);                        
     
136. #endif
     
137. }
     
138. 
     
139. #define MYLOG(level, fmt, ...)    do                \
     
140. {                                                \
     
141.     if ( log_enable && level >= log_level ) {    \
     
142.         static TCHAR level_as_string[][8] = {_TEXT("TRACE"), _TEXT("DEBUG"), _TEXT("INFO "), \
     
143.                                             _TEXT("WARN "), _TEXT("ERROR"), _TEXT("FATAL")}; \
     
144.         TCHAR datetime_str[20];                    \
     
145.         current_time(datetime_str, sizeof(datetime_str)); \
     
146.         _ftprintf(fp, _TEXT("%d::%s::%s::%s::")fmt,                                                        \
     
147.                         GetCurrentThreadId(), level_as_string[level], datetime_str, __FUNCTION__, ##__VA_ARGS__); \
     
148.         fflush(fp);                                \
     
149.     }                                            \
     
150. } while (0)
     
151. 
     
152. 
     
153. 
     
154. #define L_FATAL(fmt, ...) MYLOG(LOG_LEVEL__FATAL, fmt, ## __VA_ARGS__)
     
155. #define L_ERROR(fmt, ...) MYLOG(LOG_LEVEL__ERROR, fmt, ## __VA_ARGS__)
     
156. #define L_WARNING(fmt, ...) MYLOG(LOG_LEVEL__WARNING, fmt, ## __VA_ARGS__)
     
157. #define L_INFO(fmt, ...) MYLOG(LOG_LEVEL__INFO, fmt, ## __VA_ARGS__)
     
158. #define L_DEBUG(fmt, ...) MYLOG(LOG_LEVEL__DEBUG, fmt, ## __VA_ARGS__)
     
159. #define L_TRACE(fmt, ...) MYLOG(LOG_LEVEL__TRACE, fmt, ## __VA_ARGS__)
     
160. #define L_ENTER()        L_DEBUG("Enter.\n")
     
161. #define L_LEAVE()        L_DEBUG("Leave.\n")
     
162. 
     
163. #ifdef __cplusplus
     
164. }
     
165. #endif
     
166. 
     
167. #endif