feature:根据文件大小分割日志

src/CMakeLists.txt

@@ -3,6 +3,7 @@ project(logging)
 aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR} SRC)
 aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR}/handler SRC)
 aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR}/interceptor SRC)
+aux_source_directory(${CMAKE_CURRENT_SOURCE_DIR}/utils SRC)
 
 if (SHARED)
     add_library(${PROJECT_NAME} SHARED ${SRC})

src/handler/logging-handler-file.c

@@ -1,28 +1,73 @@
 #include "logging/logging-handler.h"
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
+
+#define Handler_FILE_EX_PRT(fp)                                                \
+    ((log_Handler_file_ex_t *)(fp + sizeof(log_Handler)))
+
+typedef struct log_Handler_file_ex_s {
+    unsigned int file_size;
+    unsigned int file_size_max;
+    unsigned int suffix;
+    char        *file_name;
+} log_Handler_file_ex_t;
+
+static unsigned int getFileSize(FILE *fp) {
+    fseek(fp, 0L, SEEK_END);
+    return ftell(fp);
+}
 
 static void __freeFileHandler(log_Handler *handler) {
     fclose(handler->stream);
+    free(Handler_FILE_EX_PRT(handler)->file_name);
     free(handler);
 }
 
+static void changeFile(log_Handler *handler) {
+    log_Handler_file_ex_t *handler_ex = Handler_FILE_EX_PRT(handler);
+    fclose(handler->stream);
+    char new_file_name[30];
+    sprintf(new_file_name, "%s_%d.log", handler_ex->file_name, ++handler_ex->suffix);
+    handler->stream = fopen(new_file_name, "at");
+    handler_ex->file_size = getFileSize(handler->stream);
+}
+
 static void outputFileHandler(log_Handler *handler, const char *message) {
     fputs(message, handler->stream);
+    log_Handler_file_ex_t *handler_ex = Handler_FILE_EX_PRT(handler);
+    handler_ex->file_size += strlen(message);
+    if (handler_ex->file_size > handler_ex->file_size_max)
+        changeFile(handler);
 }
 
-log_Handler *loggingFileHandler(const char *name) {
-    char new_file_name[100];
-    sprintf(new_file_name, "%s.log", name);
+log_Handler *loggingFileHandler(const char *name, unsigned int max_size) {
+    char         new_file_name[30];
+    int          suffix = 0;
+    unsigned int file_size;
+    FILE        *fp;
 
-    FILE *fp             = fopen(new_file_name, "at");
+    do {
+        sprintf(new_file_name, "%s_%d.log", name, suffix++);
+        fp        = fopen(new_file_name, "at");
+        file_size = getFileSize(fp);
+    } while (file_size > max_size);
 
-    log_Handler *handler = (log_Handler *)malloc(sizeof(log_Handler));
+    /// 分配log_Handler与记录文件大小的空间
+    log_Handler           *handler = (log_Handler *)malloc(sizeof(log_Handler) +
+                                                 sizeof(log_Handler_file_ex_t));
 
-    handler->stream      = fp;
-    handler->apply_color = false;
-    handler->_free       = __freeFileHandler;
-    handler->output      = outputFileHandler;
+    log_Handler_file_ex_t *handler_ex = Handler_FILE_EX_PRT(handler);
+    handler_ex->file_size_max        = max_size;
+    handler_ex->file_size             = file_size;
+    handler_ex->suffix                = suffix;
+    handler_ex->file_name             = malloc(strlen(new_file_name) + 1);
+    strcpy(handler_ex->file_name, name);
+
+    handler->stream                   = fp;
+    handler->apply_color              = false;
+    handler->_free                    = __freeFileHandler;
+    handler->output                   = outputFileHandler;
 
     return handler;
 }

src/logging.c

@@ -1,5 +1,6 @@
 #include "logging.h"
 #include "logging/logging-handler.h"
+#include "utils/logging-utils.h"
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -17,14 +18,10 @@
 
 Logger *G_LOGGER = NULL;
 
-static void getTimeStr(char *timeStr) {
-    time_t     t = time(NULL);
-    struct tm *p = localtime(&t);
-    char       _timeStr[20];
-    strftime(_timeStr, sizeof(_timeStr), "%Y-%m-%d %H:%M:%S", p);
-    strcpy(timeStr, _timeStr);
-}
-
+/**
+ * @brief 为日志添加一个handler
+ * @param handler 处理器对象
+ */
 static void addHandler(log_Handler *handler) {
     if (G_LOGGER == NULL) {
         return;
@@ -53,6 +50,14 @@ void addInterceptor(log_Interceptor *Interceptor) {
     G_LOGGER->interceptor = Interceptor;
 }
 
+/**
+ * @brief 内部日志打印处理核心函数
+ * @param level 日志等级
+ * @param color 应用的颜色
+ * @param message 日志内容
+ * @param ... 格式化参数列表
+ * @return
+ */
 static void
 _builtin_log(char *level, const char *color, const char *message, ...) {
     if (G_LOGGER == NULL) {
@@ -77,7 +82,7 @@ _builtin_log(char *level, const char *color, const char *message, ...) {
 
     if (handler->apply_color)
         sprintf(logStr,
-                "%s: %s %s%s%s %s\n",
+                "[%s]: %s %s%s%s %s\n",
                 G_LOGGER->name,
                 timeStr,
                 color,
@@ -86,7 +91,7 @@ _builtin_log(char *level, const char *color, const char *message, ...) {
                 message);
     else
         sprintf(
-            logStr, "%s: %s %s %s\n", G_LOGGER->name, timeStr, level, message);
+            logStr, "[%s]: %s %s %s\n", G_LOGGER->name, timeStr, level, message);
 
     handler->output(handler, logStr);
 }

src/utils/logging-utils.c

@@ -0,0 +1,15 @@
+#include "logging-utils.h"
+#include <time.h>
+#include <string.h>
+
+/**
+* @brief 获取当前时间字符串
+* @param timeStr 存储时间字符串缓冲区指针
+*/
+void getTimeStr(char *timeStr) {
+    time_t     t = time(NULL);
+    struct tm *p = localtime(&t);
+    char       _timeStr[20];
+    strftime(_timeStr, sizeof(_timeStr), "%Y-%m-%d %H:%M:%S", p);
+    strcpy(timeStr, _timeStr);
+}

src/utils/logging-utils.h

@@ -0,0 +1,5 @@
+#ifndef __LOGGING_UTILS_H__
+#define __LOGGING_UTILS_H__
+
+void getTimeStr(char *timeStr);
+#endif // __LOGGING_UTILS_H__