135 lines
3 KiB
C++
135 lines
3 KiB
C++
#pragma once
|
|
|
|
#include <io.h>
|
|
#include <fcntl.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "NonCopyable.h"
|
|
#include "StringUtil.h"
|
|
#include "WinUtil.h"
|
|
|
|
#ifndef LOGGER_DISABLE
|
|
class Logger : NonCopyable
|
|
{
|
|
public:
|
|
Logger() : m_systime(), m_console(INVALID_HANDLE_VALUE), m_file(INVALID_HANDLE_VALUE) {}
|
|
|
|
Logger::~Logger()
|
|
{
|
|
if (m_console)
|
|
FreeConsole();
|
|
|
|
if (m_file)
|
|
CloseHandle(m_file);
|
|
}
|
|
|
|
static Logger& Logger::Get()
|
|
{
|
|
static Logger instance;
|
|
return instance;
|
|
};
|
|
|
|
bool File(const std::string& filename)
|
|
{
|
|
std::string logpath = FullPathFromPath(filename);
|
|
m_file = CreateFileA(logpath.c_str(), GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
OutputDebugStringA(logpath.c_str());
|
|
|
|
if (m_file != INVALID_HANDLE_VALUE)
|
|
PrintStamp(false);
|
|
|
|
return m_file != INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
bool Console(const char* title)
|
|
{
|
|
AllocConsole();
|
|
|
|
m_console = GetStdHandle(STD_OUTPUT_HANDLE);
|
|
if (m_console != INVALID_HANDLE_VALUE)
|
|
{
|
|
ShowWindow(GetConsoleWindow(), SW_MAXIMIZE);
|
|
if (title) SetConsoleTitleA(title);
|
|
}
|
|
|
|
if (m_console != INVALID_HANDLE_VALUE)
|
|
PrintStamp(true);
|
|
|
|
return m_console != INVALID_HANDLE_VALUE;
|
|
}
|
|
|
|
void Print(const char* format, va_list args)
|
|
{
|
|
bool to_console = m_console != INVALID_HANDLE_VALUE;
|
|
bool to_file = m_file != INVALID_HANDLE_VALUE;
|
|
|
|
if ((to_console || to_file) && format)
|
|
{
|
|
int outsize = _vscprintf(format, args) + 1;
|
|
std::unique_ptr<char[]> buffer(new char[outsize]);
|
|
CharArrayFromFormatV(buffer.get(), outsize, format, args);
|
|
|
|
#ifdef LOGGER_DISABLE_TIME
|
|
std::string to_print(buffer.get(), outsize - 1);
|
|
#else
|
|
std::string to_print;
|
|
GetTime(&to_print);
|
|
to_print.append(buffer.get(), outsize - 1);
|
|
#endif
|
|
|
|
to_print.append("\r\n");
|
|
|
|
DWORD lenout = 0;
|
|
if (to_console) WriteConsoleA(m_console, to_print.c_str(), (DWORD)to_print.size(), &lenout, NULL);
|
|
if (to_file) WriteFile(m_file, to_print.c_str(), (DWORD)to_print.size(), &lenout, NULL);
|
|
}
|
|
}
|
|
|
|
private:
|
|
void PrintStamp(bool console)
|
|
{
|
|
static char stamp[] = "[TIME]\t\t[THREAD]\t[LOG]\r\n";
|
|
DWORD lenout = 0;
|
|
|
|
if (console) WriteConsoleA(m_console, stamp, _countof(stamp) - 1, &lenout, NULL);
|
|
else WriteFile(m_file, stamp, _countof(stamp) - 1, &lenout, NULL);
|
|
}
|
|
|
|
void GetTime(std::string* out)
|
|
{
|
|
GetLocalTime(&m_systime);
|
|
*out = StringFromFormat("%02u:%02u:%02u.%03u\t%08u\t", m_systime.wHour, m_systime.wMinute,
|
|
m_systime.wSecond, m_systime.wMilliseconds, GetCurrentThreadId());
|
|
|
|
}
|
|
|
|
SYSTEMTIME m_systime;
|
|
HANDLE m_console;
|
|
HANDLE m_file;
|
|
};
|
|
|
|
inline void LogFile(const std::string& logname)
|
|
{
|
|
Logger::Get().File(logname);
|
|
}
|
|
|
|
inline void LogConsole(const char* title = nullptr)
|
|
{
|
|
Logger::Get().Console(title);
|
|
}
|
|
|
|
inline void PrintLog(const char* format, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, format);
|
|
Logger::Get().Print(format, args);
|
|
va_end(args);
|
|
}
|
|
|
|
#else
|
|
#define LogFile(logname) (logname)
|
|
#define LogConsole(title, notice) (title)
|
|
#define PrintLog(format, ...) (format)
|
|
#endif |