完善日志库代码

fufu

2024-10-31 57a43626b15297d191382d40e63f7ba07b024f01

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

@@ -3,95 +3,100 @@

#include <stdio.h>
#include <iostream>
#include <string>
#include <stdarg.h>
#include <pthread.h>
#include <thread>
#include "block_queue.h"
#include "locker.h"
#include <fstream>
#include <sqlite3.h> // æ·»åŠ  SQLite æ•°æ®åº“的头文件

using namespace std;

// æ—¥å¿—级别枚举
enum LogLevel {
    DEBUG = 0,
    INFO,
    WARN,
    ERROR
};

// å­˜å‚¨è§£æžæ—¥å¿—文件的结构体
struct ParsedLog {
    char timestamp[64]; // æ—¶é—´æˆ³
    char device_id[64]; // è®¾å¤‡ID
    char level[16];     // æ—¥å¿—级别
    char content[256];  // æ—¥å¿—内容
    char source[64];    // æ¥æºæ¨¡å—
    char user_id[64];   // ç”¨æˆ·ID
};

class Log
{
public:
    // C++11以后,使用局部变量懒汉不用加锁
    // èŽ·å–单例实例
    static Log *get_instance()
    {
        static Log instance;
        return &instance;
    }

    static void *flush_log_thread(void *args)
    {
        Log::get_instance()->async_write_log();
    }
    // åˆ·æ–°æ—¥å¿—线程
    void flush_log_thread(); // ä¿®æ”¹ä¸ºéžé™æ€æˆå‘˜å‡½æ•°

    // å¯é€‰æ‹©çš„参数有日志文件、日志缓冲区大小、最大行数以及最长日志条队列
    bool init(const char *file_name, int close_log, int log_buf_size = 8192, int split_lines = 5000000, int max_queue_size = 0);
    // åˆå§‹åŒ–日志系统，参数包括文件名、是否关闭日志、缓冲区大小、最大行数和最大队列大小
    bool init(const char *file_name, int close_log, int log_buf_size, int split_lines, int max_queue_size);

    // å†™å…¥æ—¥å¿—
    void write_log(int level, const char *format, ...);
    
    // åˆ·æ–°æ—¥å¿—文件
    void flush(void);

    // æŽ¥æ”¶åŽŸå§‹æ—¥å¿—
    void receiveLog(const char* raw_log);

    // å°†æ—¥å¿—写入数据库
    void writeLogToDatabase(const ParsedLog& log);

private:
    Log();
    virtual ~Log();
    Log(); // æž„造函数
    virtual ~Log(); // æžæž„函数

    void async_write_log(); // å¼‚步写入日志
    bool check_log_size(); // æ£€æŸ¥æ—¥å¿—文件大小
    void rotate_logs(); // è½®æ¢æ—¥å¿—文件
    
    void *async_write_log()
    {
        string single_log;
        // ä»Žé˜»å¡žé˜Ÿåˆ—中取出一个日志string，写入文件
        while (m_log_queue->pop(single_log))
        {
            m_mutex.lock();
            if (check_log_size()) // æ£€æŸ¥æ–‡ä»¶å¤§å°
            {
                rotate_logs(); // è½®æ¢æ—¥å¿—
            }
            fputs(single_log.c_str(), m_fp);
            m_mutex.unlock();
        }
    }
    ParsedLog parseLog(const char* log); // è§£æžæ—¥å¿—
    bool init(const char *file_name, int close_log, int log_buf_size, int split_lines, int max_queue_size);
    void queryLogs(const char *device_id); // æŸ¥è¯¢æ—¥å¿—

    // æ£€æŸ¥æ—¥å¿—文件大小
    bool 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 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; // é‡ç½®æ—¥å¿—计数
    }
    const char* log_level_to_string(int level); // å°†æ—¥å¿—级别转换为字符串
    int levelToInt(const char *level);

private:
    char dir_name[128]; // è·¯å¾„名
    char log_name[128]; // log文件名
    char dir_name[128]; // æ—¥å¿—文件目录
    char log_name[128]; // æ—¥å¿—文件名
    int m_split_lines;  // æ—¥å¿—最大行数
    int m_log_buf_size; // æ—¥å¿—缓冲区大小
    long long m_count;  // æ—¥å¿—行数记录
    int m_today;        // å› ä¸ºæŒ‰å¤©åˆ†ç±»,记录当前时间是那一天
    FILE *m_fp;         // æ‰“å¼€log的文件指针
    char *m_buf;
    block_queue<string> *m_log_queue; // é˜»å¡žé˜Ÿåˆ—
    bool m_is_async;                  // æ˜¯å¦åŒæ­¥æ ‡å¿—位
    locker m_mutex;
    int m_close_log; // å…³é—­æ—¥å¿—
    long max_size = 10 * 1024 * 1024; // 10MB
    int m_today;        // è®°å½•å½“前时间
    std::ofstream m_fp; // å°† FILE * m_fp æ›´æ”¹ä¸º std::ofstream m_fp
    char *m_buf;        // æ—¥å¿—缓冲区
    block_queue<char*> *m_log_queue; // é˜»å¡žé˜Ÿåˆ—
    bool m_is_async;                  // æ˜¯å¦å¼‚步标志位
    locker m_mutex;                   // äº’斥锁
    int m_close_log; // å…³é—­æ—¥å¿—的标志
    long max_size = 10 * 1024 * 1024; // å•ä¸ªæ—¥å¿—文件的大小 10MB

    // æ•°æ®åº“相关
    sqlite3 *db; // SQLite æ•°æ®åº“指针
    void openDatabase(const char *db_name); // æ‰“开数据库
    void closeDatabase(); // å…³é—­æ•°æ®åº“
};

#define LOG_DEBUG(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(0, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_INFO(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(1, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_WARN(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(2, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_ERROR(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(3, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_DEBUG(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(DEBUG, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_INFO(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(INFO, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_WARN(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(WARN, format, ##__VA_ARGS__); Log::get_instance()->flush();}
#define LOG_ERROR(format, ...) if(0 == m_close_log) {Log::get_instance()->write_log(ERROR, format, ##__VA_ARGS__); Log::get_instance()->flush();}

#endif