log.hh

Go to the documentation of this file.

#ifndef DATAQUEUE_LOG_HH
#define DATAQUEUE_LOG_HH

#include <sstream>
#include <string>
#include <map>
#include <boost/thread/mutex.hpp>

/**
 * \author cpellegrino
 */

namespace Log {

enum LogLevel {cDebug, cWarning, cError, cNotice, cStatus};

class Logger
{
  std::ostringstream m_stream;

  LogLevel m_level;

 public:

  Logger(LogLevel level)
    :
    m_level(level)
  {}

  std::ostringstream& stream()
  {
    return m_stream;
  }

  ~Logger();
};

class Counter
{
  typedef std::map< std::string, unsigned int> container_t;

  container_t m_counters;
  mutable boost::mutex m_mutex;

  Counter() {}

  Counter(Counter const&);
  Counter& operator =(Counter const&);

 public:

  static
  Counter& get()
  {
    static Counter c;

    return c;
  }

  void add(std::string const& tag)
  {
    boost::mutex::scoped_lock lock(m_mutex);

    container_t::iterator const it = m_counters.find(tag);

    if (it != m_counters.end()) {
      ++it->second;
    } else {
      m_counters.insert(std::make_pair(tag, 1));
    }
  }

  void reset()
  {
    boost::mutex::scoped_lock lock(m_mutex);
    m_counters.clear();
  }

  void reset(std::string const& tag)
  {
    boost::mutex::scoped_lock lock(m_mutex);

    container_t::iterator const it = m_counters.find(tag);

    if (it != m_counters.end()) {
      it->second = 0;
    }
  }

  friend
  std::ostream& operator <<(std::ostream& stream, Counter const& c)
  {
    boost::mutex::scoped_lock lock(c.m_mutex);

    for (
        container_t::const_iterator it = c.m_counters.begin(), et = c.m_counters.end();
        it != et;
        ++it
    ) {
      stream << it->first << ": " << it->second << ", ";
    }

    return stream;
  }
};

} // ns Log

#define LOG_DEBUG   Log::Logger(Log::cDebug  ).stream()
#define LOG_WARNING Log::Logger(Log::cWarning).stream()
#define LOG_ERROR   Log::Logger(Log::cError  ).stream()
#define LOG_NOTICE  Log::Logger(Log::cNotice ).stream()
#define LOG_STATUS  Log::Logger(Log::cDebug  ).stream()

#endif // DATAQUEUE_LOG_HH