Merge pull request #39 from Jin-W-FS/master

实现了syslog形式的log-rotate

demo/os/linux/easylogger/plugins/file/elog_file_cfg.h

@@ -35,4 +35,7 @@
 /* EasyLogger file log plugin's using file max size */
 #define ELOG_FILE_MAX_SIZE  (10 * 1024 * 1024)
 
+/* EasyLogger file log plugin's using max rotate file count */
+#define ELOG_FILE_MAX_ROTATE 10
+
 #endif /* _ELOG_FILE_CFG_H_ */

easylogger/plugins/file/elog_file.c

@@ -31,6 +31,8 @@
 #include <sys/stat.h>
 #include <fcntl.h>
 #include <stdbool.h>
+#include <stdlib.h>
+#include <string.h>
 
 #include <file/elog_file.h>
 #include <file/elog_file_cfg.h>
@@ -53,6 +55,7 @@ ElogErrCode elog_file_init(void)
 
     cfg.name = ELOG_FILE_NAME;
     cfg.max_size = ELOG_FILE_MAX_SIZE;
+    cfg.max_rotate = ELOG_FILE_MAX_ROTATE;
 
     elog_file_config(&cfg);
 
@@ -70,8 +73,13 @@ void elog_file_write(const char *log, size_t size)
     statbuf.st_size = 0;
     fstat(fd, &statbuf);
 
-    if (unlikely(statbuf.st_size > local_cfg.max_size))
+    if (unlikely(statbuf.st_size > local_cfg.max_size)) {
-        return;
+        if (local_cfg.max_rotate > 0) {
+            elog_file_rotate();
+        } else {
+            return;
+        }
+    }
 
     elog_file_port_lock();
 
@@ -103,6 +111,42 @@ void elog_file_config(ElogFileCfg *cfg)
 
     local_cfg.name = cfg->name;
     local_cfg.max_size = cfg->max_size;
+    local_cfg.max_rotate = cfg->max_rotate;
+
+    fp = fopen(local_cfg.name, "a+");
+    if (fp)
+        fd = fileno(fp);
+    else
+        fd = -1;
+
+    elog_file_port_unlock();
+}
+
+void elog_file_rotate(void)
+{
+    if (local_cfg.max_rotate <= 0) return;
+
+    elog_file_port_lock();
+
+    if (fp) {
+        fclose(fp);
+        fp = NULL;
+    }
+
+    /* mv xxx.log.n-1 => xxx.log.n, and xxx.log => xxx.log.0 */
+    size_t base = strlen(local_cfg.name);
+    char* oldpath = (char*)malloc(base + 10); // use c99 variable length array or alloca(3) instead?
+    char* newpath = (char*)malloc(base + 10);
+    memcpy(oldpath, local_cfg.name, base);
+    memcpy(newpath, local_cfg.name, base);
+    int n;
+    for (n = local_cfg.max_rotate - 1; n >= 0; --n) {
+        sprintf(oldpath + base, n ? ".%d" : "", n-1);
+        sprintf(newpath + base, ".%d", n);
+        rename(oldpath, newpath);
+    }
+    free(oldpath);
+    free(newpath);
 
     fp = fopen(local_cfg.name, "a+");
     if (fp)

easylogger/plugins/file/elog_file.h

@@ -49,12 +49,14 @@ extern "C" {
 typedef struct {
     char *name;              /* file name */
     size_t max_size;         /* file max size */
+    int max_rotate;          /* max rotate file count */
 } ElogFileCfg;
 
 /* elog_file.c */
 ElogErrCode elog_file_init(void);
 void elog_file_write(const char *log, size_t size);
 void elog_file_config(ElogFileCfg *cfg);
+void elog_file_rotate(void);
 void elog_file_deinit(void);
 
 /* elog_file_port.c */

easylogger/plugins/file/elog_file_cfg.h

@@ -35,4 +35,7 @@
 /* EasyLogger file log plugin's using file max size */
 #define ELOG_FILE_MAX_SIZE             /* @note you must define it for a value */
 
+/* EasyLogger file log plugin's using max rotate file count */
+#define ELOG_FILE_MAX_ROTATE           /* @note you must define it for a value */
+
 #endif /* _ELOG_FILE_CFG_H_ */