Salome HOME
[EDF28648] : On container launch time from ContainerManager it s possible to execute...
[modules/kernel.git] / src / Container / SALOME_Container_i.hxx
index 56ba2be21c9d078d33b768d6887e136cce6bb349..ca2fac0a354aeb7b477a68bdfbbb6c61f4e2714a 100644 (file)
@@ -1,4 +1,4 @@
-// Copyright (C) 2007-2019  CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2007-2023  CEA, EDF, OPEN CASCADE
 //
 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 #include <list>
 #include <string>
 
-class SALOME_NamingService;
+class SALOME_NamingService_Container_Abstract;
 
-class CONTAINER_EXPORT Engines_Container_i:
-  public virtual POA_Engines::Container,
-  public virtual PortableServer::ServantBase
+class CONTAINER_EXPORT Abstract_Engines_Container_i : public virtual POA_Engines::Container,
+                                                      public virtual PortableServer::ServantBase
 {
 public:
-  Engines_Container_i();
-  Engines_Container_i(CORBA::ORB_ptr orb, 
-                      PortableServer::POA_ptr poa,
-                      char * containerName ,
-                      int argc, char* argv[],
-                      bool activAndRegist = true,
-                      bool isServantAloneInProcess = true);
-  virtual ~Engines_Container_i();
+  Abstract_Engines_Container_i();
+  Abstract_Engines_Container_i(CORBA::ORB_ptr orb,
+                               PortableServer::POA_ptr poa,
+                               char *containerName,
+                               int argc, char *argv[],
+                               SALOME_NamingService_Container_Abstract *ns = nullptr,
+                               bool isServantAloneInProcess = true);
+  virtual ~Abstract_Engines_Container_i();
+
+  virtual bool isSSLMode() const = 0;
 
   // --- CORBA methods
 
-  virtual bool load_component_Library(const charcomponentName, CORBA::String_out reason);
+  virtual bool load_component_Library(const char *componentName, CORBA::String_out reason);
 
   virtual Engines::EngineComponent_ptr
-  create_component_instance( const char* componentName);
+  create_component_instance(const char *componentName);
+
+  void override_environment_python( const Engines::FieldsDict& env ) override;
+
+  void override_environment( const Engines::FieldsDict& env ) override;
+
+  Engines::FieldsDict *get_os_environment() override;
+  
+  void execute_python_code(const char *code) override;
 
   virtual Engines::EngineComponent_ptr
-  create_component_instance_env( const char* componentName,
-                                 const Engines::FieldsDict& env,
-                                 CORBA::String_out reason);
+  create_component_instance_env(const char *componentName,
+                                const Engines::FieldsDict &env,
+                                CORBA::String_out reason);
 
   virtual char *
-  create_python_service_instance(const charCompName,
+  create_python_service_instance(const char *CompName,
                                  CORBA::String_out reason);
 
   Engines::EngineComponent_ptr
-  find_component_instance( const char* registeredName);
+  find_component_instance(const char *registeredName);
 
   Engines::EngineComponent_ptr
-  load_impl(const char* nameToRegister,
-            const char* componentName);
+  load_impl(const char *nameToRegister,
+            const char *componentName);
+
+  bool is_SSL_mode() override { return this->isSSLMode(); }
+
+  Engines::EmbeddedNamingService_ptr get_embedded_NS_if_ssl() override;
 
+  SALOME_NamingService_Container_Abstract *getNS() const { return _NS; }
 
   void remove_impl(Engines::EngineComponent_ptr component_i);
   void finalize_removal();
 
   virtual void ping();
-  char* name();
-  char* workingdir();
-  char* logfilename();
-  void logfilename(const char* name);
+  CORBA::Long getNumberOfCPUCores();
+  Engines::vectorOfDouble* loadOfCPUCores();
+  void setPyScriptForCPULoad(const char *script);
+  void resetScriptForCPULoad();
+  CORBA::Long getTotalPhysicalMemory();
+  CORBA::Long getTotalPhysicalMemoryInUse();
+  CORBA::Long getTotalPhysicalMemoryInUseByMe();
+  char *name();
+  char *workingdir();
+  char *logfilename();
+  void logfilename(const char *name);
 
   virtual void Shutdown();
-  chargetHostName();
+  char *getHostName();
   CORBA::Long getPID();
   //! Kill current container
   bool Kill_impl();
 
-  Engines::fileRef_ptr createFileRef(const charorigFileName);
+  Engines::fileRef_ptr createFileRef(const char *origFileName);
   Engines::fileTransfer_ptr getFileTransfer();
 
-  virtual Engines::Salome_file_ptr createSalome_file(const charorigFileName);
-  void copyFile(Engines::Container_ptr container, const char* remoteFile, const char* localFile);
-  Engines::PyNode_ptr createPyNode(const char* nodeName, const char* code);
+  virtual Engines::Salome_file_ptr createSalome_file(const char *origFileName);
+  void copyFile(Engines::Container_ptr container, const char *remoteFile, const char *localFile);
+  Engines::PyNode_ptr createPyNode(const char *nodeName, const char *code);
   Engines::PyNode_ptr getDefaultPyNode(const char *nodeName);
-  Engines::PyScriptNode_ptr createPyScriptNode(const char* nodeName, const char* code);
+  Engines::PyScriptNode_ptr createPyScriptNode(const char *nodeName, const char *code);
+  void removePyScriptNode(const char *nodeName) override;
+  void cleanAllPyScripts() override;
   Engines::PyScriptNode_ptr getDefaultPyScriptNode(const char *nodeName);
   // --- local C++ methods
 
   Engines::EngineComponent_ptr
-  find_or_create_instance( std::string genericRegisterName,
-                           std::string componentLibraryName);
+  find_or_create_instance(std::string genericRegisterName,
+                          std::string componentLibraryName);
 
-  bool load_component_CppImplementation(const char* componentName,std::string& reason);
-  bool load_component_PythonImplementation(const char* componentName,std::string& reason);
-  bool load_component_ExecutableImplementation(const char* componentName,std::string& reason);
+  bool load_component_CppImplementation(const char *componentName, std::string &reason);
+  bool load_component_PythonImplementation(const char *componentName, std::string &reason);
+  bool load_component_ExecutableImplementation(const char *componentName, std::string &reason);
 
-  Engines::EngineComponent_ptr createPythonInstance(std::string CompName, std::stringerror);
-  Engines::EngineComponent_ptr createExecutableInstance(std::string CompName, const Engines::FieldsDict& env, std::string& error);
-  Engines::EngineComponent_ptr createInstance(std::string genericRegisterName, void *handle, std::stringerror);
+  Engines::EngineComponent_ptr createPythonInstance(std::string CompName, std::string &error);
+  Engines::EngineComponent_ptr createExecutableInstance(std::string CompName, const Engines::FieldsDict &env, std::string &error);
+  Engines::EngineComponent_ptr createInstance(std::string genericRegisterName, void *handle, std::string &error);
 
-  static bool isPythonContainer(const charContainerName);
+  static bool isPythonContainer(const char *ContainerName);
   static void decInstanceCnt(std::string genericRegisterName);
   //??? char* machineName();
 
@@ -135,40 +158,78 @@ public:
   int getArgc() { return _argc; }
   char **getArgv() { return _argv; }
 
-  void registerTemporaryFile( const std::string& fileName );
-  void unregisterTemporaryFile( const std::string& fileName );
+  void registerTemporaryFile(const std::string &fileName);
+  void unregisterTemporaryFile(const std::string &fileName);
   void clearTemporaryFiles();
+  PortableServer::ObjectId *getCORBAId() const { return _id; }
 
 protected:
-
   static std::map<std::string, int> _cntInstances_map;
-  static std::map<std::string, void *> _library_map; // library names, loaded
-  static std::map<std::string, void *> _toRemove_map;// library names to remove
-  static omni_mutex _numInstanceMutex ; // lib and instance protection
+  static std::map<std::string, void *> _library_map;  // library names, loaded
+  static std::map<std::string, void *> _toRemove_map; // library names to remove
+  static omni_mutex _numInstanceMutex;                // lib and instance protection
 
   bool _isSupervContainer;
 
-  SALOME_NamingService *_NS ;
+  SALOME_NamingService_Container_Abstract *_NS;
   std::string _library_path;
   std::string _containerName;
   std::string _logfilename;
+  std::string _load_script;
   CORBA::ORB_var _orb;
   PortableServer::POA_var _poa;
-  PortableServer::ObjectId * _id ;
-  int _numInstance ;
-  std::map<std::string,Engines::EngineComponent_var> _listInstances_map;
-  std::map<std::string,Engines::fileRef_var> _fileRef_map;
-  std::map<std::string,Engines::Salome_file_var> _Salome_file_map;
-  std::map<std::string,Engines::PyScriptNode_var> _dftPyScriptNode;
-  std::map<std::string,Engines::PyNode_var> _dftPyNode;
+  PortableServer::ObjectId *_id;
+  int _numInstance;
+  std::map<std::string, Engines::EngineComponent_var> _listInstances_map;
+  std::map<std::string, Engines::fileRef_var> _fileRef_map;
+  std::map<std::string, Engines::Salome_file_var> _Salome_file_map;
+  std::map<std::string, Engines::PyScriptNode_var> _dftPyScriptNode;
+  std::map<std::string, Engines::PyNode_var> _dftPyNode;
   Utils_Mutex _mutexForDftPy;
   std::list<std::string> _tmp_files;
   Engines::fileTransfer_var _fileTransfer;
 
-  int    _argc ;
-  char** _argv ;
-  long   _pid;
-  bool   _isServantAloneInProcess;
+  int _argc;
+  char **_argv;
+  long _pid;
+  bool _isServantAloneInProcess;
+};
+
+class CONTAINER_EXPORT Engines_Container_i : public Abstract_Engines_Container_i
+{
+public:
+  Engines_Container_i() { }
+  Engines_Container_i(CORBA::ORB_ptr orb,
+                      PortableServer::POA_ptr poa,
+                      char *containerName,
+                      int argc, char *argv[],
+                      SALOME_NamingService_Container_Abstract *ns = nullptr,
+                      bool isServantAloneInProcess = true) :
+                      Abstract_Engines_Container_i(orb, poa, containerName, argc, argv, ns, isServantAloneInProcess) {}
+  bool isSSLMode() const override { return false; }
+};
+
+class CONTAINER_EXPORT Engines_Container_SSL_i : public Abstract_Engines_Container_i
+{
+public:
+  Engines_Container_SSL_i();
+  Engines_Container_SSL_i(CORBA::ORB_ptr orb,
+                          PortableServer::POA_ptr poa,
+                          char *containerName,
+                          int argc, char *argv[],
+                          SALOME_NamingService_Container_Abstract *ns = nullptr,
+                          bool isServantAloneInProcess = true) :
+                          Abstract_Engines_Container_i(orb, poa, containerName, argc, argv, ns, isServantAloneInProcess) {}
+    bool isSSLMode() const override { return true; }
 };
 
+/*!
+ * Methods to be used in SSL mode to skip NS.
+ */
+namespace KERNEL
+{
+  CONTAINER_EXPORT Engines_Container_SSL_i *getContainerSA();
+  CONTAINER_EXPORT Engines::Container_var getContainerRefSA();
+} // namespace KERNEL
+
 #endif