receive packets from mrzcpd; log output supports stderr or file
This commit is contained in:
luwenpeng
308
src/log/log.cpp
308
src/log/log.cpp
@@ -1,3 +1,307 @@
|
||||
#include "log.h"
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <errno.h>
|
||||
#include <fcntl.h>
|
||||
#include <unistd.h>
|
||||
#include <stdarg.h>
|
||||
#include <time.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
// TODO
|
||||
#include "log.h"
|
||||
#include "toml.h"
|
||||
|
||||
enum log_output
|
||||
{
|
||||
LOG_OUTPUT_STDERR,
|
||||
LOG_OUTPUT_FILE,
|
||||
};
|
||||
|
||||
struct log_config
|
||||
{
|
||||
enum log_output output;
|
||||
enum log_level level;
|
||||
char log_file[256];
|
||||
};
|
||||
|
||||
struct log_context
|
||||
{
|
||||
struct log_config config;
|
||||
int log_fd;
|
||||
int log_file_reopen_day;
|
||||
};
|
||||
|
||||
struct log_context g_log_context;
|
||||
struct log_context *g_log_ctx = &g_log_context;
|
||||
|
||||
static unsigned char weekday_str[7][4] = {"Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"};
|
||||
static unsigned char month_str[12][4] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
|
||||
static unsigned char level_str[7][6] = {"TRACE", "DEBUG", "INFO", "WARN", "ERROR", "FATAL", "STATE"};
|
||||
|
||||
static inline void local_time(struct tm *local)
|
||||
{
|
||||
time_t t;
|
||||
time(&t);
|
||||
localtime_r(&t, local);
|
||||
}
|
||||
|
||||
static inline enum log_level check_level(const char *level)
|
||||
{
|
||||
if (level == NULL)
|
||||
{
|
||||
return LOG_NONE;
|
||||
}
|
||||
|
||||
if (strcasecmp(level, "TRACE") == 0)
|
||||
{
|
||||
return LOG_TRACE;
|
||||
}
|
||||
else if (strcasecmp(level, "DEBUG") == 0)
|
||||
{
|
||||
return LOG_DEBUG;
|
||||
}
|
||||
else if (strcasecmp(level, "INFO") == 0)
|
||||
{
|
||||
return LOG_INFO;
|
||||
}
|
||||
else if (strcasecmp(level, "WARN") == 0)
|
||||
{
|
||||
return LOG_WARN;
|
||||
}
|
||||
else if (strcasecmp(level, "ERROR") == 0)
|
||||
{
|
||||
return LOG_ERROR;
|
||||
}
|
||||
else if (strcasecmp(level, "FATAL") == 0)
|
||||
{
|
||||
return LOG_FATAL;
|
||||
}
|
||||
else if (strcasecmp(level, "STATE") == 0)
|
||||
{
|
||||
return LOG_STATE;
|
||||
}
|
||||
else
|
||||
{
|
||||
return LOG_NONE;
|
||||
}
|
||||
}
|
||||
|
||||
// return 0: success
|
||||
// return -1: failed
|
||||
static int parse_config(struct log_config *config, const char *cfg_file)
|
||||
{
|
||||
int ret = -1;
|
||||
FILE *fp = NULL;
|
||||
char errbuf[200];
|
||||
const char *ptr;
|
||||
toml_table_t *log_section = NULL;
|
||||
toml_table_t *conf_table = NULL;
|
||||
|
||||
fp = fopen(cfg_file, "r");
|
||||
if (fp == NULL)
|
||||
{
|
||||
fprintf(stderr, "open config file %s failed, %s\n", cfg_file, strerror(errno));
|
||||
goto error_out;
|
||||
}
|
||||
|
||||
conf_table = toml_parse_file(fp, errbuf, sizeof(errbuf));
|
||||
if (conf_table == NULL)
|
||||
{
|
||||
fprintf(stderr, "parse config file %s failed, %s\n", cfg_file, errbuf);
|
||||
goto error_out;
|
||||
}
|
||||
|
||||
log_section = toml_table_in(conf_table, "log");
|
||||
if (log_section == NULL)
|
||||
{
|
||||
fprintf(stderr, "config file %s missing log section\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
|
||||
// output
|
||||
ptr = toml_raw_in(log_section, "output");
|
||||
if (ptr == NULL)
|
||||
{
|
||||
fprintf(stderr, "config file %s missing log.output\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
if (strcasecmp(ptr, "stderr") == 0)
|
||||
{
|
||||
config->output = LOG_OUTPUT_STDERR;
|
||||
}
|
||||
else if (strcasecmp(ptr, "file") == 0)
|
||||
{
|
||||
config->output = LOG_OUTPUT_FILE;
|
||||
}
|
||||
else
|
||||
{
|
||||
fprintf(stderr, "config file %s invalid log.output\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
|
||||
// file
|
||||
if (config->output == LOG_OUTPUT_FILE)
|
||||
{
|
||||
ptr = toml_raw_in(log_section, "file");
|
||||
if (ptr == NULL)
|
||||
{
|
||||
fprintf(stderr, "config file %s missing log.file\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
strncpy(config->log_file, ptr, sizeof(config->log_file) - 1);
|
||||
}
|
||||
|
||||
// level
|
||||
ptr = toml_raw_in(log_section, "level");
|
||||
if (ptr == NULL)
|
||||
{
|
||||
fprintf(stderr, "config file %s missing log.level\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
config->level = check_level(ptr);
|
||||
if (config->level == LOG_NONE)
|
||||
{
|
||||
fprintf(stderr, "config file %s invalid log.level\n", cfg_file);
|
||||
goto error_out;
|
||||
}
|
||||
|
||||
ret = 0;
|
||||
|
||||
error_out:
|
||||
if (conf_table)
|
||||
{
|
||||
toml_free(conf_table);
|
||||
}
|
||||
|
||||
if (fp)
|
||||
{
|
||||
fclose(fp);
|
||||
}
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
// return 0: success
|
||||
// return -1: failed
|
||||
static int log_reopen()
|
||||
{
|
||||
int new_fd;
|
||||
int old_fd;
|
||||
struct tm local;
|
||||
char buff[512] = {0};
|
||||
local_time(&local);
|
||||
snprintf(buff, sizeof(buff), "%s.%d-%02d-%02d", g_log_ctx->config.log_file, local.tm_year + 1900, local.tm_mon + 1, local.tm_mday);
|
||||
|
||||
new_fd = open(buff, O_WRONLY | O_APPEND | O_CREAT, 0644);
|
||||
if (new_fd == -1)
|
||||
{
|
||||
fprintf(stderr, "open() \"%s\" failed, %s", buff, strerror(errno));
|
||||
return -1;
|
||||
}
|
||||
|
||||
g_log_ctx->log_file_reopen_day = local.tm_mday;
|
||||
old_fd = g_log_ctx->log_fd;
|
||||
g_log_ctx->log_fd = new_fd;
|
||||
|
||||
if (old_fd > 0)
|
||||
{
|
||||
close(old_fd);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
/******************************************************************************
|
||||
* Public API
|
||||
******************************************************************************/
|
||||
|
||||
// return 0: success
|
||||
// return -1: failed
|
||||
int log_init(const char *config_file)
|
||||
{
|
||||
memset(g_log_ctx, 0, sizeof(struct log_context));
|
||||
|
||||
if (parse_config(&g_log_ctx->config, config_file) != 0)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
|
||||
if (g_log_context.config.output == LOG_OUTPUT_FILE)
|
||||
{
|
||||
if (log_reopen() != 0)
|
||||
{
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void log_free()
|
||||
{
|
||||
if (g_log_ctx->config.output == LOG_OUTPUT_FILE)
|
||||
{
|
||||
if (g_log_ctx->log_fd > 0)
|
||||
{
|
||||
close(g_log_ctx->log_fd);
|
||||
g_log_ctx->log_fd = -1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void log_reload_level(const char *config_file)
|
||||
{
|
||||
struct log_config config;
|
||||
if (parse_config(&config, config_file) != 0)
|
||||
{
|
||||
return;
|
||||
}
|
||||
g_log_context.config.level = config.level;
|
||||
}
|
||||
|
||||
void log_print(enum log_level level, const char *module, const char *fmt, ...)
|
||||
{
|
||||
if (level < g_log_ctx->config.level)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
int nwrite;
|
||||
char buf[2048] = {0};
|
||||
char *p = buf;
|
||||
char *end = buf + sizeof(buf);
|
||||
va_list args;
|
||||
struct tm local;
|
||||
|
||||
local_time(&local);
|
||||
// add time
|
||||
p += snprintf(p, end - p, "%s %s %d %02d:%02d:%02d %d ",
|
||||
weekday_str[local.tm_wday], month_str[local.tm_mon], local.tm_mday, local.tm_hour, local.tm_min, local.tm_sec, local.tm_year + 1900);
|
||||
// add level
|
||||
p += snprintf(p, end - p, "%s ", level_str[level]);
|
||||
// add module
|
||||
p += snprintf(p, end - p, "(%s), ", module);
|
||||
// add content
|
||||
va_start(args, fmt);
|
||||
p += vsnprintf(p, end - p, fmt, args);
|
||||
va_end(args);
|
||||
// add end of line
|
||||
p += snprintf(p, end - p, "\n");
|
||||
|
||||
if (g_log_ctx->config.output == LOG_OUTPUT_STDERR)
|
||||
{
|
||||
fprintf(stderr, "%s", buf);
|
||||
return;
|
||||
}
|
||||
else
|
||||
{
|
||||
if (g_log_ctx->log_file_reopen_day != local.tm_mday)
|
||||
{
|
||||
log_reopen();
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
nwrite = write(g_log_ctx->log_fd, buf, p - buf);
|
||||
} while (nwrite == -1 && errno == EINTR);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user