339 lines
10 KiB
C
339 lines
10 KiB
C
#ifndef LOG_H
|
||
#define LOG_H
|
||
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
#include <time.h>
|
||
#include <wchar.h>
|
||
#include <locale.h>
|
||
|
||
// ANSI颜色代码
|
||
#define COLOR_RESET "\033[0m"
|
||
#define COLOR_RED "\033[31m"
|
||
#define COLOR_GREEN "\033[32m"
|
||
#define COLOR_YELLOW "\033[33m"
|
||
#define COLOR_BLUE "\033[34m"
|
||
#define COLOR_MAGENTA "\033[35m"
|
||
#define COLOR_CYAN "\033[36m"
|
||
#define COLOR_WHITE "\033[37m"
|
||
#define COLOR_GRAY "\033[90m"
|
||
|
||
// 辅助函数声明
|
||
const char* get_color_code(const char* color_name);
|
||
void get_time_string(char* buffer, size_t size);
|
||
|
||
// 内部打印函数
|
||
static inline void _print_int(int x) { printf("%d", x); }
|
||
static inline void _print_double(double x) { printf("%.2f", x); }
|
||
static inline void _print_str(const char* x) { printf("%s", x); }
|
||
static inline void _print_char(int x) { printf("%c", x); }
|
||
static inline void _print_wstr(const wchar_t* x) {
|
||
static int locale_set = 0;
|
||
if (!locale_set) {
|
||
setlocale(LC_ALL, "");
|
||
locale_set = 1;
|
||
}
|
||
wprintf(L"%ls", x);
|
||
}
|
||
|
||
// 类型自动打印（使用C23的_Generic）
|
||
#define PRINT_VALUE(x) _Generic((x), \
|
||
int: _print_int, \
|
||
long: _print_int, \
|
||
double: _print_double, \
|
||
float: _print_double, \
|
||
char: _print_char, \
|
||
char*: _print_str, \
|
||
const char*: _print_str, \
|
||
wchar_t*: _print_wstr, \
|
||
const wchar_t*: _print_wstr \
|
||
)(x)
|
||
|
||
// log函数重载 - 单参数
|
||
static inline void _log_int(int x) {
|
||
printf("%d\n", x);
|
||
}
|
||
|
||
static inline void _log_double(double x) {
|
||
printf("%.2f\n", x);
|
||
}
|
||
|
||
static inline void _log_str(const char* x) {
|
||
printf("%s\n", x);
|
||
}
|
||
|
||
// log函数重载 - 格式化字符串 + 一个参数
|
||
static inline void _log_fmt_int(const char* fmt, int b) {
|
||
const char* p = fmt;
|
||
int done = 0;
|
||
while (*p) {
|
||
if (*p == '{' && *(p+1) == '}' && !done) {
|
||
printf("%d", b);
|
||
done = 1;
|
||
p += 2;
|
||
} else {
|
||
putchar(*p);
|
||
p++;
|
||
}
|
||
}
|
||
printf("\n");
|
||
}
|
||
|
||
// log函数重载 - 格式化字符串 + 两个参数
|
||
static inline void _log_fmt_2(const char* fmt, int b, double c) {
|
||
const char* p = fmt;
|
||
int done = 0;
|
||
while (*p) {
|
||
if (*p == '{' && *(p+1) == '}') {
|
||
if (done == 0) {
|
||
printf("%d", b);
|
||
done = 1;
|
||
} else if (done == 1) {
|
||
printf("%.2f", c);
|
||
done = 2;
|
||
}
|
||
p += 2;
|
||
} else {
|
||
putchar(*p);
|
||
p++;
|
||
}
|
||
}
|
||
printf("\n");
|
||
}
|
||
|
||
// 格式化字符串 + 三个参数
|
||
#define _log_fmt_3(fmt, a, b, c) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 四个参数
|
||
#define _log_fmt_4(fmt, a, b, c, d) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 五个参数
|
||
#define _log_fmt_5(fmt, a, b, c, d, e) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 六个参数
|
||
#define _log_fmt_6(fmt, a, b, c, d, e, f) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
else if (_done == 5) { PRINT_VALUE(f); _done = 6; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 七个参数
|
||
#define _log_fmt_7(fmt, a, b, c, d, e, f, g) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
else if (_done == 5) { PRINT_VALUE(f); _done = 6; } \
|
||
else if (_done == 6) { PRINT_VALUE(g); _done = 7; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 八个参数
|
||
#define _log_fmt_8(fmt, a, b, c, d, e, f, g, h) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
else if (_done == 5) { PRINT_VALUE(f); _done = 6; } \
|
||
else if (_done == 6) { PRINT_VALUE(g); _done = 7; } \
|
||
else if (_done == 7) { PRINT_VALUE(h); _done = 8; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 九个参数
|
||
#define _log_fmt_9(fmt, a, b, c, d, e, f, g, h, i) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
else if (_done == 5) { PRINT_VALUE(f); _done = 6; } \
|
||
else if (_done == 6) { PRINT_VALUE(g); _done = 7; } \
|
||
else if (_done == 7) { PRINT_VALUE(h); _done = 8; } \
|
||
else if (_done == 8) { PRINT_VALUE(i); _done = 9; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 格式化字符串 + 十个参数
|
||
#define _log_fmt_10(fmt, a, b, c, d, e, f, g, h, i, j) do { \
|
||
const char* _p = (fmt); \
|
||
int _done = 0; \
|
||
while (*_p) { \
|
||
if (*_p == '{' && *(_p+1) == '}') { \
|
||
if (_done == 0) { PRINT_VALUE(a); _done = 1; } \
|
||
else if (_done == 1) { PRINT_VALUE(b); _done = 2; } \
|
||
else if (_done == 2) { PRINT_VALUE(c); _done = 3; } \
|
||
else if (_done == 3) { PRINT_VALUE(d); _done = 4; } \
|
||
else if (_done == 4) { PRINT_VALUE(e); _done = 5; } \
|
||
else if (_done == 5) { PRINT_VALUE(f); _done = 6; } \
|
||
else if (_done == 6) { PRINT_VALUE(g); _done = 7; } \
|
||
else if (_done == 7) { PRINT_VALUE(h); _done = 8; } \
|
||
else if (_done == 8) { PRINT_VALUE(i); _done = 9; } \
|
||
else if (_done == 9) { PRINT_VALUE(j); _done = 10; } \
|
||
_p += 2; \
|
||
} else { \
|
||
putchar(*_p); \
|
||
_p++; \
|
||
} \
|
||
} \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
// 主log宏 - 支持最多11个参数（格式化字符串 + 10个变量）
|
||
#define _GET_11TH_ARG(_1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, N, ...) N
|
||
|
||
#define log(...) _GET_11TH_ARG(__VA_ARGS__, \
|
||
_log_fmt_10, _log_fmt_9, _log_fmt_8, _log_fmt_7, _log_fmt_6, \
|
||
_log_fmt_5, _log_fmt_4, _log_fmt_3, _log_2, _log_1, _log_1)(__VA_ARGS__)
|
||
|
||
#define _log_1(x) _Generic((x), \
|
||
int: _log_int, \
|
||
double: _log_double, \
|
||
char*: _log_str, \
|
||
const char*: _log_str \
|
||
)(x)
|
||
|
||
#define _log_2(a, b) _Generic((b), \
|
||
int: _log_fmt_int \
|
||
)(a, b)
|
||
|
||
#define _log_3(a, b, c) _Generic((b), \
|
||
int: _log_fmt_2 \
|
||
)(a, b, c)
|
||
|
||
// 颜色log
|
||
#define clog(color, msg) do { \
|
||
printf("%s", get_color_code(color)); \
|
||
PRINT_VALUE(msg); \
|
||
printf("%s\n", COLOR_RESET); \
|
||
} while(0)
|
||
|
||
// 日志等级宏
|
||
#define logerr(msg) do { \
|
||
char time_str[64]; \
|
||
get_time_string(time_str, sizeof(time_str)); \
|
||
printf("%s[%s ERROR]%s ", COLOR_RED, time_str, COLOR_RESET); \
|
||
PRINT_VALUE(msg); \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
#define logok(msg) do { \
|
||
char time_str[64]; \
|
||
get_time_string(time_str, sizeof(time_str)); \
|
||
printf("%s[%s OK]%s ", COLOR_GREEN, time_str, COLOR_RESET); \
|
||
PRINT_VALUE(msg); \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
#define loginfo(msg) do { \
|
||
char time_str[64]; \
|
||
get_time_string(time_str, sizeof(time_str)); \
|
||
printf("%s[%s INFO]%s ", COLOR_CYAN, time_str, COLOR_RESET); \
|
||
PRINT_VALUE(msg); \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
#define logwarn(msg) do { \
|
||
char time_str[64]; \
|
||
get_time_string(time_str, sizeof(time_str)); \
|
||
printf("%s[%s WARN]%s ", COLOR_YELLOW, time_str, COLOR_RESET); \
|
||
PRINT_VALUE(msg); \
|
||
printf("\n"); \
|
||
} while(0)
|
||
|
||
#endif
|