/**

 *  单个头文件，可即时独立使用 ，只要定义了宏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
#endif // NOT_WITH_MOOON
 
#include
#include
#include
#include
#if !defined(NOT_WITH_MOOON)
SYS_NAMESPACE_BEGIN
#endif // NOT_WITH_MOOON
 
/***
  * 万能型类型转换函数
  */
template
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(_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