来自mooon的最简单的日志类CSimpleLogger

/**

 * 单个头文件，可即时独立使用 ，只要定义了宏NOT_WITH_MOOON，即不依赖于mooon

 * 简单的写日志类，非线程安全，提供按大小滚动功能

 * 不追求功能，也不追求性能，只求简单，若要功能强、性能高，可以使用CLogger

 *

 * 使用方法：

 * 1) 构造一个 CSimpleLogger对象

 *     CSimpleLogger logger(".", "test.log", 1024*1024, 10);

 * 2) 调用print方法写日志

 *    logger. print("%s\n", "test");

 */

#ifndef MOOON_SYS_SIMPLE_LOGGER_H

#define MOOON_SYS_SIMPLE_LOGGER_H

// 只要定义了NOT_WITH_MOOON宏，

// 则本文件和mooon无任何关系，方便集成到自己的代码中

#define NOT_WITH_MOOON

#if !defined(NOT_WITH_MOOON)

#include <sys/config.h>

#endif // NOT_WITH_MOOON

#include <stdio.h>

#include <stdarg.h>

#include <time.h>

#include <sstream>

#if !defined(NOT_WITH_MOOON)

SYS_NAMESPACE_BEGIN

#endif // NOT_WITH_MOOON

/***

  * 万能型类型转换函数

  */

template <typename AnyType>

inline std::string any2string(AnyType any_value)

{

    std::stringstream result_stream;

    result_stream << any_value;

    return result_stream.str();

}

/***

  * 取当前时间，和date_util.h有重复，但为保持simple_logger.h的独立性，在所难免

  */

inline void get_current_datetime(char* datetime_buffer, size_t datetime_buffer_size)

{

    struct tm result;

    time_t now = time(NULL);

    localtime_r(&now, &result);

    snprintf(datetime_buffer, datetime_buffer_size

        ,"%04d-%02d-%02d %02d:%02d:%02d"

        ,result.tm_year+1900, result.tm_mon+1, result.tm_mday

        ,result.tm_hour, result.tm_min, result.tm_sec);

}

class CSimpleLogger

{

public:

    /***

      * 构造一个CSimpleLogger，并创建或打开日志文件

      * @log_dir 日志存放的目录，不需要以斜杠结尾，目录必须已经存在

      * @filename 日志的文件名，不包含目录部分，

      *           由log_dir和filename共同组成日志文件路径

      * @log_size 每个日志文件的大小，单位为字节数，如果小于1024，则会被强制为1024

      * @log_numer 日志滚动的个数

      * @record_size 单条日志的大小，超过会被截断，单位为字节数，如果小于1024，则会被强制为1024

      */

    CSimpleLogger(const std::string& log_dir

                 ,const std::string& filename

                 ,unsigned int log_size = 1024*1024*100

                 ,unsigned char log_numer = 10

                 ,unsigned short record_size = 8192);

    ~CSimpleLogger();

    /** 日志文件是否创建或打开成功 */

    bool is_ok() const;

    

    /** 输出日志，象printf一样使用，不自动加换行符 */

    void print(const char* format, ...);

     /** 刷新日志，因为使用FILE是带缓存的 */

    void flush();

private:

    void reset();    /** 复位状态值 */

    void roll_log(); /** 滚动日志 */

private:

    FILE* _fp;                    /** 当前正在写的日志文件描述符 */

    char* _log_buffer;            /** 存放日志的Buffer */

    int _bytes_writed;            /** 已经写入的字节数 */

    std::string _log_dir;         /** 日志存放目录 */

    std::string _filename;        /** 日志文件名，不包含目录部分 */

    unsigned int _log_size;       /** 单个日志文件的大小 */

    unsigned char _log_numer;     /** 日志滚动的个数 */

    unsigned short _record_size;  /** 单条日志的大小，单位为字节数 */

};

inline CSimpleLogger::CSimpleLogger(

                     const std::string& log_dir 

                    ,const std::string& filename

                    ,unsigned int log_size

                    ,unsigned char log_numer

                    ,unsigned short record_size)

 :_fp(NULL)

 ,_log_buffer(NULL)

 ,_bytes_writed(0)

 ,_log_dir(log_dir)

 ,_filename(filename)

 ,_log_size(log_size)

 ,_log_numer(log_numer)

 ,_record_size(record_size)

{

    std::string log_path = log_dir + std::string("/") + filename;

    _fp = fopen(log_path.c_str(), "a");

    

    if (_fp != NULL)

    {

        if (-1 == fseek(_fp, 0, SEEK_END))

        {

            // 失败，将不会写日志

            fclose(_fp);

            _fp = NULL;

        }

        else

        {

            // 取得已有大小

            _bytes_writed = ftell(_fp);

            // 不能太小气了          

            if (_log_size < 1024)

            {

                _log_size = 1024;

            }

            

            // 同样不能太小气

            if (_record_size < 1024)

            {

                _record_size = 1024;

            }

            

            _log_buffer = new char[_record_size];

        }

    }

}

inline CSimpleLogger::~CSimpleLogger()

{

    if (_fp != NULL)

        fclose(_fp);

        

    delete []_log_buffer;

}

inline bool CSimpleLogger::is_ok() const

{

    return _fp != NULL;

}

inline void CSimpleLogger::print(const char* format, ...)

{

    if (_fp != NULL)

    {

        va_list ap;

        va_start(ap, format);

        char datetime_buffer[sizeof("2012-12-21 00:00:00")]; // 刚好世界末日

        get_current_datetime(datetime_buffer, sizeof(datetime_buffer));

        vsnprintf(_log_buffer, _record_size, format, ap);

        int bytes_writed = fprintf(_fp, "[%s]%s", datetime_buffer, _log_buffer);

        if (bytes_writed > 0)

            _bytes_writed += bytes_writed;

        if (_bytes_writed > static_cast<int>(_log_size))

        {

            roll_log();

        }

        va_end(ap);

    }

}

inline void CSimpleLogger::roll_log()

{

    std::string new_path; // 滚动后的文件路径，包含目录和文件名

    std::string old_path; // 滚动前的文件路径，包含目录和文件名

    

    reset(); // 轮回，一切重新开始

    

    // 历史滚动

    for (int i=_log_numer-1; i>0; --i)

    {

        new_path = _log_dir + std::string("/") + _filename + std::string(".") + any2string(i);

        old_path = _log_dir + std::string("/") + _filename + std::string(".") + any2string(i-1);

        if (0 == access(old_path.c_str(), F_OK))

        {

            rename(old_path.c_str(), new_path.c_str());

        }

    }

    if (_log_numer > 0)

    {

        // 当前滚动

        new_path = _log_dir + std::string("/") + _filename + std::string(".1");

        old_path = _log_dir + std::string("/") + _filename;

        if (0 == access(old_path.c_str(), F_OK))

        {

            rename(old_path.c_str(), new_path.c_str());

        }

    }

    // 重新创建

    _fp = fopen(old_path.c_str(), "w+");

}

inline void CSimpleLogger::reset()

{

    _bytes_writed = 0;

    if (_fp != NULL)

    {

        fclose(_fp);

        _fp = NULL;

    }

}

inline void CSimpleLogger::flush()

{

    if (_fp != NULL)

        fflush(_fp);

}

/***

  * 测试代码

#include "simple_logger.h"

int main()

{

    CSimpleLogger logger(".", "test.log", 10240);

    for (int i=0; i<100000; ++i)

        logger.print("%d ==> abcdefghijklmnopqrestuvwxyz.\n", i);

        

    return 0;

}

*/

#if !defined(NOT_WITH_MOOON)

SYS_NAMESPACE_END

#endif // NOT_WITH_MOOON

#endif // MOOON_SYS_SIMPLE_LOGGER_H