Merge branch 'master' of ssh://115.28.86.8:29418/~admin/昆仑_1025

Administrator

2024-11-01 b462e21965b8332b17ad3bdfa5ed38b67f4dc2bb

 Server/ÂíÀöƼ/code/log/log.cpp

@@ -1,21 +1,14 @@
#include <string.h>
#include <time.h>
#include <chrono>
#include <ctime>
#include <stdarg.h>
#include "log.h"
#include <sstream>
#include <fstream>
#include <iostream>
#include <chrono>

using namespace std;

// æž„造函数
Log::Log() {
    m_count = 0; // åˆå§‹åŒ–日志行数
    m_is_async = false; // åˆå§‹åŒ–为同步
}
Log::Log() : m_count(0), m_is_async(false) {}

// æžæž„函数
Log::~Log() {
@@ -24,31 +17,82 @@
    }
}

// æŽ¥æ”¶åŽŸå§‹æ—¥å¿—并解析
void Log::receiveLog(const std::string &raw_log) {
    ParsedLog parsed_log = parseLog(raw_log);
    write_log(levelToInt(parsed_log.level), "%s [%s] %s", parsed_log.timestamp.c_str(), parsed_log.device_id.c_str(), parsed_log.content.c_str());
//异步写入
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::check_log_size() {
    if (m_fp.tellp() >= max_size) {
        rotate_logs(); // å¦‚果日志文件大小超过最大值，轮换日志
        return true;
    }
    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 std::string &log) {
ParsedLog Log::parseLog(const char* log) {
    ParsedLog parsed_log;
    std::istringstream iss(log);
    std::string level;
    char level[16];

    // è¯»å–时间戳、设备ID和日志级别
    iss >> parsed_log.timestamp >> parsed_log.device_id >> level;
    std::getline(iss, parsed_log.content);
    parsed_log.level = level; // è®¾ç½®è§£æžåŽçš„级别
    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 std::string &file_name, int close_log, int log_buf_size, int split_lines, int max_queue_size) {
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); // åˆ›å»ºé˜»å¡žé˜Ÿåˆ—
        m_log_queue = new block_queue<char*>(max_queue_size); // åˆ›å»ºé˜»å¡žé˜Ÿåˆ—
        std::thread(&Log::flush_log_thread, this).detach(); // åˆ›å»ºåˆ·æ–°æ—¥å¿—线程
    }

@@ -60,43 +104,76 @@

    time_t t = time(NULL);
    struct tm *sys_tm = localtime(&t);
    std::string log_full_name;
    char log_full_name[256];

    // æ ¹æ®æ–‡ä»¶è·¯å¾„和当前日期生成完整日志文件名
    if (file_name.find('/') == std::string::npos) {
        log_full_name = to_string(sys_tm->tm_year + 1900) + "_" + to_string(sys_tm->tm_mon + 1) + "_" + to_string(sys_tm->tm_mday) + "_" + file_name;
    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 {
        std::string log_name = file_name.substr(file_name.find_last_of('/') + 1);
        std::string dir_name = file_name.substr(0, file_name.find_last_of('/') + 1);

        log_full_name = dir_name + to_string(sys_tm->tm_year + 1900) + "_" + to_string(sys_tm->tm_mon + 1) + "_" + to_string(sys_tm->tm_mday) + "_" + log_name;
        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);
        
        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);
    }

    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; // åˆå§‹åŒ–成功
}

#include <chrono>
//查询日志草最
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);
    
    std::string s;
    char s[32];

    // æ ¹æ®æ—¥å¿—级别设置前缀
    switch (level) {
        case DEBUG: s = "[debug]: "; break;
        case INFO: s = "[info]: "; break;
        case WARN: s = "[warn]: "; break;
        case ERROR: s = "[error]: "; break;
        default: s = "[info]: "; break;
        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(); // é”å®šäº’斥体
@@ -104,7 +181,7 @@

    // æ£€æŸ¥æ˜¯å¦éœ€è¦è½®æ¢æ—¥å¿—
    if (m_today != sys_tm->tm_mday || m_count % m_split_lines == 0) {
        // ... (日志轮换代码)
        rotate_logs(); // è°ƒç”¨æ—¥å¿—轮换函数
    }

    // å†™å…¥æ—¥å¿—内容
@@ -113,6 +190,7 @@
    vsnprintf(m_buf, m_log_buf_size, format, valist); // å†™å…¥æ—¥å¿—内容

    m_fp << s << m_buf << std::endl; // ä½¿ç”¨ofstream写入文件
    va_end(valist);

    m_mutex.unlock(); // è§£é”äº’斥体

@@ -133,18 +211,65 @@
void Log::flush_log_thread() {
    while (m_is_async) {
        // ä»Žé˜Ÿåˆ—中获取日志并写入
        string log;
        char* log;
        if (m_log_queue->pop(log)) {
            write_log(INFO, "%s", log.c_str());
            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 std::string &level) {
    if (level == "[debug]:") return DEBUG;
    if (level == "[info]:") return INFO;
    if (level == "[warn]:") return WARN;
    if (level == "[error]:") return ERROR;
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; // é‡ç½®æ—¥å¿—计数
}