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 #include "../Scheduler/ClockService.hh"
  37
  38 //#include "IOStreamTarget.mpp"
  39 #define prefix_
  40 ///////////////////////////////cc.p////////////////////////////////////////
  41
  42 ///////////////////////////////////////////////////////////////////////////
  43 // senf::log::IOStreamTarget
  44
  45 prefix_ senf::log::IOStreamTarget::IOStreamTarget(std::ostream & os)
  46     : stream_ (os), tag_ (detail::getDefaultTag()), noformat_ (false), showTime_ (true),
  47       showStream_ (false), showLevel_ (true), showArea_ (true), timeBase_ (-1)
  48 {
  49     std::locale const & loc (datestream_.getloc());
  50     datestream_.imbue( std::locale(
  51                            loc, new boost::posix_time::time_facet("%Y-%m-%d %H:%M:%S.%f-0000")) );
  52
  53 }
  54
  55 prefix_ void senf::log::IOStreamTarget::timeFormat(std::string const & format)
  56 {
  57     if (format.empty()) {
  58         noformat_ = true;
  59         timeBase_ = -1;
  60     } else {
  61         noformat_ = false;
  62         std::locale const & loc (datestream_.getloc());
  63         datestream_.imbue( std::locale(
  64                                loc, new boost::posix_time::time_facet(format.c_str())) );
  65     }
  66 }
  67
  68 ////////////////////////////////////////
  69 // private members
  70
  71 prefix_ void senf::log::IOStreamTarget::v_write(time_type timestamp,
  72                                                 std::string const & stream,
  73                                                 std::string const & area, unsigned level,
  74                                                 std::string const & message)
  75 {
  76     std::string m (message);
  77     boost::trim_right(m);
  78     detail::quoteNonPrintable(m);
  79
  80     if (tag_.empty() && !showTime_ && !showStream_ && !showLevel_ && !showArea_)
  81         stream_ << m << std::endl;
  82     else {
  83         typedef boost::char_separator<char> Separator;
  84         typedef boost::tokenizer<Separator> Tokenizer;
  85         Separator separator ("\n");
  86         Tokenizer tokenizer (m, separator);
  87         Tokenizer::iterator i (tokenizer.begin());
  88         Tokenizer::iterator const i_end (tokenizer.end());
  89
  90         if (showTime_) {
  91             if (noformat_) {
  92                 if (timeBase_ == -1) timeBase_ = timestamp;
  93                 time_type delta (timestamp - timeBase_);
  94                 datestream_ << std::setfill('0') << std::setw(10)
  95                             << (delta / 1000000000ll) << '.'
  96                             << std::setfill('0') << std::setw(9)
  97                             << (delta % 1000000000ll);
  98             }
  99             else
 100                 datestream_ << senf::ClockService::abstime(timestamp);
 101             datestream_ << ' ';
 102         }
 103         if (!tag_.empty())
 104             datestream_ << tag_ << ": ";
 105         if (showStream_)
 106             datestream_ << '[' << stream << "] ";
 107         if (showLevel_)
 108             datestream_ << '[' << LEVELNAMES[level] << "] ";
 109         if (showArea_ && area != "senf::log::DefaultArea")
 110             datestream_ << '[' << area << "] ";
 111
 112         for (; i != i_end; ++i)
 113             stream_ << datestream_.str() << *i << "\n";
 114         stream_ << std::flush;
 115         datestream_.str("");
 116     }
 117 }
 118
 119 ///////////////////////////////////////////////////////////////////////////
 120
 121 prefix_ void senf::log::detail::quoteNonPrintable(std::string & s)
 122 {
 123     for (std::string::iterator i (s.begin()); i != s.end(); ++i)
 124         if (*i < ' ' && *i != '\n')
 125             *i = ' ';
 126 }
 127
 128 prefix_ std::string senf::log::detail::getDefaultTag()
 129 {
 130     std::stringstream ss;
 131     ss << ::program_invocation_short_name << '[' << ::getpid() << ']';
 132     return ss.str();
 133 }
 134
 135 ///////////////////////////////cc.e////////////////////////////////////////
 136 #undef prefix_
 137 //#include "IOStreamTarget.mpp"
 138
 139 \f
 140 // Local Variables:
 141 // mode: c++
 142 // fill-column: 100
 143 // comment-column: 40
 144 // c-file-style: "senf"
 145 // indent-tabs-mode: nil
 146 // ispell-local-dictionary: "american"
 147 // compile-command: "scons -u test"
 148 // End: