WatcherServer.C

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2010 by Olivier Stezowski                               *
 *   stezow(AT)ipnl.in2p3.fr                                               *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

// ADF includes

#include "GwLogCollector.h"

#ifndef ADF_FrameBlock
#include "FrameBlock.h"
#endif
#ifndef ADF_NarvalInterface
#include "NarvalInterface.h"
#endif

// RO0T includes
#ifndef ROOT_TServerSocket
#include "TServerSocket.h"
#endif
#ifndef ROOT_TMonitor
#include "TMonitor.h"
#endif
#ifndef ROOT_TList
#include "TList.h"
#endif

// system include
#include <string>

using namespace std;
using namespace ADF;

template <typename Prod_actor>
class WatcherServer
{
private:
        LogMessage fLog;
        
private:
        Prod_actor *fProducer;
        OneBlock fBProd;
        string fPathProd;
        
private:
        UInt_t fPort;
        string fUrl;

private:        
        TServerSocket *fServ;      // server socket
        TMonitor      *fMon;       // socket monitor
        TList         *fSockets;   // list of open spy sockets
        
protected:
        void HandleSocket(TSocket *s);

public:
        WatcherServer(const char *path, UInt_t bsize = aMByte);
        ~WatcherServer();
        
        Bool_t Init();  
        Bool_t Run(Int_t askedloop = -1);
};

template <typename Prod_actor> 
WatcherServer<Prod_actor>::WatcherServer(const char *path, UInt_t bsize):
        fLog("WatcherServer"),
        fProducer(0x0),
        fBProd(bsize,ConfAgent::kWrite),
        fPathProd(path),
        fPort(9090),
        fUrl("localhost"),
        fServ(0x0),
        fMon(0x0),
        fSockets( new TList() ) 
{
        fSockets->SetOwner(true);       
}

template <typename Prod_actor> 
WatcherServer<Prod_actor>::~WatcherServer()
{
        // 
        if ( fProducer ) 
                delete fProducer; 
        if ( fMon ) 
                fMon->Remove(fServ);
        if ( fServ ) 
                delete fServ;   
        if ( fMon ) 
                delete fMon;
        if ( fSockets )
                delete fSockets;        
}

template <typename Prod_actor> 
Bool_t WatcherServer<Prod_actor>::Init()
{
        UInt_t err = 0u;
        fLog.SetProcessMethod("Init");
        
        // just in case a actor forget to call it and it is required.
        NarvalInterface::process_config(fPathProd.data(),&err);
        if ( err > 0u ) {
                fLog << warning << " in NarvalInterface::process_config " << nline; 
        } 
        
        err = 0u;
        // it creates and inits the different actors
        // producer
        Prod_actor::process_config(fPathProd.data(),&err); 
        if ( err > 0u ) {
                fLog << error << " in process_config for Producer " << dolog; 
                return false;
        }
        fProducer = new Prod_actor(); 
        fProducer->process_initialise(&err);
        if ( err > 0u )  {
                fLog << error << " in process_initialse for Producer " << dolog; 
                return false;
        }
        
        // connection to send buffers
        fServ = new TServerSocket(fPort, kTRUE);

        // Add server socket to monitor so we are notified when a client needs to be    
        // accepted
        fMon  = new TMonitor();
        fMon->Add(fServ);
        
        return true;
}


template <typename Prod_actor> 
Bool_t WatcherServer<Prod_actor>::Run(Int_t askedloop)
{
        printf("************** EMULATOR MAIN LOOP ************** \n");
        // loop on events
        Int_t nb_loop = 0, nb_loop_throwaway = 0; UInt_t err = 0u;      
        while ( 1 ) {
                // to get an input block                
                if ( gDebug )
                        cout << " Get Block " <<  endl;
                err = fProducer->ProcessBlock(fBProd);
                if ( gDebug )
                        cout << " Block " << fBProd.GetSize() << endl;
                 
                if ( err != 0u ) 
                        break;
                if ( fBProd.GetSize() == 0u )
                        break;
                
                TSocket *s;
        //      if ( (s = fMon->Select(2000)) != (TSocket*)-1 )
                if ( (s = fMon->Select(1)) != (TSocket*)-1 ) {                  
                        HandleSocket(s);
                }
                else {
                        nb_loop_throwaway++;
                        if ( gDebug )
                                cout << " Empty cycle " << nb_loop_throwaway << endl;
                }
                nb_loop++;
                
                if ( nb_loop == askedloop ) 
                        break;  
                // stop from ROOT
//              if (gROOT->IsInterrupted())
//                      break;
        }
        printf("nb_loop %d, nb_loop_unsent %d, error %d\n",nb_loop,nb_loop_throwaway,err);
        printf("************** EMULATOR MAIN LOOP ************** \n");
        return true;
}

template <typename Prod_actor> 
void WatcherServer<Prod_actor>::HandleSocket(TSocket *s)
{
        UInt_t size;
        if ( gDebug ) 
                printf("HANDLE SOCKET \n");
        
        // accept new connection from spy and send it data 
        if (s->IsA() == TServerSocket::Class()) {
                TSocket *sock = ((TServerSocket*)s)->Accept();
                fMon->Add(sock);
                fSockets->Add(sock);
                if ( gDebug )           
                  fLog << info << "accepted connection from " 
                       << sock->GetInetAddress().GetHostName() << dolog;
                       
                if ( gDebug )   
                        printf("SEND DATA \n"); 
                size = fBProd.GetSize() ;
                sock->SendRaw(&size,sizeof(size));
                if ( size > 0u ) 
                        sock->SendRaw(fBProd.GetBufferIO()->GetAddress(),size);         
        } 
        else {  // means a socket wants to close the connection 
                fMon->Remove(s);
                fSockets->Remove(s);
                if ( gDebug )   
                         fLog << info << "closed connection from " 
                              << s->GetInetAddress().GetHostName() << dolog;
                delete s;
        }
}

#ifndef ADF_BasicAFP
#include "BasicAFP.h"
#endif

#ifndef ADF_RootAFP
#include "RootAFP.h"
#endif

void RunWatcherServer(const char *path, Int_t nbloop = -1, UInt_t bsize = aMByte) 
{
  WatcherServer< RootAFP > aserver(path,bsize);
  
  // Init the Producer Actor
  aserver.Init();
  
  // loop
  aserver.Run(nbloop);
}

void InfiniteWatcherServer(const char *path, UInt_t bsize = aMByte, UInt_t nb_times = 10000u) 
{
  for (UInt_t i = 0u; i <nb_times ; i++) {
        RunWatcherServer(path,-1,bsize);
  }
}