2 * Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
4 * This file is part of CasparCG (www.casparcg.com).
6 * CasparCG is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * CasparCG is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with CasparCG. If not, see <http://www.gnu.org/licenses/>.
19 * Author: Robert Nagy, ronag89@gmail.com
34 #include <boost/shared_ptr.hpp>
35 #include <boost/make_shared.hpp>
36 #include <boost/filesystem/convenience.hpp>
37 #include <boost/date_time/posix_time/posix_time.hpp>
38 #include <boost/algorithm/string.hpp>
40 #include <boost/log/core.hpp>
41 #include <boost/log/trivial.hpp>
42 #include <boost/log/expressions.hpp>
44 #include <boost/log/sinks/text_file_backend.hpp>
45 #include <boost/log/sinks/text_ostream_backend.hpp>
46 #include <boost/log/sinks/sync_frontend.hpp>
47 #include <boost/log/sinks/async_frontend.hpp>
48 #include <boost/log/core/record.hpp>
49 #include <boost/log/attributes/attribute_value.hpp>
50 #include <boost/log/attributes/current_thread_id.hpp>
51 #include <boost/log/utility/setup/common_attributes.hpp>
53 #include <boost/core/null_deleter.hpp>
54 #include <boost/lambda/lambda.hpp>
55 #include <boost/bind.hpp>
56 #include <boost/lexical_cast.hpp>
58 #include <tbb/atomic.h>
60 namespace caspar { namespace log {
62 using namespace boost;
64 template<typename Stream>
65 void append_timestamp(Stream& stream)
67 auto timestamp = boost::posix_time::microsec_clock::local_time();
68 auto date = timestamp.date();
69 auto time = timestamp.time_of_day();
70 auto milliseconds = time.fractional_seconds() / 1000; // microseconds to milliseconds
72 std::wstringstream buffer;
77 << std::setw(4) << date.year() << L"-" << std::setw(2) << date.month().as_number() << "-" << std::setw(2) << date.day().as_number()
79 << std::setw(2) << time.hours() << L":" << std::setw(2) << time.minutes() << L":" << std::setw(2) << time.seconds()
81 << std::setw(3) << milliseconds
84 stream << buffer.str();
89 tbb::atomic<int> column_width_;
91 column_writer(int initial_width = 0)
93 column_width_ = initial_width;
96 template<typename Stream, typename Val>
97 void write(Stream& out, const Val& value)
99 std::string to_string = boost::lexical_cast<std::string>(value);
100 int length = static_cast<int>(to_string.size());
103 while ((read_width = column_width_) < length && column_width_.compare_and_swap(length, read_width) != read_width);
104 read_width = column_width_;
106 out << L"[" << to_string << L"] ";
108 for (int n = 0; n < read_width - length; ++n)
113 template<typename Stream>
114 void my_formatter(bool print_all_characters, const boost::log::record_view& rec, Stream& strm)
116 static column_writer thread_id_column;
117 static column_writer severity_column(7);
118 namespace expr = boost::log::expressions;
120 append_timestamp(strm);
122 thread_id_column.write(strm, boost::log::extract<boost::log::attributes::current_thread_id::value_type>("ThreadID", rec).get().native_id());
123 severity_column.write(strm, boost::log::extract<boost::log::trivial::severity_level>("Severity", rec));
125 if (print_all_characters)
127 strm << rec[expr::message];
131 strm << replace_nonprintable_copy(rec[expr::message].get<std::wstring>(), L'?');
139 boost::log::add_common_attributes();
140 typedef boost::log::sinks::asynchronous_sink<boost::log::sinks::wtext_ostream_backend> stream_sink_type;
142 auto stream_backend = boost::make_shared<boost::log::sinks::wtext_ostream_backend>();
143 stream_backend->add_stream(boost::shared_ptr<std::wostream>(&std::wcout, boost::null_deleter()));
144 stream_backend->auto_flush(true);
146 auto stream_sink = boost::make_shared<stream_sink_type>(stream_backend);
148 bool print_all_characters = false;
149 stream_sink->set_formatter(boost::bind(&my_formatter<boost::log::wformatting_ostream>, print_all_characters, _1, _2));
151 boost::log::core::get()->add_sink(stream_sink);
156 void add_file_sink(const std::wstring& folder)
158 typedef boost::log::sinks::synchronous_sink<boost::log::sinks::text_file_backend> file_sink_type;
162 if (!boost::filesystem::is_directory(folder))
163 BOOST_THROW_EXCEPTION(directory_not_found());
165 auto file_sink = boost::make_shared<file_sink_type>(
166 boost::log::keywords::file_name = (folder + L"caspar_%Y-%m-%d.log"),
167 boost::log::keywords::time_based_rotation = boost::log::sinks::file::rotation_at_time_point(0, 0, 0),
168 boost::log::keywords::auto_flush = true,
169 boost::log::keywords::open_mode = std::ios::app
172 bool print_all_characters = true;
174 file_sink->set_formatter(boost::bind(&my_formatter<boost::log::formatting_ostream>, print_all_characters, _1, _2));
175 boost::log::core::get()->add_sink(file_sink);
179 std::wcerr << L"Failed to Setup File Logging Sink" << std::endl << std::endl;
183 void set_log_level(const std::wstring& lvl)
185 if (boost::iequals(lvl, L"trace"))
186 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::trace);
187 else if (boost::iequals(lvl, L"debug"))
188 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::debug);
189 else if (boost::iequals(lvl, L"info"))
190 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::info);
191 else if (boost::iequals(lvl, L"warning"))
192 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::warning);
193 else if (boost::iequals(lvl, L"error"))
194 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::error);
195 else if (boost::iequals(lvl, L"fatal"))
196 boost::log::core::get()->set_filter(boost::log::trivial::severity >= boost::log::trivial::fatal);