From 63989e7910ddd40c5dae64022e80c07a3b598043 Mon Sep 17 00:00:00 2001
From: wxx <1321839169@qq.com>
Date: 星期三, 30 十月 2024 15:06:24 +0800
Subject: [PATCH] Merge branch 'master' of ssh://115.28.86.8:29418/~admin/昆仑_1025
---
Server/马丽萍/code/log/log.cpp | 166 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 166 insertions(+), 0 deletions(-)
diff --git "a/Server/\351\251\254\344\270\275\350\220\215/code/log/log.cpp" "b/Server/\351\251\254\344\270\275\350\220\215/code/log/log.cpp"
new file mode 100644
index 0000000..7368c50
--- /dev/null
+++ "b/Server/\351\251\254\344\270\275\350\220\215/code/log/log.cpp"
@@ -0,0 +1,166 @@
+#include <string.h>
+#include <time.h>
+#include <sys/time.h>
+#include <stdarg.h>
+#include "log.h"
+#include <pthread.h>
+using namespace std;
+
+Log::Log()
+{
+ m_count = 0;
+ m_is_async = false;
+}
+
+Log::~Log()
+{
+ if (m_fp != NULL)
+ {
+ fclose(m_fp);
+ }
+}
+
+bool Log::init(const char *file_name, int close_log, int log_buf_size, int split_lines, int max_queue_size)
+{
+ if (max_queue_size >= 1)
+ {
+ m_is_async = true;
+ m_log_queue = new block_queue<string>(max_queue_size);
+ pthread_t tid;
+ pthread_create(&tid, NULL, flush_log_thread, NULL);
+ }
+
+ m_close_log = close_log;
+ m_log_buf_size = log_buf_size;
+ m_buf = new char[m_log_buf_size];
+ memset(m_buf, '\0', m_log_buf_size);
+ m_split_lines = split_lines;
+
+ time_t t = time(NULL);
+ struct tm *sys_tm = localtime(&t);
+ const char *p = strrchr(file_name, '/');
+ char log_full_name[256] = {0};
+
+ if (p == NULL)
+ {
+ snprintf(log_full_name, 255, "%d_%02d_%02d_%s", sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday, file_name);
+ }
+ else
+ {
+ strcpy(log_name, p + 1);
+ strncpy(dir_name, file_name, p - file_name + 1);
+ snprintf(log_full_name, 255, "%s%d_%02d_%02d_%s", dir_name, sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday, log_name);
+ }
+
+ m_today = sys_tm->tm_mday;
+ m_fp = fopen(log_full_name, "a");
+ if (m_fp == NULL)
+ {
+ return false;
+ }
+
+ return true;
+}
+
+void Log::write_log(int level, const char *format, ...)
+{
+ struct timeval now;
+ gettimeofday(&now, NULL);
+ time_t t = now.tv_sec;
+ struct tm *sys_tm = localtime(&t);
+ char s[16] = {0};
+
+ switch (level)
+ {
+ case 0: strcpy(s, "[debug]:"); break;
+ case 1: strcpy(s, "[info]:"); break;
+ case 2: strcpy(s, "[warn]:"); break;
+ case 3: strcpy(s, "[erro]:"); break;
+ default: strcpy(s, "[info]:"); break;
+ }
+
+ m_mutex.lock();
+ m_count++;
+
+ if (m_today != sys_tm->tm_mday || m_count % m_split_lines == 0)
+ {
+ char new_log[256] = {0};
+ fflush(m_fp);
+ fclose(m_fp);
+ char tail[16] = {0};
+ snprintf(tail, 16, "%d_%02d_%02d_", sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday);
+
+ if (m_today != sys_tm->tm_mday)
+ {
+ snprintf(new_log, 255, "%s%s%s", dir_name, tail, log_name);
+ m_today = sys_tm->tm_mday;
+ m_count = 0;
+ }
+ else
+ {
+ snprintf(new_log, 255, "%s%s%s.%lld", dir_name, tail, log_name, m_count / m_split_lines);
+ }
+ m_fp = fopen(new_log, "a");
+ }
+
+ // 鍦ㄥ啓鍏ヤ箣鍓嶆鏌ユ棩蹇楁枃浠跺ぇ灏�
+ if (check_log_size())
+ {
+ rotate_logs(); // 濡傛灉瓒呰繃鏈�澶ф枃浠跺ぇ灏忥紝杩涜鏃ュ織杞崲
+ }
+
+ va_list valst;
+ va_start(valst, format);
+
+ string log_str;
+ int n = snprintf(m_buf, 48, "%d-%02d-%02d %02d:%02d:%02d.%06ld %s ",
+ sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday,
+ sys_tm->tm_hour, sys_tm->tm_min, sys_tm->tm_sec, now.tv_usec, s);
+ int m = vsnprintf(m_buf + n, m_log_buf_size - n - 1, format, valst);
+ m_buf[n + m] = '\n';
+ m_buf[n + m + 1] = '\0';
+ log_str = m_buf;
+
+ m_mutex.unlock();
+
+ if (m_is_async && !m_log_queue->full())
+ {
+ m_log_queue->push(log_str);
+ }
+ else
+ {
+ m_mutex.lock();
+ fputs(log_str.c_str(), m_fp);
+ m_mutex.unlock();
+ }
+
+ va_end(valst);
+}
+
+void Log::flush(void)
+{
+ m_mutex.lock();
+ fflush(m_fp);
+ m_mutex.unlock();
+}
+
+// 娣诲姞鐨勬枃浠跺ぇ灏忔鏌ュ嚱鏁�
+bool Log::check_log_size()
+{
+ if (m_fp)
+ {
+ fseek(m_fp, 0, SEEK_END);
+ long file_size = ftell(m_fp);
+ return file_size >= max_size; // 杩斿洖鏂囦欢鏄惁瓒呭嚭闄愬埗
+ }
+ return false;
+}
+
+void Log::rotate_logs()
+{
+ fclose(m_fp); // 鍏抽棴褰撳墠鏃ュ織鏂囦欢
+ char new_log[256] = {0};
+ snprintf(new_log, sizeof(new_log), "%s/%s.%lld", dir_name, log_name, m_count / m_split_lines);
+ m_fp = fopen(new_log, "a"); // 鎵撳紑鏂扮殑鏃ュ織鏂囦欢
+ m_count = 0; // 閲嶇疆鏃ュ織璁℃暟
+}
--
Gitblit v1.8.0