测试

 

#include "stdafx.h"
#include "GB_Logger.h"
#include <imagehlp.h>
#include <time.h>
#include <stdarg.h>
#include <tchar.h>
 


GB_Logger* GB_Logger::logger = NULL; 


TCHAR g_LOGDIRECTORYPATH[MAX_STR_LEN] = {0};       //存放日志目录

//默认构造函数
GB_Logger::GB_Logger()
{
    //初始化
    //memset(g_LOGDIRECTORYPATH, 0, MAX_STR_LEN);
    memset(m_strCurLogName, 0, MAX_STR_LEN);
 
    //设置默认的写日志级别
    m_nLogLevel = EnumLogLevel::LogLevelNormal;
    GenerateLogName();
    //初始化临界区变量
    InitializeCriticalSection(&m_cs);
     
    //创建日志文件名
    
}

//析构函数
GB_Logger::~GB_Logger()
{
    //释放临界区
    DeleteCriticalSection(&m_cs);
    //关闭文件流
    if (m_hFile)
    {
        CloseHandle(m_hFile);
    }
}

int GB_Logger::TraceLogger(EnumLogType pLogType, const TCHAR * strInfo, ...)
{
    if (!strInfo)
        return 0;
    //TCHAR* str_buffer  = (TCHAR *)malloc(sizeof(TCHAR)* MAX_STR_LEN);
    TCHAR str_Buffer[MAX_STR_LEN] = { 0 }; 
    GetCurrentTimeToTChar(str_Buffer, MAX_STR_LEN);
    switch (pLogType)
    {
    case ErrInfo:
        if (m_nLogLevel >= EnumLogLevel::LogLevelStop)   //不写日志。
            return 2;
        lstrcat(str_Buffer, _T(" Error: "));
        break;
    case WarnInfo:
        if (m_nLogLevel >= EnumLogLevel::LogLevelNormal) //只写错误日志。
            return 2;
        lstrcat(str_Buffer, _T(" Warning:"));
        break;
    case trackingInfo:
        if (m_nLogLevel >= EnumLogLevel::LogLevelMid)   //当前只记录错误和警告信息。
            return 2;
        lstrcat(str_Buffer, _T(" trackingInfo:"));
        break;
    default:
        lstrcat(str_Buffer, _T(":"));
        return 2;
    }
    va_list arg_ptr = NULL;
    va_start(arg_ptr, strInfo);
    TCHAR p_Content[MAX_STR_LEN] = { 0 };
    _vsntprintf_s(p_Content,sizeof(TCHAR)*MAX_STR_LEN, strInfo, arg_ptr);
    lstrcat(str_Buffer, p_Content);
    va_end(arg_ptr);
    arg_ptr = NULL;
    WriteStrToLoggerFile(str_Buffer);
    return 1;
}

//将信息发送到指定的windows窗体
int GB_Logger::SendInfoToWindows(HWND hWnd,const TCHAR * strFormat, ...)
{
    //判断当前的写日志级别，若设置只写错误和警告信息则函数返回
    if (!strFormat)
        return 0;
    TCHAR prefix[MAX_STR_LEN] = { 0 };
    TCHAR str_Buffer[MAX_STR_LEN] = { 0 };
    GetCurrentTimeToTChar(str_Buffer,MAX_STR_LEN);
    lstrcat(str_Buffer, _T(": "));
    va_list arg_ptr = NULL;
    va_start(arg_ptr, strFormat);  //让arg_ptr指向参数列表中的第一参数地址。注意：函数参数是以数据结构，栈的形式存取，从右至左入栈。
    TCHAR p_Content[MAX_STR_LEN] = { 0 };
    _vsntprintf_s(p_Content, MAX_STR_LEN, strFormat, arg_ptr);
    lstrcat(str_Buffer, p_Content);
    va_end(arg_ptr);
    arg_ptr = NULL;
    ::SendMessage(hWnd, WM_USERDEFINEWINDOWSINFO, 0, (LPARAM)&str_Buffer); //同步
    return 1;
}

int GB_Logger::SetLogDirectory(const TCHAR * strDirectoryPath)
{
    __try{
        //进入临界区
        EnterCriticalSection(&m_cs);
        if (m_hFile)
        {
            if (CloseHandle(m_hFile) != 0)
                perror("close file fail!");
            else
                m_hFile = nullptr;
        }
        _tcscpy_s(g_LOGDIRECTORYPATH, MAX_STR_LEN, strDirectoryPath);
        if (0 != _tcslen(g_LOGDIRECTORYPATH))
        {
            lstrcat(g_LOGDIRECTORYPATH, _T("//"));
        }
        int hr = CreateDirectory(g_LOGDIRECTORYPATH, NULL);
        if (hr<0)
        {
            return hr;
        }
        GenerateLogName();
        return 1;
    }
    __finally{
        LeaveCriticalSection(&m_cs);
    }
}

//获取系统当前时间
int GB_Logger::GetCurrentTimeToTChar(OUT TCHAR* p_ResultBuffer, int p_SizeOfResultBuffer)
{
    time_t curTime;
    tm pTimeInfo;
    time(&curTime);
    localtime_s(&pTimeInfo, &curTime);
    _stprintf_s(p_ResultBuffer, p_SizeOfResultBuffer, _T("%02d:%02d:%02d"), pTimeInfo.tm_hour, pTimeInfo.tm_min, pTimeInfo.tm_sec);
    return 1;
}

//写文件操作
int GB_Logger::WriteStrToLoggerFile(const TCHAR * strInfo)
{
    if (!strInfo)
        return 0;
    try
    {
        //进入临界区
        EnterCriticalSection(&m_cs);
        //若文件流没有打开，则重新打开
        if (!m_hFile)
        {
            HANDLE hFile;
            TCHAR stBuffer[1024] = { 0 };
            lstrcat(stBuffer, g_LOGDIRECTORYPATH);
            lstrcat(stBuffer, m_strCurLogName);
            hFile = CreateFile(stBuffer,               //指向文件名的指针 
                GENERIC_WRITE | GENERIC_READ,          //访问模式（读/写） 写，读
                FILE_SHARE_READ,                       // 共享模式  不共享
                NULL,                                  //指向安全属性的指针 
                OPEN_ALWAYS,                           //如何让创建
                FILE_ATTRIBUTE_NORMAL,                 //文件属性
                NULL);                                 //用于复制文件句柄
            if (hFile == INVALID_HANDLE_VALUE)
            {
                AfxMessageBox(_T("创建日志文件失败"));
                return GetLastError();
            }
            this->m_hFile = hFile;
        }
        if (m_hFile)
        {
            //写日志信息到文件流
            TCHAR pLogbuff[MAX_PATH] = { 0 };
            _stprintf_s(pLogbuff, _T("%s\n"), strInfo);

            if (SetFilePointer(m_hFile, 0, NULL, FILE_END) == -1)
            {
                printf("SetFilePointer error\n");
                return 0;
            }

            DWORD ReturnCharNumber;
            WriteFile(m_hFile, pLogbuff, _tcslen(pLogbuff)*sizeof(TCHAR), &ReturnCharNumber, NULL);
            //判断当前日志文件大小，单位是字节。
            //LONGLONG file_size = 0;
            //file_size = GetFileSize(m_hFile, NULL);
             
            LARGE_INTEGER FileSize;
            GetFileSizeEx(m_hFile, &FileSize);
            if (FileSize.QuadPart >= MAX_LOGFILESIZE)
            {
                CloseHandle(m_hFile);
                TCHAR str_Buffer[1024] = { 0 };
                TCHAR* str_TimeBuffer = (TCHAR *)malloc(sizeof(TCHAR)* MAX_STR_LEN);
                GetCurrentTimeToTChar(str_TimeBuffer, MAX_STR_LEN);
                lstrcat(str_Buffer, m_strCurLogName);
                lstrcat(str_Buffer, str_TimeBuffer);
                memset(m_strCurLogName, 0, MAX_STR_LEN);
                lstrcat(m_strCurLogName, str_Buffer);
            }
        }

        //离开临界区
        LeaveCriticalSection(&m_cs);
        return 1;
    }
    //若发生异常，则先离开临界区，防止死锁
    catch (...)
    {
        LeaveCriticalSection(&m_cs);
        return 0;
    }
    return 1;
}

//创建日志文件的名称
void GB_Logger::GenerateLogName()
{
    time_t curTime;
    tm pTimeInfo;
    time(&curTime);
    localtime_s(&pTimeInfo, &curTime);
    TCHAR temp[1024] = { 0 };
    //日志的名称如：2013-01-01.log
    _stprintf_s(temp, _T("%04d-%02d-%02d-%02d.log"), pTimeInfo.tm_year + 1900, pTimeInfo.tm_mon + 1, pTimeInfo.tm_mday,pTimeInfo.tm_hour);
    if (0 != _tcscmp(m_strCurLogName, temp))  //如果文件名称不存在，创建该文件。
    {
        _tcscpy_s(m_strCurLogName, temp);
        if (m_hFile)
            CloseHandle(m_hFile);
        TCHAR temp[1024] = { 0 };
        lstrcat(temp, g_LOGDIRECTORYPATH);
        lstrcat(temp, m_strCurLogName);
        //以追加的方式打开文件流
        //_tfopen_s(&m_pFileStream, temp, _T("a+"));
    }

}

GB_Logger* GB_Logger::GetInstance()
{
    if (NULL == logger)
    { 
        GB_Logger* pLog = new GB_Logger();
        logger = pLog;
    }
    return logger;
}