JLogger.cc

Go to the documentation of this file.

#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"
 
 
namespace {
 
  using namespace JPP;
 
 
  const std::string  save   =   "save";    //!< Close and re-open file;
  const std::string  stop   =   "stop";    //!< Stop process.
 
 
  const int    MINIMUM_NUMBER_OF_MESSAGES  =  100;
  const double MINIMUM_ELAPSED_TIME_S      =  60*60;
 
 
  /**
   * Logger file.
   */
  struct JLoggerFile :
    public ogzstream,
    public JDateAndTime
  {
    static const int MAXIMUM_FILE_NUMBER  = 100;
 
    /**
     * Constructor.
     *
     * \param  path          path
     */
    JLoggerFile(const std::string& path) :
      path(path)
    {}
 
 
    /**
     * Get current file name.
     *
     * \return               file name
     */
    std::string getFilename() const
    {
      return file_name;
    }
 
 
    /**
     * Open file.
     */
    void open()
    {
      using namespace std;
 
      JDateAndTime::set();
 
      for (int i = 0; !this->is_open() && i != MAXIMUM_FILE_NUMBER; ++i) {
 
        ostringstream os;
 
        os << getFullPath(path) 
           << "KM3NeT" 
           << "_"  << this->getYear() << '-' << FILL(2,'0') << this->getMonth() << '-' << FILL(2,'0') << this->getDay();
        
        if (i != 0) {
          os << "_" << i;
        }
        
        os << ".txt.gz";
 
        file_name = os.str();
 
        if (!getFileStatus(file_name.c_str())) {
          ogzstream::open(file_name.c_str());
        }
      }
    }
 
 
    /**
     * Close file.
     */
    void close()
    {
      using namespace std;
 
      ogzstream::close();
 
      file_name = "";
    }    
 
  protected:
    std::string path;
    std::string file_name;
  };
}
 
 
/**
 * \file
 *
 * Auxiliary program to save logger messages from ControlHost server.
 *
 * The option <tt>-H <hostname>[:port]</tt> 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
 */
int main(int argc, const char *argv[])
{
  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);
  }
}