LiteLog/litelog.cpp at master · bubingy/LiteLog · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#include <ctime>
#include <map>
#include <iostream>
#include "litelog.h"

using namespace std;
using namespace litelog;

string
BaseLogger::getCurrentDatetime()
{
	time_t nowtime;
	nowtime = time(NULL);
	char tmp[64];
	strftime(tmp,
             sizeof(tmp),
             "%Y-%m-%d %H:%M:%S",
             localtime(&nowtime));
	return tmp;
}

static const map<Level, const char*>
LevelStr =
{
	{ Level::Debug,     "DEBUG"   },
	{ Level::Info,      "INFO"    },
	{ Level::Warn,      "WARNING" },
	{ Level::Error,     "ERROR"   },
	{ Level::Fatal,     "FATAL"   },
};

BaseLogger::LogStream BaseLogger::operator()(Level nLevel, string sTitle)
{
	return LogStream(*this, nLevel, sTitle);
}

void
BaseLogger::endline(Level nLevel,
                    string sTitle,
                    string oMessage)
{
    _lock.lock();
    output(getCurrentDatetime(), LevelStr.find(nLevel)->second, sTitle, oMessage);
    _lock.unlock();
}

void
ConsoleLogger::output(string localtime,
                      string level,
                      string title,
                      string message)
{
    cout << '[' << localtime << ']'
        << " - " << title
        << " - " << level
        << " - " << message << endl;
    cout.flush();
}

FileLogger::FileLogger(string filename) noexcept
    : BaseLogger()
{
    _file.open(filename,
        fstream::out | fstream::app | fstream::ate);
}

FileLogger::~FileLogger()
{
    _file.flush();
    _file.close();
}

void
FileLogger::output(string localtime,
                   string level,
                   string title,
                   string message)
{
    _file << '[' << localtime << ']'
        << " - " << title
        << " - " << level
        << " - " << message << endl;
    _file.flush();
}