log4cplus_helper.h 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
  1. #ifndef _VTM_LOG4CPLUS_HELPER_H_
  2. #define _VTM_LOG4CPLUS_HELPER_H_
  3. #pragma once
  4. #include "log4vendor.h"
  5. #include <cstdint>
  6. #include <iostream>
  7. #include <set>
  8. #include <queue>
  9. #include <log4cplus/consoleappender.h>
  10. #include <log4cplus/fileappender.h>
  11. #include <log4cplus/socketappender.h>
  12. #include <log4cplus/asyncappender.h>
  13. #include <log4cplus/nullappender.h>
  14. #include <log4cplus/appender.h>
  15. #include <log4cplus/layout.h>
  16. #include <log4cplus/logger.h>
  17. #include <log4cplus/ndc.h>
  18. #include <log4cplus/helpers/loglog.h>
  19. #include <log4cplus/streams.h>
  20. #include <log4cplus/loggingmacros.h>
  21. #include <log4cplus/configurator.h>
  22. #include "mutex.h"
  23. namespace cmb {
  24. enum cpluslog_level
  25. {
  26. log_level_debug = 0,
  27. log_level_trace,
  28. log_level_info,
  29. log_level_warn,
  30. log_level_error,
  31. log_level_fatal,
  32. log_level_end
  33. };
  34. enum cpluslog_append_type
  35. {
  36. log_append_none = 0,
  37. log_append_console = 1,
  38. log_append_file = 2,
  39. log_append_console_file = 3,
  40. log_append_socket = 4,
  41. log_append_console_socket = 5,
  42. log_append_file_socket = 6,
  43. log_append_all = 7
  44. };
  45. class log4cplus_helper : public log4vendor
  46. {
  47. public:
  48. ~log4cplus_helper(void);
  49. static log4cplus_helper* get_instance();
  50. bool init(const char* name);
  51. void uninit();
  52. std::ostream& stream();
  53. void trace(const vtm_string& text);
  54. void info(const vtm_string& text);
  55. void debug(const vtm_string& text);
  56. void warn(const vtm_string& text);
  57. void error(const vtm_string& text);
  58. void fatal(const vtm_string& text);
  59. void trace(const char* text);
  60. void info(const char* text);
  61. void debug(const char* text);
  62. void warn(const char* text);
  63. void error(const char* text);
  64. void fatal(const char* text);
  65. bool set_log_append_type(int type) {
  66. if((type & log_append_console) || (type & log_append_file) || (type & log_append_socket)) {
  67. _append_type = type;
  68. return true;
  69. }
  70. return false;
  71. }
  72. /*!
  73. * @brief 设置日志文件存储的路径文件夹
  74. * @param[in] 存储的文件夹路径,兼容不加 '\\' 的情况
  75. * @return :
  76. */
  77. bool set_log_dir(const vtm_string& log_file) {
  78. vtm_string tmp(log_file);
  79. if(tmp.empty())
  80. return false;
  81. if(tmp[tmp.size()-1] != '\\' && tmp[tmp.size()-1] != '/') {
  82. #if defined(_MSC_VER)
  83. tmp.push_back('\\');
  84. #else
  85. tmp.push_back('/');
  86. #endif //_MSC_VER
  87. }
  88. _log_dir = tmp;
  89. return true;
  90. }
  91. bool set_log_level(const int level) {
  92. const int ll = level * 10000;
  93. if(ll < 0 || ll > log4cplus::OFF_LOG_LEVEL) {
  94. return false;
  95. }
  96. _log_level = ll;
  97. return true;
  98. }
  99. void update_map_value(const vtm_string& value) {
  100. if(!value.empty()) {
  101. log4cplus::getMDC ().put("key", value);
  102. }
  103. }
  104. vtm_string get_map_value() {
  105. vtm_string value("");
  106. if(!log4cplus::getMDC().get(&value, "key")) {
  107. return vtm_string("");
  108. }
  109. return value;
  110. }
  111. int run_log_sender(const vtm_string& ip, const unsigned short port);
  112. vtm_string format(const char *ft, ...);
  113. vtm_string format(const char *ft,va_list& arg);
  114. void log( const char *file_name, const char *func_name, int line, const char* ft, int type = log_level_debug);
  115. void log( const char *file_name, const char *func_name, int line, int type ,const char *ft = "", ...);
  116. void log(int log_level, const vtm_string& text);
  117. void logw(int log_level, const std::wstring& wtext);
  118. log4cplus::Logger get_logger() {
  119. return log4cplus::Logger::getInstance(LOG4CPLUS_TEXT(_log_key_name.c_str()));
  120. }
  121. private:
  122. static log4cplus_helper* _instance;
  123. log4cplus_helper(void);
  124. void config_remote_logger(const char* remote_name);
  125. log4cplus::helpers::SharedObjectPtr<log4cplus::Appender> _append_console;
  126. log4cplus::helpers::SharedObjectPtr<log4cplus::Appender> _append_file;
  127. log4cplus::helpers::SharedObjectPtr<log4cplus::Appender> _append_socket;
  128. log4cplus::helpers::SharedObjectPtr<log4cplus::Appender> _append_none;
  129. log4cplus::tstring _log_dir;
  130. log4cplus::tstring _log_server_ip;
  131. unsigned short _log_server_port;
  132. vtm_string _log_client_key_name;
  133. int _append_type;
  134. vtm_string _log_key_name;
  135. bool _socket_send_running;
  136. static mutex _mutex;
  137. bool _initialized;
  138. unsigned short _log_listen_port;
  139. mutex _mutex_4_update_socket;
  140. bool _initilzed_zip;
  141. std::queue<vtm_string> _zip_logs;
  142. vtm_string _cur_upload_log;
  143. uint64_t _cur_upload_offset;
  144. int _log_level;
  145. };
  146. }
  147. #endif //_VTM_LOG4CPLUS_HELPER_H_