log4cplus_helper.h 4.5 KB

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