simple_logger/log_manager.cpp at master · zhskyy/simple_logger · 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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
#include "log_manager.hpp"
#include <iostream>
#include <fstream>

std::map<std::string, std::shared_ptr<lg::logger_interface>> lg::log_manager::consumers_;
std::unique_ptr<lg::log_manager> lg::log_manager::instance_;
std::once_flag lg::log_manager::onceFlag_;

lg::logger_helper::logger_helper()
{
}

lg::logger_helper::~logger_helper()
{
	log_manager::instance()->flush(std::move(m_stream_));
}

// endl, flush, setw, setfill, etc.
lg::logger_helper& lg::logger_helper::operator<<(std::ostream&(*mip)(std::ostream&))
{
	mip(m_stream_);

	/*if (mip == static_cast<std::ostream&  (*)(std::ostream&)>(std::flush)
		|| mip == static_cast<std::ostream&  (*)(std::ostream&)>(std::endl))
		this->flush();*/

	return *this;
}

// setiosflags, resetiosflags
lg::logger_helper& lg::logger_helper::operator<<(std::ios_base&(*mip)(std::ios_base&))
{
	mip(m_stream_);
	return *this;
}


std::string lg::logger_interface::current_date_time()
{
	const auto now = time(nullptr);
	struct tm  tstruct;
	localtime_s(&tstruct, &now);
	std::stringstream ts;
	ts << "[" << 1900 + tstruct.tm_year << "-" << tstruct.tm_mon << "-" << tstruct.tm_mday << "T" << tstruct.tm_hour << ":" << tstruct.tm_min << ":" << tstruct.tm_sec << "]";
	return ts.str();
}

std::string lg::logger_interface::head(log_level l)
{
	std::stringstream ts;
	ts << "[";
	const auto o = levels_str.find(l);
	ts << (o == levels_str.end() ? "-----" : o->second);
	ts << "]";
	return ts.str();
}

void lg::console_logger::accept(const std::string& data, log_level level)
{
	if(level >= min_level())
	{
		std::cout << head(level) << current_date_time() << " " << data;
	}
}

lg::file_logger::file_logger(const std::string& file_name, log_level min_level)
	: logger_interface(min_level), filename_(file_name)
{
	std::ofstream fout;
	fout.open((filename_.empty() ? std::string("app.log") : filename_).c_str(),
		std::ios_base::out | std::ios_base::app);
	fout << std::endl << " ------------------------------------- " << std::endl;
}

void lg::file_logger::accept(const std::string& data, log_level level)
{
	if (level >= min_level())
	{
		std::ofstream fout;
		fout.open((filename_.empty() ? std::string("app.log") : filename_).c_str(),
			std::ios_base::out | std::ios_base::app);
		fout << head(level) << current_date_time() << " " << data;
	}
}

lg::log_manager* lg::log_manager::instance()
{
	std::call_once(onceFlag_, [] {
		instance_.reset(new log_manager);
	});
	return instance_.get();
}

lg::logger_helper lg::log_manager::helper()
{
	return logger_helper();
}

void lg::log_manager::add_consumer(const std::string& name, std::shared_ptr<logger_interface> logger)
{
	if (consumers_.end() == consumers_.find(name))
		consumers_[name] = logger;
}

lg::log_manager& lg::log_manager::operator()(const log_level e)
{
	m_log_level_ = e;
	return *this;
}

void lg::log_manager::flush(std::stringstream&& stream)
{
	for (const auto& consumer : consumers_)
	{
		consumer.second->accept(stream.str(), m_log_level_);
	}
}