JLogger.cc File Reference

Auxiliary program to save logger messages from ControlHost server. More...

#include <string>
#include <iostream>
#include <fstream>
#include <sstream>
#include <iomanip>
#include <set>
#include "JLogger/JMessageLogger.hh"
#include "JNet/JControlHost.hh"
#include "JLang/JTimeval.hh"
#include "JLang/JException.hh"
#include "JLang/gzstream.h"
#include "JSystem/JDateAndTime.hh"
#include "JSystem/JStat.hh"
#include "Jeep/JeepToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Functions
int	main (int argc, const char *argv[])

Detailed Description

Auxiliary program to save logger messages from ControlHost server.

The option -H <hostname>[:port] correponds to the hostname and the port of the server, respectively. Each message tag will be written to file in the directory specified. The program will terminate when it receives message <stop>.

Author

mdejong

Definition in file JLogger.cc.

Function Documentation

int main	(	int	argc,
		const char *	argv[]
	)

Definition at line 126 of file JLogger.cc.

{
  using namespace std;
  using namespace JPP;

  string          hostname;
  set<JTag>       taglist;
  string          path;
  JTimeval        timeout;
  set<int>        happy_hour;
  int             debug;

  taglist.insert(MESSAGE_TAG);
  
  try {

    JParser<> zap("Auxiliary program to save logger messages from ControlHost server.");

    zap['H'] = make_field(hostname)        = "localhost";
    zap['T'] = make_field(taglist)         = JPARSER::initialised();
    zap['D'] = make_field(path)            = "/tmp/";
    zap['t'] = make_field(timeout)         = JTimeval(1, 0);
    zap['W'] = make_field(happy_hour)      = JPARSER::initialised();
    zap['d'] = make_field(debug)           = 1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  if (happy_hour.empty()) {

    WARNING("No happy hours (option -W); set to midnight." << endl);

    happy_hour.insert(0);
  }


  JControlHost::Throw(true); 
  
  try {

    JControlHost in(hostname);

    {
      JSubscriptionList buffer;
      
      for (set<JTag>::const_iterator i = taglist.begin(); i != taglist.end(); ++i) {
        buffer.add(JSubscriptionAll(*i));
      }

      in.Subscribe(buffer);
      in.SendMeAlways();
    }


    JLoggerFile out(path);

    out.open();

    DEBUG("Open file  " << out.getFilename() << endl);

    JPrefix prefix;
    string  buffer;

    JDateAndTime clock;

    for (int number_of_messages; buffer.size() != stop.size() || string(buffer.data(), stop.size()) != stop; ) {

      const int check = in.CheckHead(prefix, timeout);

      DEBUG("Check head " << check << endl);

      if        (check  < 0) {

        FATAL("Error at JControlHost::CheckHead " << check << endl);

      } else if (check == 1) {

        in.GetFullString(buffer);

        DEBUG("Message <" << buffer << ">" << endl);

        if        (buffer == stop) {

          break;

        } else if (buffer == save) {

          DEBUG("Close file " << out.getFilename() << endl);

          out.close();
          out.open();

          DEBUG("Open file  " << out.getFilename() << endl);

        } else {

          ++number_of_messages;

          if ( ! (out << buffer << endl)) {
            FATAL("Error writing to file " << out.getFilename() << endl);
          }
        }
      }

      if (check == 0 || number_of_messages > MINIMUM_NUMBER_OF_MESSAGES) {

        number_of_messages = 0;

        clock.set();

        DEBUG("Test " << setw(2) << clock.getHour() << ' ' << happy_hour.count(clock.getHour()) << ' ' << clock.getElapsedTime(out) << endl);

        if (happy_hour.count(clock.getHour()) != 0 && clock.getElapsedTime(out) > MINIMUM_ELAPSED_TIME_S) {

          DEBUG("Close file " << out.getFilename() << endl);

          out.close();
          out.open();

          DEBUG("Open file " << out.getFilename() << endl);
        }
      }
    }

    out.close();
  }    
  catch(const JControlHostException& error) {
    ERROR(error << endl);
  }
}