log mylog

/*
 * 支持linux，windows平台，不支持UNICODE
 * 用例：
#include "mylog.h"
#ifdef _WIN32
int tmain(int argc, _TCHAR* argv[])
#else
int main(int argc, char* argv[])
#endif
{
    log_init("1.txt");
    //log_init(NULL);
    L_ENTER();
    L_FATAL(_TEXT("fatal %d\n"), 1);
    L_ERROR(_TEXT("fatal %d\n"), 1);
    L_WARNING(_TEXT("fatal %d\n"), 1);
    L_INFO(_TEXT("fatal %d\n"), 1);
    L_DEBUG(_TEXT("fatal %d\n"), 1);
    L_TRACE(_TEXT("fatal %d\n"), 1);
    L_LEAVE();

    return 0;
}
 */
#ifndef __MYLOG_H__
#define __MYLOG_H__

// 日志格式
//1812::INFO::2014-04-15 14:48:54::log_version::0.1.0.0

#ifdef __cplusplus
extern "C" {
#endif


#include <stdio.h>
#include <stdint.h>
#include <time.h>
#include <wchar.h>

#ifdef _WIN32
#include <windows.h>
#else
#include <sys/types.h>
#include <unistd.h>
#endif


enum LOG_LEVEL_ {
    LOG_LEVEL__TRACE = 0,
    LOG_LEVEL__DEBUG,
    LOG_LEVEL__INFO,
    LOG_LEVEL__WARNING,
    LOG_LEVEL__ERROR,
    LOG_LEVEL__FATAL
};

/*
 *    配置
 * MY_DEFAULT_LOG_FILE: 默认日志文件名
 * MY_DEFAULT_LOG_LEVEL: 默认输出的日志级别
 */
#define MY_DEFAULT_LOG_FILE    "log.txt"
#define MY_DEFAULT_LOG_LEVEL LOG_LEVEL__DEBUG


/**************************************************************************/
/**************************************************************************/


#ifndef TRUE
#define TRUE 1
#endif

#ifndef FALSE
#define FALSE 0
#endif

#ifndef _WIN32
#define _TEXT(x)    x
#define _tfopen     fopen
#define _ftprintf   fprintf
#define GetCurrentThreadId getpid
#define TCHAR       char
#endif

#undef LOG_FILE_NAME
#define LOG_FILE_NAME    _TEXT(MY_DEFAULT_LOG_FILE)

static uint32_t log_level = MY_DEFAULT_LOG_LEVEL;
static uint32_t log_enable = FALSE;
static FILE *fp = NULL;

#ifdef _WIN32
# pragma warning(push)
/* C4996: _CRT_SECURE_NO_WARNINGS. */
# pragma warning(disable: 4996)
#endif

uint32_t log_init(const TCHAR* filename)
{
    if ( filename == NULL ) {
        filename = LOG_FILE_NAME;
    }

    if ( fp != NULL ) {
        fclose(fp);
    }

    fp = _tfopen(filename, _TEXT("a+"));
    if ( fp == NULL ) {
        _ftprintf(stderr, _TEXT("open %s failed, log disable\n"), filename);
        return FALSE;
    }

    log_enable = TRUE;
    return TRUE;
}

#ifdef _WIN32
# pragma warning(pop)
#endif

void current_time(TCHAR *out, int out_len)
{
    time_t timer;
    timer = time(NULL);
#ifdef _WIN32
    struct tm today;
    localtime_s(&today, &timer);
    strftime(out, out_len, "%Y-%m-%d %H:%M:%S", &today);
#else
    struct tm *today;
    today = localtime(&timer);
    strftime(out, out_len, "%Y-%m-%d %H:%M:%S", today);
#endif
}

#define MYLOG(level, fmt, ...)    do                \
{                                                \
    if ( log_enable && level >= log_level ) {    \
        static TCHAR level_as_string[][8] = {_TEXT("TRACE"), _TEXT("DEBUG"), _TEXT("INFO "), \
                                            _TEXT("WARN "), _TEXT("ERROR"), _TEXT("FATAL")}; \
        TCHAR datetime_str[20];                    \
        current_time(datetime_str, sizeof(datetime_str)); \
        _ftprintf(fp, _TEXT("%d::%s::%s::%s::")fmt,                                                        \
                        GetCurrentThreadId(), level_as_string[level], datetime_str, __FUNCTION__, ##__VA_ARGS__); \
        fflush(fp);                                \
    }                                            \
} while (0)



#define L_FATAL(fmt, ...) MYLOG(LOG_LEVEL__FATAL, fmt, ## __VA_ARGS__)
#define L_ERROR(fmt, ...) MYLOG(LOG_LEVEL__ERROR, fmt, ## __VA_ARGS__)
#define L_WARNING(fmt, ...) MYLOG(LOG_LEVEL__WARNING, fmt, ## __VA_ARGS__)
#define L_INFO(fmt, ...) MYLOG(LOG_LEVEL__INFO, fmt, ## __VA_ARGS__)
#define L_DEBUG(fmt, ...) MYLOG(LOG_LEVEL__DEBUG, fmt, ## __VA_ARGS__)
#define L_TRACE(fmt, ...) MYLOG(LOG_LEVEL__TRACE, fmt, ## __VA_ARGS__)
#define L_ENTER()        L_DEBUG("Enter.\n")
#define L_LEAVE()        L_DEBUG("Leave.\n")

#ifdef __cplusplus
}
#endif

#endif