MONITOR/alistoragemanager/AliStorageServerThread.cxx

   1 #include "AliStorageServerThread.h"
   2 #include "AliStorageTypes.h"
   3 #include "AliESDEvent.h"
   4
   5 #include <iostream>
   6 #include <fstream>
   7
   8 #include <TFile.h>
   9 #include <TThread.h>
  10
  11 using namespace std;
  12
  13 AliStorageServerThread::AliStorageServerThread() :
  14         fDatabase(0),
  15         fStoragePath("")
  16 {
  17         TThread::Lock();
  18         fDatabase = new AliStorageDatabase();
  19         //load parameters from config file
  20         ifstream configFile (GetConfigFilePath());
  21
  22
  23         if (configFile.is_open())
  24         {
  25                 string line;
  26                 int from,to;
  27                 while(configFile.good())
  28                 {
  29                         getline(configFile,line);
  30                         from = line.find("\"")+1;
  31                         to = line.find_last_of("\"");
  32                         if(line.find("STORAGE_PATH=")==0)
  33                         {
  34                                 fStoragePath=line.substr(from,to-from);
  35                         }
  36                 }
  37                 if(configFile.eof())
  38                 {
  39                         configFile.clear();
  40                 }
  41                 configFile.close();
  42         }
  43         else
  44         {
  45                 cout<<"SERVER -- Unable to open config file"<<endl;
  46         }
  47         TThread::UnLock();
  48
  49         //start communication on socket
  50         StartCommunication();
  51 }
  52
  53 AliStorageServerThread::~AliStorageServerThread()
  54 {
  55         cout<<"SERVER -- AliStorageServerThread destructor called";
  56         cout<<" --- OK"<<endl;
  57 }
  58
  59 void AliStorageServerThread::StartCommunication()
  60 {
  61         AliStorageEventManager *eventManager = AliStorageEventManager::GetEventManagerInstance();
  62         storageSockets socket = SERVER_COMMUNICATION_REP;
  63         eventManager->CreateSocket(socket);
  64
  65         struct serverRequestStruct *request;
  66
  67         while(1)
  68         {
  69                 request = eventManager->GetServerStruct(socket);
  70
  71                 switch(request->messageType)
  72                 {
  73                 case REQUEST_LIST_EVENTS:
  74                 {
  75                         vector<serverListStruct> result = fDatabase->GetList(request->list);
  76                         eventManager->Send(result,socket);
  77                         break;
  78                 }
  79                 case REQUEST_GET_EVENT:
  80                 {
  81                         AliESDEvent *event = fDatabase->GetEvent(request->event);
  82                         eventManager->Send(event,socket);
  83                         delete event;
  84                         break;
  85                 }
  86                 case REQUEST_GET_NEXT_EVENT:
  87                 {
  88                         AliESDEvent *event = fDatabase->GetNextEvent(request->event);
  89                         eventManager->Send(event,socket);
  90                         delete event;
  91                         break;
  92                 }
  93                 case REQUEST_GET_LAST_EVENT:
  94                 {
  95                         AliESDEvent *event = fDatabase->GetLastEvent();
  96                         eventManager->Send(event,socket);
  97                         delete event;
  98                         break;
  99                 }
 100                 case REQUEST_MARK_EVENT:
 101                 {
 102                         struct eventStruct *markData  = &(request->event);
 103                         eventManager->Send(MarkEvent(*markData),socket);
 104                         break;
 105                 }
 106                 default:break;
 107                 }
 108         }
 109 }
 110
 111 bool AliStorageServerThread::MarkEvent(struct eventStruct event)
 112 {
 113         string pathToFile = fDatabase->GetFilePath(event);
 114         TFile *tmpFile = new TFile(pathToFile.c_str(),"read");
 115         if(!tmpFile)
 116         {
 117                 cout<<"SERVER -- couldn't open temp file"<<endl;
 118                 return false;
 119         }
 120         AliESDEvent *eventToMark = (AliESDEvent*)tmpFile->Get(Form("event%d",event.eventNumber));
 121         if(!eventToMark)
 122         {
 123                 cout<<"SERVER -- couldn't find such event"<<endl;
 124                 if(tmpFile){delete tmpFile;}
 125                 return false;
 126         }
 127         cout<<"SERVER -- Marking event:"<<eventToMark->GetEventNumberInFile()<<endl;
 128
 129         TFile *permFile = new TFile(Form("%s/permEvents.root",fStoragePath.c_str()),"update");//open/create perm file
 130
 131         if(!permFile)
 132         {
 133                 cout<<"SERVER -- Couldn't open perm file"<<endl;
 134                 if(tmpFile){delete tmpFile;}
 135                 if(eventToMark){delete eventToMark;}
 136                 return false;
 137         }
 138
 139         //create new directory for this run
 140         TDirectory *currentRun;
 141         if((currentRun = permFile->mkdir(Form("run%d",event.runNumber))))
 142         {
 143                 cout<<"SERVER -- creating new directory for this run"<<endl;
 144                 currentRun->cd();
 145         }
 146         else
 147         {
 148                 cout<<"SERVER -- opening existing directory for this run"<<endl;
 149                 permFile->cd(Form("run%d",event.runNumber));
 150         }
 151
 152         //try to add record to the database
 153         if(!fDatabase->MarkEvent(event))
 154         {
 155                 cout<<"SERVER -- could not mark event in the database"<<endl;
 156                 if(tmpFile){delete tmpFile;}
 157                 if(eventToMark){delete eventToMark;}
 158                 if(permFile){delete permFile;}
 159                 return false;
 160         }
 161
 162         eventToMark->Write(Form("event%d",event.eventNumber));
 163         permFile->Close();
 164         tmpFile->Close();
 165
 166         if(tmpFile){delete tmpFile;}
 167         if(eventToMark){delete eventToMark;}
 168         if(permFile){delete permFile;}
 169 //      if(currentRun)delete currentRun;//this line crashes if there is no permanent file yet
 170         return true;
 171 }
 172