FairMQ/fairmq/FairMQLogger.cxx

/**
 * FairMQLogger.cxx
 *
 * @since 2012-12-04
 * @author D. Klein, A. Rybalchenko
 */

#include <string>
#include <stdio.h>

#include "FairMQLogger.h"

using std::string;
using std::cout;
using std::endl;

FairMQLogger::FairMQLogger()
{
}

FairMQLogger::~FairMQLogger()
{
  cout << os.str() << endl;
}

std::ostringstream& FairMQLogger::Log(int type)
{
  timestamp_t tm = get_timestamp();
  timestamp_t ms = tm / 1000.0L;
  timestamp_t s = ms / 1000.0L;
  std::time_t t = s;
  std::size_t fractional_seconds = ms % 1000;
  char mbstr[100];
  std::strftime(mbstr, 100, "%H:%M:%S:", std::localtime(&t));

  string type_str;
  switch (type) {
    case DEBUG:
      type_str = "\033[01;34mDEBUG\033[0m";
      break;
    case INFO:
      type_str = "\033[01;32mINFO\033[0m";
      break;
    case ERROR:
      type_str = "\033[01;31mERROR\033[0m";
      break;
    case STATE:
      type_str = "\033[01;33mSTATE\033[0m";
    default:
      break;
  }

  os << "[\033[01;36m" <<  mbstr << fractional_seconds << "\033[0m]" << "[" << type_str << "]" << " ";

  return os;
}

timestamp_t get_timestamp ()
{
  struct timeval now;
  gettimeofday (&now, NULL);
  return now.tv_usec + (timestamp_t)now.tv_sec * 1000000;
}