Utils/Logger/IOStreamTarget.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2007
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief IOStreamTarget non-inline non-template implementation */
  25
  26 #include "IOStreamTarget.hh"
  27 #include "IOStreamTarget.ih"
  28
  29 // Custom includes
  30 #include <errno.h>
  31 #include <locale>
  32 #include <sstream>
  33 #include <boost/algorithm/string/trim.hpp>
  34 #include <boost/tokenizer.hpp>
  35 #include <boost/date_time/posix_time/posix_time.hpp>
  36
  37 //#include "IOStreamTarget.mpp"
  38 #define prefix_
  39 ///////////////////////////////cc.p////////////////////////////////////////
  40
  41 ///////////////////////////////////////////////////////////////////////////
  42 // senf::log::IOStreamTarget
  43
  44 prefix_ senf::log::IOStreamTarget::IOStreamTarget(std::ostream & os)
  45     : stream_ (os), tag_ (detail::getDefaultTag()), noformat_ (false), showTime_ (true),
  46       showStream_ (false), showLevel_ (true), showArea_ (true)
  47 {
  48     std::locale const & loc (datestream_.getloc());
  49     datestream_.imbue( std::locale(
  50                            loc, new boost::posix_time::time_facet("%Y-%m-%d %H:%M:%S.%f-0000")) );
  51
  52 }
  53
  54 prefix_ void senf::log::IOStreamTarget::timeFormat(std::string const & format)
  55 {
  56     if (format.empty())
  57         noformat_ = true;
  58     else {
  59         noformat_ = false;
  60         std::locale const & loc (datestream_.getloc());
  61         datestream_.imbue( std::locale(
  62                                loc, new boost::posix_time::time_facet(format.c_str())) );
  63     }
  64 }
  65
  66 ////////////////////////////////////////
  67 // private members
  68
  69 prefix_ void senf::log::IOStreamTarget::v_write(time_type timestamp,
  70                                                 std::string const & stream,
  71                                                 std::string const & area, unsigned level,
  72                                                 std::string const & message)
  73 {
  74     std::string m (message);
  75     boost::trim_right(m);
  76     detail::quoteNonPrintable(m);
  77
  78     if (tag_.empty() && !showTime_ && !showStream_ && !showLevel_ && !showArea_)
  79         stream_ << m << std::endl;
  80     else {
  81         typedef boost::char_separator<char> Separator;
  82         typedef boost::tokenizer<Separator> Tokenizer;
  83         Separator separator ("\n");
  84         Tokenizer tokenizer (m, separator);
  85         Tokenizer::iterator i (tokenizer.begin());
  86         Tokenizer::iterator const i_end (tokenizer.end());
  87
  88         if (showTime_) {
  89             if (noformat_)
  90                 datestream_ << std::setfill('0') << std::setw(19) << timestamp;
  91             else
  92                 datestream_ << senf::ClockService::abstime(timestamp);
  93             datestream_ << ' ';
  94         }
  95         if (!tag_.empty())
  96             datestream_ << tag_ << ": ";
  97         if (showStream_)
  98             datestream_ << '[' << stream << "] ";
  99         if (showLevel_)
 100             datestream_ << '[' << LEVELNAMES[level] << "] ";
 101         if (showArea_ && area != "senf::log::DefaultArea")
 102             datestream_ << '[' << area << "] ";
 103
 104         for (; i != i_end; ++i)
 105             stream_ << datestream_.str() << *i << "\n";
 106         stream_ << std::flush;
 107         datestream_.str("");
 108     }
 109 }
 110
 111 ///////////////////////////////////////////////////////////////////////////
 112
 113 prefix_ void senf::log::detail::quoteNonPrintable(std::string & s)
 114 {
 115     for (std::string::iterator i (s.begin()); i != s.end(); ++i)
 116         if (*i < ' ' && *i != '\n')
 117             *i = ' ';
 118 }
 119
 120 prefix_ std::string senf::log::detail::getDefaultTag()
 121 {
 122     std::stringstream ss;
 123     ss << ::program_invocation_short_name << '[' << ::getpid() << ']';
 124     return ss.str();
 125 }
 126
 127 ///////////////////////////////cc.e////////////////////////////////////////
 128 #undef prefix_
 129 //#include "IOStreamTarget.mpp"
 130
 131 \f
 132 // Local Variables:
 133 // mode: c++
 134 // fill-column: 100
 135 // comment-column: 40
 136 // c-file-style: "senf"
 137 // indent-tabs-mode: nil
 138 // ispell-local-dictionary: "american"
 139 // compile-command: "scons -u test"
 140 // End: