git.sesse.net Git - casparcg/blob - common/log/log.cpp

   1 /*\r
   2 * copyright (c) 2010 Sveriges Television AB <info@casparcg.com>\r
   3 *\r
   4 *  This file is part of CasparCG.\r
   5 *\r
   6 *    CasparCG is free software: you can redistribute it and/or modify\r
   7 *    it under the terms of the GNU General Public License as published by\r
   8 *    the Free Software Foundation, either version 3 of the License, or\r
   9 *    (at your option) any later version.\r
  10 *\r
  11 *    CasparCG is distributed in the hope that it will be useful,\r
  12 *    but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  13 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  14 *    GNU General Public License for more details.\r
  15 \r
  16 *    You should have received a copy of the GNU General Public License\r
  17 *    along with CasparCG.  If not, see <http://www.gnu.org/licenses/>.\r
  18 *\r
  19 */\r
  20 #include "../stdafx.h"\r
  21 \r
  22 #if defined(_MSC_VER)\r
  23 #pragma warning (disable : 4100) // 'identifier' : unreferenced formal parameter\r
  24 #pragma warning (disable : 4512) // 'class' : assignment operator could not be generated\r
  25 #endif\r
  26 \r
  27 #include "log.h"\r
  28 \r
  29 #include "../exception/exceptions.h"\r
  30 \r
  31 #include <ios>\r
  32 #include <string>\r
  33 #include <ostream>\r
  34 \r
  35 #include <boost/shared_ptr.hpp>\r
  36 #include <boost/make_shared.hpp>\r
  37 #include <boost/filesystem/convenience.hpp>\r
  38 #include <boost/date_time/posix_time/posix_time.hpp>\r
  39 \r
  40 #include <boost/log/core/core.hpp>\r
  41 \r
  42 #include <boost/log/formatters/stream.hpp>\r
  43 #include <boost/log/formatters/attr.hpp>\r
  44 #include <boost/log/formatters/date_time.hpp>\r
  45 #include <boost/log/formatters/message.hpp>\r
  46 \r
  47 #include <boost/log/filters/attr.hpp>\r
  48 \r
  49 #include <boost/log/sinks/text_file_backend.hpp>\r
  50 \r
  51 #include <boost/log/detail/universal_path.hpp>\r
  52 \r
  53 #include <boost/log/sinks/text_file_backend.hpp>\r
  54 #include <boost/log/sinks/text_ostream_backend.hpp>\r
  55 #include <boost/log/sinks/sync_frontend.hpp>\r
  56 #include <boost/log/sinks/async_frontend.hpp>\r
  57 #include <boost/log/core/record.hpp>\r
  58 #include <boost/log/utility/attribute_value_extractor.hpp>\r
  59 \r
  60 #include <boost/log/utility/init/common_attributes.hpp>\r
  61 #include <boost/log/utility/empty_deleter.hpp>\r
  62 #include <boost/lambda/lambda.hpp>\r
  63 \r
  64 namespace caspar{ namespace log{\r
  65 \r
  66 using namespace boost;\r
  67 \r
  68 \r
  69 class system_uptime : public boost::log::attribute\r
  70 {\r
  71     typedef boost::log::attributes::basic_attribute_value<unsigned int> attribute_value_type;\r
  72 \r
  73 public:\r
  74     boost::shared_ptr<boost::log::attribute_value> get_value()\r
  75     {\r
  76         unsigned int up;\r
  77         up = GetTickCount() / 1000;\r
  78         return boost::shared_ptr<boost::log::attribute_value>(new attribute_value_type(up));\r
  79     }\r
  80 };\r
  81 \r
  82 void my_formatter(std::wostream& strm, boost::log::basic_record<wchar_t> const& rec)\r
  83 {\r
  84     namespace lambda = boost::lambda;\r
  85         \r
  86         #pragma warning(disable : 4996)\r
  87         time_t rawtime;\r
  88         struct tm* timeinfo;\r
  89         time(&rawtime );\r
  90         timeinfo = localtime ( &rawtime );\r
  91         char buffer [80];\r
  92         strftime (buffer,80, "%c", timeinfo);\r
  93         strm << L"[" << buffer << L"] ";\r
  94                 \r
  95     boost::log::attributes::current_thread_id::held_type thread_id;\r
  96     if(boost::log::extract<boost::log::attributes::current_thread_id::held_type>(L"ThreadID", rec.attribute_values(), lambda::var(thread_id) = lambda::_1))\r
  97         strm << L"[" << thread_id << L"] ";\r
  98 \r
  99     severity_level severity;\r
 100     if(boost::log::extract<severity_level>(boost::log::sources::aux::severity_attribute_name<wchar_t>::get(), rec.attribute_values(), lambda::var(severity) = lambda::_1))\r
 101         strm << L"[" << severity << L"] ";\r
 102 \r
 103     strm << rec.message();\r
 104 }\r
 105 \r
 106 namespace internal{\r
 107         \r
 108 void init()\r
 109 {       \r
 110         boost::log::wcore::get()->add_global_attribute(L"ASD", boost::make_shared<system_uptime>());\r
 111 \r
 112         boost::log::add_common_attributes<wchar_t>();\r
 113         typedef boost::log::aux::add_common_attributes_constants<wchar_t> traits_t;\r
 114 \r
 115         typedef boost::log::sinks::synchronous_sink<boost::log::sinks::wtext_file_backend> file_sink_type;\r
 116 \r
 117         typedef boost::log::sinks::asynchronous_sink<boost::log::sinks::wtext_ostream_backend> stream_sink_type;\r
 118 \r
 119         auto stream_backend = boost::make_shared<boost::log::sinks::wtext_ostream_backend>();\r
 120         stream_backend->add_stream(boost::shared_ptr<std::wostream>(&std::wcout, boost::log::empty_deleter()));\r
 121         stream_backend->auto_flush(true);\r
 122 \r
 123         auto stream_sink = boost::make_shared<stream_sink_type>(stream_backend);\r
 124         \r
 125         stream_sink->locked_backend()->set_formatter(&my_formatter);\r
 126 \r
 127         boost::log::wcore::get()->add_sink(stream_sink);\r
 128 }\r
 129 }\r
 130 \r
 131 void add_file_sink(const std::wstring& folder)\r
 132 {       \r
 133         boost::log::add_common_attributes<wchar_t>();\r
 134         typedef boost::log::aux::add_common_attributes_constants<wchar_t> traits_t;\r
 135 \r
 136         typedef boost::log::sinks::synchronous_sink<boost::log::sinks::wtext_file_backend> file_sink_type;\r
 137 \r
 138         try\r
 139         {\r
 140                 if(!boost::filesystem::is_directory(folder))\r
 141                         BOOST_THROW_EXCEPTION(directory_not_found());\r
 142 \r
 143                 auto file_sink = boost::make_shared<file_sink_type>(\r
 144                         boost::log::keywords::file_name = (folder + L"caspar_%Y-%m-%d_%H-%M-%S.%N.log"),\r
 145                         boost::log::keywords::time_based_rotation = boost::log::sinks::file::rotation_at_time_point(0, 0, 0),\r
 146                         boost::log::keywords::auto_flush = true\r
 147                 );\r
 148 \r
 149                 //file_sink->locked_backend()->set_formatter(\r
 150                 //      boost::log::formatters::wstream\r
 151                 //              //<< L"[" << boost::log::formatters::date_time(L"TimeStamp") << L"] "\r
 152                 //              << L"[" << boost::log::formatters::attr<boost::log::attributes::current_thread_id::held_type >(L"ThreadID") << L"] "\r
 153                 //              << L"[" << boost::log::formatters::attr<severity_level>(boost::log::sources::aux::severity_attribute_name<wchar_t>::get()) << L"] "\r
 154                 //              << boost::log::formatters::message<wchar_t>()\r
 155                 //);\r
 156 \r
 157                 file_sink->set_filter(boost::log::filters::attr<severity_level>(boost::log::sources::aux::severity_attribute_name<wchar_t>::get()) >= info);\r
 158                 \r
 159                 boost::log::wcore::get()->add_sink(file_sink);\r
 160 \r
 161                 CASPAR_LOG(info) << L"Logging [info] or higher severity to " << folder << std::endl << std::endl;\r
 162         }\r
 163         catch(...)\r
 164         {\r
 165                 std::wcerr << L"Failed to Setup File Logging Sink" << std::endl << std::endl;\r
 166         }\r
 167 }\r
 168 \r
 169 }}