src/Container/SALOME_ContainerManager.cxx

   1 #include "SALOME_ContainerManager.hxx"
   2 #include "SALOME_NamingService.hxx"
   3 #include "OpUtil.hxx"
   4 #include <sys/types.h>
   5 #include <unistd.h>
   6 #include <vector>
   7
   8 #define TIME_OUT_TO_LAUNCH_CONT 21
   9
  10 using namespace std;
  11
  12 const char *SALOME_ContainerManager::_ContainerManagerNameInNS="ContainerManager";
  13
  14 SALOME_ContainerManager::SALOME_ContainerManager(CORBA::ORB_ptr orb)
  15 {
  16   _NS=new SALOME_NamingService(orb);
  17   PortableServer::POA_var root_poa=PortableServer::POA::_the_root_poa();
  18   PortableServer::ObjectId_var id=root_poa->activate_object(this);
  19   CORBA::Object_var obj=root_poa->id_to_reference(id);
  20   Engines::ContainerManager_var refContMan = Engines::ContainerManager::_narrow(obj);
  21   _NS->Register(refContMan,_ContainerManagerNameInNS);
  22 }
  23
  24 SALOME_ContainerManager::~SALOME_ContainerManager()
  25 {
  26   delete _NS;
  27 }
  28
  29 void SALOME_ContainerManager::Shutdown()
  30 {
  31   ShutdownContainers();
  32   PortableServer::ObjectId_var oid = _default_POA()->servant_to_id(this);
  33   _default_POA()->deactivate_object(oid);
  34   _remove_ref();
  35
  36 }
  37
  38 void SALOME_ContainerManager::ShutdownContainers()
  39 {
  40   _NS->Change_Directory("/Containers");
  41   vector<string> vec=_NS->list_directory_recurs();
  42   for(vector<string>::iterator iter=vec.begin();iter!=vec.end();iter++)
  43     {
  44       CORBA::Object_var obj=_NS->Resolve((*iter).c_str());
  45       Engines::Container_var cont=Engines::Container::_narrow(obj);
  46       if(!CORBA::is_nil(cont))
  47         cont->Shutdown();
  48     }
  49 }
  50
  51 Engines::Container_ptr SALOME_ContainerManager::FindOrStartContainer(const char *containerName, const Engines::MachineList& possibleComputers)
  52 {
  53   Engines::Container_ptr ret=FindContainer(containerName,possibleComputers);
  54   if(!CORBA::is_nil(ret))
  55     return ret;
  56   // Container doesn't exist try to launch it ...
  57   vector<string> vector;
  58   string theMachine=_LoadManager.FindBest(possibleComputers);
  59   string command;
  60   if(theMachine==GetHostname())
  61     command=_ResManager.BuildCommandToLaunchLocalContainer(containerName);
  62   else
  63     command=_ResManager.BuildTempFileToLaunchRemoteContainer(theMachine,containerName);
  64   _ResManager.RmTmpFile();
  65   int status=system(command.c_str());
  66   if (status == -1) {
  67     MESSAGE("SALOME_LifeCycleCORBA::StartOrFindContainer rsh failed (system command status -1)");
  68     return Engines::Container::_nil();
  69   }
  70   else if (status == 217) {
  71     MESSAGE("SALOME_LifeCycleCORBA::StartOrFindContainer rsh failed (system command status 217)");
  72     return Engines::Container::_nil();
  73   }
  74   else {
  75     int count=TIME_OUT_TO_LAUNCH_CONT;
  76     while ( CORBA::is_nil(ret) && count ) {
  77       sleep( 1 ) ;
  78       count-- ;
  79       if ( count != 10 )
  80         MESSAGE( count << ". Waiting for FactoryServer on " << theMachine);
  81       string containerNameInNS=BuildContainerNameInNS(containerName,theMachine.c_str());
  82       CORBA::Object_var obj = _NS->Resolve(containerNameInNS.c_str());
  83       ret=Engines::Container::_narrow(obj);
  84     }
  85     if ( CORBA::is_nil(ret) ) {
  86       MESSAGE("SALOME_LifeCycleCORBA::StartOrFindContainer rsh failed");
  87     }
  88     return ret;
  89   }
  90 }
  91
  92 Engines::MachineList *SALOME_ContainerManager::GetFittingResources(const Engines::MachineParameters& params, const char *componentName)
  93 {
  94   vector<string> vec=_ResManager.GetFittingResources(params,componentName);
  95   Engines::MachineList *ret=new Engines::MachineList;
  96   ret->length(vec.size());
  97   for(unsigned int i=0;i<vec.size();i++)
  98     {
  99       (*ret)[i]=(vec[i]).c_str();
 100     }
 101   return ret;
 102 }
 103
 104 char* SALOME_ContainerManager::FindBest(const Engines::MachineList& possibleComputers)
 105   {
 106     string theMachine=_LoadManager.FindBest(possibleComputers);
 107     return CORBA::string_dup(theMachine.c_str());
 108   }
 109
 110 Engines::Container_ptr SALOME_ContainerManager::FindContainer(const char *containerName,const char *theMachine)
 111 {
 112   string containerNameInNS(BuildContainerNameInNS(containerName,theMachine));
 113   CORBA::Object_var obj = _NS->Resolve(containerNameInNS.c_str());
 114   if( !CORBA::is_nil(obj) )
 115     return Engines::Container::_narrow(obj);
 116   else
 117     return Engines::Container::_nil();
 118 }
 119
 120 Engines::Container_ptr SALOME_ContainerManager::FindContainer(const char *containerName,const Engines::MachineList& possibleComputers)
 121 {
 122   for(unsigned int i=0;i<possibleComputers.length();i++)
 123     {
 124       Engines::Container_ptr cont=FindContainer(containerName,possibleComputers[i]);
 125       if( !CORBA::is_nil(cont) )
 126         return cont;
 127     }
 128   return Engines::Container::_nil();
 129 }
 130
 131 string SALOME_ContainerManager::BuildContainerNameInNS(const char *containerName,const char *machineName)
 132 {
 133   string containerNameInNS("/Containers/");
 134   containerNameInNS+=machineName;
 135   containerNameInNS+="/";
 136   containerNameInNS+=containerName;
 137   return containerNameInNS;
 138 }