dbharpo.cxx

Go to the documentation of this file.

//
// harpo database manipulation utility
// Write access to harpo database was required !!!
// Can be run on harpo servers ONLY !!!
//

#include "HarpoDB.h"
#include "HarpoDBrunconfig.h"
#include "TXMLFile.h"
#include "TString.h"
#include "TSystem.h"
#include <iostream>
#include <stdlib.h>
#include <unistd.h>
#include <getopt.h>
#include <libgen.h>

typedef enum DBACTIONS_
  { RNONE, RSTART, REND, RRUN, RLAST, RGEN, RQUALITY } DBACTIONS; 

static const char* const short_options = "hvDnc:s:e:rlg::Q:";

static const struct option long_options[] = {
  { "help", 0, NULL, 'h' },
  { "verbose", 0, NULL, 'v' },
  { "debug", 0, NULL, 'D' },
  { "dry-run", 0, NULL, 'n' },
  { "startrun", 1, NULL, 's' },
  { "endrun", 1, NULL, 'e' },
  { "running", 0, NULL, 'r' },
  { "lastrun", 0, NULL, 'l' },
  { "config", 1, NULL, 'c' },
  { "generate",2,NULL,'g' },
  { "runquality",1,NULL,'Q' },
  { NULL, 0, NULL, 0 }
};

static TString *ProgramName;
static TString *CfgFileName;
static int Verbose = 0;
static int Debug = 0;
  
void help()
{
  printf("Usage: %s -c cfgfile [OPTIONS] [ACTION] ...\n",ProgramName->Data());
  printf("  Where options :\n\n"
         "  -h, --help\t print this text and exit\n"
         "  -v, --verbose\t increase verbosity level, can be repeated several times\n"
         "  -D, --debug\t increase debug level can repeat several times\n"
         "  -n, --dry-run\t perform a trial run with no changes made\n"
         "  -c file , --config\t use file us config file (defult harpodbrc.xml)\n"
         "actions:\n"
         "  -sNNNN , --startrun\t create new run record in DB and mark the run us running\n"
         "  -eNNNN , --endrun\t mark run us finished and update run duration\n"
         "  -r, --running\t show runs marked us running in DB\n"
         "  -l, --lastrun\t show the last run number in DB\n"
         "  -g[NNNN] , --generate\t crete configuration file using information from run NNNN\n"
         "  -QNNNN str(12) , --runquality\t set RunQuality for run NNNN to str\n"
         "\n");
  exit(1);
}

typedef dbResults::size_type dbResSize;

dbResSize RunningList(dbResults &run_list)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  HarpoDB *db;
  
  db = new HarpoDB;
  db->Open();
  
  if (db->ExecQuery("SELECT `run` FROM `runconfig` WHERE `RunQuality` LIKE 'Running%';",&run_list) ) {
    if ( Debug > 0)std::cout << "Ok, size of result " << run_list.size() << std::endl;
  } else
    std::cout << "Bad SELET QUERY"  << std::endl;

  db->Close();
  delete db;
  return run_list.size();
}

// Read run template from file
void ReadConfig(HarpoDBRow &rinfo)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  if ( gSystem->AccessPathName(CfgFileName->Data(),kReadPermission) ) {
    std::cout << "File not exist " <<  *CfgFileName<< std::endl;
  } else {
    HarpoDBrunconfig *entry = nullptr;

    TFile * rcfile = TXMLFile::Open(CfgFileName->Data());
    if (Debug > 0 ) rcfile->ls();
    rcfile->GetObject("HarpoDBrunconfig",entry);
    if (entry != 0 ) {
      entry->Export(rinfo);
      delete entry;
    }
    delete rcfile; // close file
  }
  if (Debug > 0 ) rinfo.list();
}


void StartRun(long int r)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  dbResults run_list;
  dbResSize nrun = RunningList(run_list);
  if ( nrun >  0 ) {
    std::cout << "Warning : " << nrun << " runs marked as running Harpo DB"  << std::endl;
  }

  HarpoDB *db = new HarpoDB;
  //db->SetUser("harpo","harpo");
  if ( ! db->Open() ) return;
  
  HarpoDBTable t0;
  db->GetTable("runconfig",t0);
  if (Verbose > 1 ) t0.list();

  // check if run in DB
  HarpoDBRow runinfo;
  Bool_t ok = t0.GetRunRow(r,runinfo);
  if (ok) {
    std::cout << "Error : Run " << r << " exist in Harpo DB"  << std::endl;
    return;
  }
  runinfo.SetFields(t0.GetFiledsTypes());
  ReadConfig(runinfo);
  runinfo.SetVal("RunQuality",TString("Running"));
  t0.TypeExpression("date");
  runinfo.SetVal("date",TString("NOW()"));
  runinfo.SetVal("Duration",TString());
  runinfo.SetVal("nevt",0L);
  runinfo.SetRun(r);
  if (Debug > 0 ) {
  std::cout << "----------------------------------------"  << std::endl;
  runinfo.list();
  }
  TString opt = ((Debug > 0) ? "verbose" : "" );
  t0.InsertRunRow(runinfo,opt);
  if ( Verbose > 1 ) std::cout << "Run " << r << " Started."  << std::endl;
  return;
  
}

void RunDone(long int r)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  HarpoDB *db;
  //HarpoDBrunconfig * entry;

  db = new HarpoDB;
  db->Open();

  HarpoDBTable t0;
  db->GetTable("runconfig",t0);
  // Narrow query to "intresting" fileds
  dbFields fields = t0.GetFiledsTypes();
  for (dbFields::iterator it=fields.begin(); it!=fields.end(); ++it) {
    if (!( it->first.Contains("run") || it->first.Contains("RunQuality")
           || it->first.Contains("Duration") ))
      { // Other, we not need
        t0.Erase(it->first);
      } 
  }
  if (Verbose > 1 ) t0.list();

  Bool_t ok;
  HarpoDBRow runinfo;
  ok = t0.GetRunRow(r,runinfo);
  if (ok) {
    runinfo.SetVal("RunQuality",TString("Done"));
    t0.TypeExpression("Duration");
    runinfo.SetVal("Duration",TString("TIMEDIFF(NOW(),`date`)"));
    if ( Verbose > 0 ) runinfo.list();
    TString opt = ((Debug > 0) ? "verbose" : "" );
    t0.UpdateRunRow(runinfo,opt);
  } else {
        std::cout << "Error : Run " << r << " not in Harpo DB"  << std::endl;

  }

  db->Close();
  delete db;
}
void EndRun(long int r)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  dbResults run_list;

  if (RunningList(run_list) > 0) {
    for(dbResults::iterator il=run_list.begin();il !=run_list.end();++il){
      for(dbFields::iterator im=il->begin();im != il->end();++im){
        if (im->first.EqualTo("run") && ( im->second.Atoll() == r)) {
          std::cout << "Ending run " << im->second << std::endl;
          RunDone(r);
          return;
        }
      }
    }
  } else {
    std::cout << "No Runs running" << std::endl;
  }
  std::cout << "Run " << r << " not running" << std::endl;
}

void GetRunning()
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  dbResults run_list;
  
  if (RunningList(run_list) > 0) {
    for(dbResults::iterator il=run_list.begin();il !=run_list.end();++il){
      if (Verbose > 0) std::cout << " ======================" << std::endl;
      for(dbFields::iterator im=il->begin();im != il->end();++im){
        if (Verbose > 0) {
          std::cout << im->first << " = " << im->second << std::endl;
        } else {
          std::cout << im->second << std::endl;
        }
      }
    }
  } else {
    std::cout << "No Runs running" << std::endl;
  }
}

void GetLastRun()
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  HarpoDB *db;
  
  db = new HarpoDB;
  db->Open();
  
  dbResults run_list;
  
  if (db->ExecQuery("SELECT MAX(`run`) FROM `runconfig`;",&run_list) ) {
    if ( Debug > 0)std::cout << "Ok, size of result " << run_list.size() << std::endl;
  } else
    std::cout << "Bad SELET QUERY"  << std::endl;
  
  for(dbResults::iterator il=run_list.begin();il !=run_list.end();++il){
    if (Verbose > 0) std::cout << " ======================" << std::endl;
    for(dbFields::iterator im=il->begin();im != il->end();++im){
      if (Verbose > 0) {
        std::cout << im->first << " = " << im->second << std::endl;
      } else {
        std::cout << im->second << std::endl;
      }
    }
  }
  db->Close();
  delete db;
}

void SetConfig(const char *file)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  delete CfgFileName;
  CfgFileName = new TString(file);
}

void GenConfig(long int r)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  HarpoDB *db;
  HarpoDBrunconfig * entry;
  TFile *rcfile;

  db = new HarpoDB;
  db->Open();

  HarpoDBTable t0;
  db->GetTable("runconfig",t0);
  if (Verbose > 1 ) t0.list();

  // runconfig raw in "ROOT" format
  entry = new HarpoDBrunconfig();
  entry->Reset(); // full wit zero values

  Bool_t ok;
  HarpoDBRow runinfo;
  ok = t0.GetRunRow(r,runinfo);
  if (ok) {
    if ( Verbose > 0 ) runinfo.list();
    entry->Import(runinfo);
    // clan up some  field
    entry->nevt = 0;  
    entry->Duration = ""; // time  
    entry->RunQuality = ""; // varchar(12)
  }

  rcfile = TXMLFile::Open(CfgFileName->Data(),"RECREATE","Harpo DB rc file",0);

  entry->Write();
  rcfile->Close();
  delete entry;
  delete rcfile;
  db->Close();
  delete db;
}

void SetRunQuality(long int r,const char *qstr)
{
  if (Debug > 0 ) std::cout << __FUNCTION__ << std::endl;
  HarpoDB *db;
  //  HarpoDBrunconfig * entry;

  db = new HarpoDB;
  db->Open();

  HarpoDBTable t0;
  db->GetTable("runconfig",t0);
  // Narrow query to "intresting" fileds
  dbFields fields = t0.GetFiledsTypes();
  for (dbFields::iterator it=fields.begin(); it!=fields.end(); ++it) {
    if (!( it->first.Contains("run") || it->first.Contains("RunQuality")))
      { // Other, we not need
        t0.Erase(it->first);
      } 
  }
  if (Verbose > 1 ) t0.list();

  Bool_t ok;
  HarpoDBRow runinfo;
  ok = t0.GetRunRow(r,runinfo);
  if (ok) {
    runinfo.SetVal("RunQuality",TString(qstr));
    if ( Verbose > 0 ) runinfo.list();
    TString opt = ((Debug > 0) ? "verbose" : "" );
    t0.UpdateRunRow(runinfo,opt);
  } else {
        std::cout << "Error : Run " << r << " not in Harpo DB"  << std::endl;

  }

  db->Close();
  delete db;
}

int main(int argc, char **argv)
{
  int c;
  long int run;
  DBACTIONS action = RNONE;
  Bool_t doUpdates = true;
  
  ProgramName = new TString(basename(argv[0]));
  CfgFileName = new TString("dbharporc.xml");
  while (1) {
    //      int this_option_optind = optind ? optind : 1;
    int option_index = 0;
    
    c = getopt_long(argc, argv, short_options,
                    long_options, &option_index);
    
    if (c == -1) break;

    switch (c) {
    case 0:
      printf("option %s", long_options[option_index].name);
      if (optarg)
        printf(" with arg %s", optarg);
      printf("\n");
      break;

    case 'h':
      help();
      break;
        
    case 'v':
      Verbose++;
      break;
      
    case 'D':
      Debug++;
      break;
      
    case 'n':
      doUpdates = false;
      break;
      
    case 's':
      //printf("option s\n");
      run = strtol(optarg,NULL,0);
      action = RSTART;
      break;
      
    case 'e':
      if ( optarg != NULL ) {
        run = strtol(optarg,NULL,0);
      } else {
        run = 0;
        if (Debug>0) std::cout << "'e' without arg" << std::endl;
      }
      action = REND;
      break;
      
    case 'r':
      action = RRUN;
      break;
      
    case 'l':
      action = RLAST;
      break;
      
    case 'c':
      SetConfig(optarg);
      break;
      
    case 'g':
      if ( optarg != NULL ) {
        run = strtol(optarg,NULL,0);
      } else {
        run = 0;
        if (Debug>0) std::cout << "'g' without arg" << std::endl;
      }
      action = RGEN;
      break;
      
    case 'Q':
      run = strtol(optarg,NULL,0);
      action = RQUALITY;
      break;
      
    case '?':
      printf("UNKNOWN option\n");
      help();
      break;

    default:
      printf("?? getopt returned character code 0%o ??\n", c);
    }
  }

  // user with update rights
  if(doUpdates)
    HarpoDB::SetUser("harpo","harpo");

  switch (action) {
  case RSTART:
   StartRun(run);
   break;
   
 case REND:
   EndRun(run);
   break;
      
 case RRUN:
   GetRunning();
   break;
      
 case RLAST:
   GetLastRun();
   break;
      
  case RGEN:
    GenConfig(run);
    break;
      
  case RQUALITY:
    if ( optind < argc ) {
      SetRunQuality(run,argv[optind]);
    } else {
    SetRunQuality(run,"");
    }
    break;
      
  default:
    help();
    break;
  }
  return 0;
}