ExternalONCRPCServerSvc.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "JiveXML/ExternalONCRPCServerSvc.h"
#include "JiveXML/ONCRPCServer.h"
#include "JiveXML/ONCRPCXDRProcs.h"
 
#include <rpc/rpc.h>
 
namespace JiveXML {
 
 
  ExternalONCRPCServerSvc::ExternalONCRPCServerSvc( const std::string& name, ISvcLocator* sl) :
    base_class ( name, sl ),
    m_client(NULL)
  {
      //declare properties
      declareProperty("Hostname",m_hostName="localhost","The name of the host the external server is running on [ default: \"localhost\" ]");
      declareProperty("Timeout",m_timeOut=10,"Timeout when calling the server in seconds [ default: 10 ]");
  }
 
  ExternalONCRPCServerSvc::~ExternalONCRPCServerSvc(){
    ATH_MSG_VERBOSE( "Destructor() "  );
  }
 
  StatusCode ExternalONCRPCServerSvc::initialize() {
    ATH_MSG_VERBOSE( "Initialize()"  );
    
    //Initialize Service base
    if (Service::initialize().isFailure()) return StatusCode::FAILURE;
    
     //Initialize message stream level
    msg().setLevel(outputLevel());
    ATH_MSG_VERBOSE( "Output level is set to " << (int)msg().level()  );
 
    GetClient().ignore();
    
    return StatusCode::SUCCESS;
  }
 
  StatusCode ExternalONCRPCServerSvc::GetClient(){
 
    ATH_MSG_VERBOSE( "GetClient()"  );
 
    //Try to create client if not there yet
#ifndef __APPLE__
    if (!m_client) m_client = clnt_create(m_hostName.c_str(), ONCRPCSERVERPROG,ONCRPCSERVERVERS, "tcp");
#else
    char proto[] = "tcp";
    if (!m_client) m_client = clnt_create((char*)m_hostName.c_str(), ONCRPCSERVERPROG,ONCRPCSERVERVERS, proto);
#endif    
    //Fail if we still don't have a client
    if (!m_client){
      ATH_MSG_WARNING( "Unable to create client for server on host " << m_hostName << " using TCP/IP "  );
      return StatusCode::FAILURE;
    }
 
// xdr_void is defined inconsistently in xdr.h and gets a warning from gcc8.
#if __GNUC__ >= 8
# pragma GCC diagnostic push
# pragma GCC diagnostic ignored "-Wcast-function-type"
#endif
    //Now try pinging the server
    clnt_stat ret = clnt_call(m_client, NULLPROC, (xdrproc_t)xdr_void, NULL,
                             (xdrproc_t)xdr_void, NULL, GetTimeout());
#if __GNUC__ >= 8
# pragma GCC diagnostic pop
#endif
 
    //And check for the result
    if (ret != RPC_SUCCESS){
      ATH_MSG_WARNING( "Failed calling the server on host " << m_hostName << " : " << clnt_sperrno(ret)  );
      //Also destroy the client in this case - otherwise we might be stranded
      //with an invalid client object
      ReleaseClient().ignore();
      //and return failure
      return StatusCode::FAILURE;
    } 
 
    //Be verbose on success
    ATH_MSG_VERBOSE( "Successfully connected to server on host " << m_hostName  );
 
    return StatusCode::SUCCESS;
  }
 
  struct timeval ExternalONCRPCServerSvc::GetTimeout(){
    //Declare return value
    struct timeval timeout = { 0, 0 };
    //Catch for negatives
    if (m_timeOut < 0)  return timeout;
    //Get seconds as integer part
    timeout.tv_sec = static_cast<long>(m_timeOut);
    //Get microseconds as fractional part
    timeout.tv_usec = static_cast<long>((m_timeOut - timeout.tv_sec)*1e6);
    //Finally return it
    return timeout;
  }
 
  StatusCode ExternalONCRPCServerSvc::ReleaseClient(){
    //Destroy the client
    if (m_client) clnt_destroy(m_client);
    //And set pointer to NULL
    m_client=NULL;
    //For now always return success
    return StatusCode::SUCCESS;
  }
 
  StatusCode ExternalONCRPCServerSvc::finalize() {
 
    ATH_MSG_VERBOSE( "Finalize()"  );
 
    ReleaseClient().ignore();
 
    return StatusCode::SUCCESS;
  }
 
  StatusCode ExternalONCRPCServerSvc::UpdateEventForStream( const EventStreamID& evtStreamID, const std::string & eventStr) {
 
    ATH_MSG_VERBOSE( "UpdateEventForStream()"  );
 
    //Check that the event stream id is valid
    if (!evtStreamID.isValid()){
      ATH_MSG_ERROR( "Invalid event stream identifier - cannot add event"  );
      return StatusCode::FAILURE;
    }
 
    //Now create an event C-struct that we can send to the server
    //NOTE: the struct has to be properly initialized
    Event event;
    event.isAvailable = false ; event.isIdentical = false ; event.isCompressed = false ;
    event.RunNumber = evtStreamID.RunNumber();
    event.EventNumber=evtStreamID.EventNumber();
    event.StreamName=strdup(evtStreamID.StreamName().c_str());
    event.NBytes = strlen(eventStr.c_str())+1;
    event.EventData = strdup(eventStr.c_str());
 
    ATH_MSG_VERBOSE( "Created event structure for event " << event.EventNumber 
                     << " from run " << event.RunNumber 
                     << " to be put in stream " << event.StreamName 
                     << " with " << event.NBytes << " bytes"  );
 
    //Try to get a client
    if (GetClient().isFailure()) {
      ATH_MSG_WARNING( "Failed obtaining a client for sever on host " << m_hostName  );
      ATH_MSG_WARNING( " while updating stream " << evtStreamID.StreamName()
                       << " with event " << evtStreamID.EventNumber() 
                       << " from run " << evtStreamID.RunNumber()  );
      return StatusCode::FAILURE;
    }
 
    //Call the client to update the event
    bool isSuccess = false;
    clnt_stat ret = clnt_call(m_client, ONCRPC_SETEVENT_PROC, (xdrproc_t)xdr_event, (caddr_t)&event,
                             (xdrproc_t)xdr_bool,(caddr_t)&isSuccess, GetTimeout());
 
    //And check for the result
    if (ret != RPC_SUCCESS){
      ATH_MSG_WARNING( "Failed calling the server on host " << m_hostName << " : " << clnt_sperrno(ret)  );
      ATH_MSG_WARNING( " while updating stream " << evtStreamID.StreamName()
                       << " with event " << evtStreamID.EventNumber() 
                       << " from run " << evtStreamID.RunNumber()  );
      return StatusCode::FAILURE;
    } 
 
    //Check if the server managed to update the event from return value
    if (!isSuccess){
      ATH_MSG_WARNING( "Server on host " << m_hostName << " returned failure "  );
      ATH_MSG_WARNING( " while updating stream " << evtStreamID.StreamName()
                       << " with event " << evtStreamID.EventNumber() 
                       << " from run " << evtStreamID.RunNumber()  );
      return StatusCode::FAILURE;
    } 
 
    //Be verbose on success
    ATH_MSG_DEBUG( "Server on host " << m_hostName << " returned success " );
    ATH_MSG_DEBUG( " while updating stream " << evtStreamID.StreamName()
                   << " with event " << evtStreamID.EventNumber() 
                   << " from run " << evtStreamID.RunNumber()  );
 
    //Finally free the memory allocated for the event structure
    xdr_free((xdrproc_t)xdr_event,(caddr_t)&event);
 
    return StatusCode::SUCCESS;
  }
 
} //namespace