From b21dfad4851f5f192b28e53d3a3ef35de7e7ba91 Mon Sep 17 00:00:00 2001
From: wangky <m1561510467@163.com>
Date: 星期六, 02 十一月 2024 15:25:36 +0800
Subject: [PATCH] 1
---
Server/马丽萍/code/log/log.cpp | 399 ++++++++++++++++++++++++++++++++++++--------------------
1 files changed, 255 insertions(+), 144 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"
index 411f8eb..3c6e62c 100644
--- "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"
@@ -1,164 +1,275 @@
-#include <string.h>
#include <time.h>
-#include <sys/time.h>
+#include <chrono>
#include <stdarg.h>
#include "log.h"
-#include <pthread.h>
+#include <fstream>
+#include <iostream>
+
using namespace std;
-Log::Log()
-{
- m_count = 0;
- m_is_async = false;
+// 鏋勯�犲嚱鏁�
+Log::Log() : m_count(0), m_is_async(false) {}
+
+// 鏋愭瀯鍑芥暟
+Log::~Log() {
+ if (m_fp.is_open()) {
+ m_fp.close(); // 鍏抽棴鏃ュ織鏂囦欢
+ }
}
-Log::~Log()
-{
- if (m_fp != NULL)
- {
- fclose(m_fp);
+//寮傛鍐欏叆
+void Log::async_write_log() {
+ while (m_is_async) {
+ char* log;
+ if (m_log_queue->pop(log)) {
+ ParsedLog parsed_log = parseLog(log);
+ write_log(INFO, "%s", log); // 鍏堝啓鍏ユ棩蹇楁枃浠�
+ writeLogToDatabase(parsed_log); // 鍐嶅啓鍏ユ暟鎹簱
+ delete[] log; // 閲婃斁鍐呭瓨
+ }
}
}
-//寮傛闇�瑕佽缃樆濉為槦鍒楃殑闀垮害锛屽悓姝ヤ笉闇�瑕佽缃�
-bool Log::init(const char *file_name, int close_log, int log_buf_size, int split_lines, int max_queue_size)
-{
- //濡傛灉璁剧疆浜唌ax_queue_size,鍒欒缃负寮傛
- if (max_queue_size >= 1)
- {
- m_is_async = true;
- m_log_queue = new block_queue<string>(max_queue_size);
- pthread_t tid;
- //flush_log_thread涓哄洖璋冨嚱鏁�,杩欓噷琛ㄧず鍒涘缓绾跨▼寮傛鍐欐棩蹇�
- pthread_create(&tid, NULL, flush_log_thread, NULL);
+
+bool Log::check_log_size() {
+ if (m_fp.tellp() >= max_size) {
+ rotate_logs(); // 濡傛灉鏃ュ織鏂囦欢澶у皬瓒呰繃鏈�澶у�硷紝杞崲鏃ュ織
+ return true;
}
-
- 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;
+ return false;
+}
+
+
+// 鎺ユ敹鍘熷鏃ュ織骞惰В鏋�
+void Log::receiveLog(const char* raw_log) {
+ ParsedLog parsed_log = parseLog(raw_log);
+ write_log(levelToInt(parsed_log.level), "%s [%s] %s",
+ parsed_log.timestamp,
+ parsed_log.device_id,
+ parsed_log.content);
+}
+
+//鍐欏叆鏁版嵁搴�
+void Log::writeLogToDatabase(const ParsedLog& log) {
+ const char* sql = "INSERT INTO parsed_logs (timestamp, device_id, level, content, source, user_id) VALUES (?, ?, ?, ?, ?, ?)";
+ sqlite3_stmt* stmt;
+
+ if (sqlite3_prepare_v2(db, sql, -1, &stmt, NULL) != SQLITE_OK) {
+ cerr << "Error: Failed to prepare statement: " << sqlite3_errmsg(db) << endl;
+ return;
+ }
+
+ sqlite3_bind_text(stmt, 1, log.timestamp, -1, SQLITE_STATIC);
+ sqlite3_bind_text(stmt, 2, log.device_id, -1, SQLITE_STATIC);
+ sqlite3_bind_text(stmt, 3, log.level, -1, SQLITE_STATIC);
+ sqlite3_bind_text(stmt, 4, log.content, -1, SQLITE_STATIC);
+ sqlite3_bind_text(stmt, 5, log.source, -1, SQLITE_STATIC);
+ sqlite3_bind_text(stmt, 6, log.user_id, -1, SQLITE_STATIC);
+
+ if (sqlite3_step(stmt) != SQLITE_DONE) {
+ cerr << "Error: Failed to execute statement: " << sqlite3_errmsg(db) << endl;
+ }
+
+ sqlite3_finalize(stmt); // 娓呯悊鍑嗗璇彞
+}
+
+
+// 瑙ｆ瀽鏃ュ織瀛楃涓�
+ParsedLog Log::parseLog(const char* log) {
+ ParsedLog parsed_log;
+ char level[16];
+
+ // 璇诲彇鏃堕棿鎴炽�佽澶嘔D鍜屾棩蹇楃骇鍒�
+ sscanf(log, "%s %s %s %[^\n]", parsed_log.timestamp, parsed_log.device_id, level, parsed_log.content);
+ // 璁剧疆瑙ｆ瀽鍚庣殑绾у埆
+ for (int i = 0; i < 16; ++i) {
+ parsed_log.level[i] = level[i];
+ }
+
+ return parsed_log;
+}
+
+// 鍒濆鍖栨棩蹇楃郴缁�
+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<char*>(max_queue_size); // 鍒涘缓闃诲闃熷垪
+ std::thread(&Log::flush_log_thread, this).detach(); // 鍒涘缓鍒锋柊鏃ュ織绾跨▼
+ }
+
+ 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);
- struct tm my_tm = *sys_tm;
+ char log_full_name[256];
-
- const char *p = strrchr(file_name, '/');
- char log_full_name[256] = {0};
-
- if (p == NULL)
- {
- snprintf(log_full_name, 255, "%d_%02d_%02d_%s", my_tm.tm_year + 1900, my_tm.tm_mon + 1, my_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, my_tm.tm_year + 1900, my_tm.tm_mon + 1, my_tm.tm_mday, log_name);
- }
-
- m_today = my_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 = {0, 0};
- gettimeofday(&now, NULL);
- time_t t = now.tv_sec;
- struct tm *sys_tm = localtime(&t);
- struct tm my_tm = *sys_tm;
- 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;
- }
- //鍐欏叆涓�涓猯og锛屽m_count++, m_split_lines鏈�澶ц鏁�
- m_mutex.lock();
- m_count++;
-
- if (m_today != my_tm.tm_mday || m_count % m_split_lines == 0) //everyday log
- {
+ // 鏍规嵁鏂囦欢璺緞鍜屽綋鍓嶆棩鏈熺敓鎴愬畬鏁存棩蹇楁枃浠跺悕
+ if (strchr(file_name, '/') == nullptr) {
+ sprintf(log_full_name, "%d_%d_%d_%s",
+ sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday, file_name);
+ } else {
+ char log_name[128];
+ char dir_name[128];
+ char* last_slash = strrchr(file_name, '/');
+ int dir_len = last_slash - file_name + 1;
+ strncpy(dir_name, file_name, dir_len);
+ dir_name[dir_len] = '\0';
+ strcpy(log_name, last_slash + 1);
- char new_log[256] = {0};
- fflush(m_fp);
- fclose(m_fp);
- char tail[16] = {0};
-
- snprintf(tail, 16, "%d_%02d_%02d_", my_tm.tm_year + 1900, my_tm.tm_mon + 1, my_tm.tm_mday);
-
- if (m_today != my_tm.tm_mday)
- {
- snprintf(new_log, 255, "%s%s%s", dir_name, tail, log_name);
- m_today = my_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");
- }
-
- m_mutex.unlock();
-
- va_list valst;
- va_start(valst, format);
-
- string log_str;
- m_mutex.lock();
-
- //鍐欏叆鐨勫叿浣撴椂闂村唴瀹规牸寮�
- int n = snprintf(m_buf, 48, "%d-%02d-%02d %02d:%02d:%02d.%06ld %s ",
- my_tm.tm_year + 1900, my_tm.tm_mon + 1, my_tm.tm_mday,
- my_tm.tm_hour, my_tm.tm_min, my_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();
+ sprintf(log_full_name, "%s%d_%d_%d_%s",
+ dir_name, sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday, log_name);
}
- va_end(valst);
+ m_today = sys_tm->tm_mday; // 璁板綍浠婂ぉ鐨勬棩鏈�
+ m_fp.open(log_full_name, std::ios::out | std::ios::app); // 鎵撳紑鏃ュ織鏂囦欢
+ if (!m_fp.is_open()) {
+ cerr << "Error: Could not open log file: " << log_full_name << endl;
+ return false; // 鎵撳紑澶辫触
+ }
+
+ return true; // 鍒濆鍖栨垚鍔�
}
-void Log::flush(void)
-{
- m_mutex.lock();
- //寮哄埗鍒锋柊鍐欏叆娴佺紦鍐插尯
- fflush(m_fp);
- m_mutex.unlock();
+//鏌ヨ鏃ュ織鑽夋渶
+void Log::queryLogs(const char *device_id) {
+ const char* sql = "SELECT * FROM parsed_logs WHERE device_id = ?";
+ sqlite3_stmt* stmt;
+
+ if (sqlite3_prepare_v2(db, sql, -1, &stmt, NULL) != SQLITE_OK) {
+ cerr << "Error: Failed to prepare statement: " << sqlite3_errmsg(db) << endl;
+ return;
+ }
+
+ sqlite3_bind_text(stmt, 1, device_id, -1, SQLITE_STATIC);
+
+ while (sqlite3_step(stmt) == SQLITE_ROW) {
+ cout << "Timestamp: " << sqlite3_column_text(stmt, 1) << ", "
+ << "Device ID: " << sqlite3_column_text(stmt, 2) << ", "
+ << "Level: " << sqlite3_column_text(stmt, 3) << ", "
+ << "Content: " << sqlite3_column_text(stmt, 4) << ", "
+ << "Source: " << sqlite3_column_text(stmt, 5) << ", "
+ << "User ID: " << sqlite3_column_text(stmt, 6) << endl;
+ }
+
+ sqlite3_finalize(stmt); // 娓呯悊鍑嗗璇彞
+}
+
+
+
+// 鍐欏叆鏃ュ織
+void Log::write_log(int level, const char *format, ...) {
+ auto now = std::chrono::system_clock::now();
+ auto now_time_t = std::chrono::system_clock::to_time_t(now);
+ struct tm *sys_tm = localtime(&now_time_t);
+
+ char s[32];
+
+ // 鏍规嵁鏃ュ織绾у埆璁剧疆鍓嶇紑
+ switch (level) {
+ case DEBUG: strcpy(s, "[debug]: "); break;
+ case INFO: strcpy(s, "[info]: "); break;
+ case WARN: strcpy(s, "[warn]: "); break;
+ case ERROR: strcpy(s, "[error]: "); break;
+ default: strcpy(s, "[info]: "); break;
+ }
+
+ m_mutex.lock(); // 閿佸畾浜掓枼浣�
+ m_count++; // 澧炲姞鏃ュ織璁℃暟
+
+ // 妫�鏌ユ槸鍚﹂渶瑕佽疆鎹㈡棩蹇�
+ if (m_today != sys_tm->tm_mday || m_count % m_split_lines == 0) {
+ rotate_logs(); // 璋冪敤鏃ュ織杞崲鍑芥暟
+ }
+
+ // 鍐欏叆鏃ュ織鍐呭
+ va_list valist;
+ va_start(valist, format);
+ vsnprintf(m_buf, m_log_buf_size, format, valist); // 鍐欏叆鏃ュ織鍐呭
+
+ m_fp << s << m_buf << std::endl; // 浣跨敤ofstream鍐欏叆鏂囦欢
+ va_end(valist);
+
+ m_mutex.unlock(); // 瑙ｉ攣浜掓枼浣�
+
+ // 寮傛鍐欏叆鏃ュ織
+ if (m_is_async && !m_log_queue->full()) {
+ m_log_queue->push(m_buf); // 灏嗘棩蹇楁帹鍏ラ槦鍒�
+ }
+}
+
+// 鍒锋柊鏃ュ織鏂囦欢
+void Log::flush(void) {
+ m_mutex.lock(); // 閿佸畾浜掓枼浣�
+ m_fp.flush(); // 鍒锋柊鏂囦欢
+ m_mutex.unlock(); // 瑙ｉ攣浜掓枼浣�
+}
+
+// 鍒锋柊鏃ュ織绾跨▼
+void Log::flush_log_thread() {
+ while (m_is_async) {
+ // 浠庨槦鍒椾腑鑾峰彇鏃ュ織骞跺啓鍏�
+ char* log;
+ if (m_log_queue->pop(log)) {
+ write_log(INFO, "%s", log);
+ }
+ }
+}
+
+//杞崲瀛楃涓�
+const char* Log::log_level_to_string(int level) {
+ switch (level) {
+ case DEBUG: return "[debug]: ";
+ case INFO: return "[info]: ";
+ case WARN: return "[warn]: ";
+ case ERROR: return "[error]: ";
+ default: return "[info]: "; // 榛樿杩斿洖INFO
+ }
+}
+
+// 鏃ュ織绾у埆杞崲涓烘暣鏁�
+int Log::levelToInt(const char* level) {
+ if (strcmp(level, "[debug]:") == 0) return DEBUG;
+ if (strcmp(level, "[info]:") == 0) return INFO;
+ if (strcmp(level, "[warn]:") == 0) return WARN;
+ if (strcmp(level, "[error]:") == 0) return ERROR;
+ return INFO; // 榛樿杩斿洖INFO
+}
+
+//鎵撳紑鍏抽棴鏁版嵁搴�
+void Log::openDatabase(const char *db_name) {
+ if (sqlite3_open(db_name, &db) != SQLITE_OK) {
+ cerr << "鎵撲笉寮�鏁版嵁搴擄紒锛侀敊璇細 " << sqlite3_errmsg(db) << endl;
+ }
+}
+
+void Log::closeDatabase() {
+ if (db) {
+ sqlite3_close(db);
+ }
+}
+
+// 鏃ュ織杞崲鍑芥暟
+void Log::rotate_logs() {
+ // 鍏抽棴褰撳墠鏂囦欢
+ m_fp.close();
+
+ // 鐢熸垚鏂版枃浠跺悕
+ time_t t = time(NULL);
+ struct tm *sys_tm = localtime(&t);
+ char new_log_name[256];
+ sprintf(new_log_name, "%d_%d_%d_log.txt",
+ sys_tm->tm_year + 1900, sys_tm->tm_mon + 1, sys_tm->tm_mday);
+
+ // 鎵撳紑鏂扮殑鏃ュ織鏂囦欢
+ m_fp.open(new_log_name, std::ios::out | std::ios::app);
+ if (!m_fp.is_open()) {
+ cerr << "Error: Could not open new log file: " << new_log_name << endl;
+ return;
+ }
+
+ m_today = sys_tm->tm_mday; // 鏇存柊浠婂ぉ鐨勬棩鏈�
+ m_count = 0; // 閲嶇疆鏃ュ織璁℃暟
}
--
Gitblit v1.8.0