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