-// Copyright (C) 2009-2012 CEA/DEN, EDF R&D
+// Copyright (C) 2009-2024 CEA, EDF
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
-// version 2.1 of the License.
+// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
BL::SALOMEServices::initNS()
{
bool return_value = true;
- _salome_naming_service = new SALOME_NamingService(_orb);
+ _salome_naming_service = new BL_SALOME_NamingService_Wrapper(_orb);
_lcc = new SALOME_LifeCycleCORBA(_salome_naming_service);
CORBA::Object_var obj = _salome_naming_service->Resolve("/SalomeLauncher");
_salome_launcher = Engines::SalomeLauncher::_narrow(obj);
}
std::list<std::string>
-BL::SALOMEServices::getResourceList()
+BL::SALOMEServices::getResourceList(bool batch_only)
{
std::list<std::string> resource_list;
{
Engines::ResourceParameters params;
_lcc->preSet(params);
+ params.can_launch_batch_jobs = batch_only;
Engines::ResourceList * resourceList = NULL;
try
{
resource_descr.batch = resource_definition->batch.in();
resource_descr.mpiImpl = resource_definition->mpiImpl.in();
resource_descr.iprotocol = resource_definition->iprotocol.in();
- resource_descr.is_cluster_head = resource_definition->is_cluster_head;
+ resource_descr.can_launch_batch_jobs = resource_definition->can_launch_batch_jobs;
+ resource_descr.can_run_containers = resource_definition->can_run_containers;
resource_descr.working_directory = resource_definition->working_directory.in();
delete resource_definition;
resource_definition->name = CORBA::string_dup(new_resource.name.c_str());
resource_definition->hostname = CORBA::string_dup(new_resource.hostname.c_str());
+ if (new_resource.batch == "none")
+ resource_definition->type = CORBA::string_dup("single_machine");
+ else
+ resource_definition->type = CORBA::string_dup("cluster");
resource_definition->protocol = CORBA::string_dup(new_resource.protocol.c_str());
resource_definition->username = CORBA::string_dup(new_resource.username.c_str());
resource_definition->applipath = CORBA::string_dup(new_resource.applipath.c_str());
resource_definition->batch = CORBA::string_dup(new_resource.batch.c_str());
resource_definition->mpiImpl = CORBA::string_dup(new_resource.mpiImpl.c_str());
resource_definition->iprotocol = CORBA::string_dup(new_resource.iprotocol.c_str());
- resource_definition->is_cluster_head = new_resource.is_cluster_head;
+ resource_definition->can_launch_batch_jobs = new_resource.can_launch_batch_jobs;
+ resource_definition->can_run_containers = new_resource.can_run_containers;
resource_definition->working_directory = CORBA::string_dup(new_resource.working_directory.c_str());
try
catch (const SALOME::SALOME_Exception & ex)
{
DEBMSG("SALOME Exception in addResource ! " << ex.details.text.in());
+ throw(BL::Exception(ex.details.text.in()));
}
catch (const CORBA::SystemException& ex)
{
DEBMSG("Receive SALOME System Exception: " << ex);
DEBMSG("Check SALOME servers...");
+ throw(BL::Exception("SALOME System Exception"));
}
}
{
job_parameters->job_type = CORBA::string_dup("python_salome");
}
+ else if (job->getType() == BL::Job::COMMAND_SALOME)
+ {
+ job_parameters->job_type = CORBA::string_dup("command_salome");
+ }
// Specific parameters
if (job->getType() == BL::Job::YACS_SCHEMA)
new_parameter->value = CORBA::string_dup(oss.str().c_str());
job_parameters->specific_parameters[job_parameters->specific_parameters.length() - 1] = new_parameter;
}
+ if (!job->getYacsDriverOptions().empty())
+ {
+ job_parameters->specific_parameters.length(job_parameters->specific_parameters.length() + 1);
+ Engines::Parameter_var new_parameter = new Engines::Parameter;
+ new_parameter->name = CORBA::string_dup("YACSDriverOptions");
+ new_parameter->value = CORBA::string_dup(job->getYacsDriverOptions().c_str());
+ job_parameters->specific_parameters[job_parameters->specific_parameters.length() - 1] = new_parameter;
+ }
}
if (job->getLoadLevelerJobType() != "")
{
job_parameters->job_name = CORBA::string_dup(job->getName().c_str());
job_parameters->job_file = CORBA::string_dup(job->getJobFile().c_str());
job_parameters->env_file = CORBA::string_dup(job->getEnvFile().c_str());
+ job_parameters->pre_command = CORBA::string_dup(job->getPreCommand().c_str());
BL::Job::FilesParam files = job->getFilesParameters();
std::list<std::string>::iterator it;
int i = 0;
job_parameters->maximum_duration = CORBA::string_dup(cpp_batch_params.maximum_duration.c_str());
job_parameters->resource_required.name = CORBA::string_dup(job->getResource().c_str());
job_parameters->resource_required.nb_proc = cpp_batch_params.nb_proc;
+ job_parameters->resource_required.nb_node = cpp_batch_params.nb_node;
job_parameters->queue = CORBA::string_dup(job->getBatchQueue().c_str());
+ job_parameters->partition = CORBA::string_dup(job->getBatchPartition().c_str());
+ job_parameters->exclusive = cpp_batch_params.exclusive;
+ job_parameters->wckey = CORBA::string_dup(job->getWCKey().c_str());
+ job_parameters->extra_params = CORBA::string_dup(job->getExtraParams().c_str());
// Memory
- CORBA::Long memory;
- std::string ram = cpp_batch_params.expected_memory.substr(0,cpp_batch_params.expected_memory.size()-2);
- std::istringstream iss(ram);
- iss >> memory;
- std::string unity = cpp_batch_params.expected_memory.substr(cpp_batch_params.expected_memory.size()-2, 2);
- if((unity.find("gb") != std::string::npos))
- memory = memory * 1024;
- job_parameters->resource_required.mem_mb = memory;
+ switch (cpp_batch_params.mem_req_type)
+ {
+ case BL::Job::MEM_PER_NODE:
+ job_parameters->resource_required.mem_mb = cpp_batch_params.mem_limit;
+ job_parameters->mem_per_cpu = 0;
+ break;
+ case BL::Job::MEM_PER_CPU:
+ job_parameters->resource_required.mem_mb = 0;
+ job_parameters->mem_per_cpu = cpp_batch_params.mem_limit;
+ break;
+ default:
+ throw Exception("Unknown memory requirement, unable to create job");
+ }
+
+ // Unused parameters
+ job_parameters->resource_required.nb_proc_per_node = -1;
+ job_parameters->resource_required.cpu_clock = -1;
+
+ // Parameters for COORM
+ job_parameters->launcher_file = CORBA::string_dup(cpp_batch_params.launcher_file.c_str());
+ job_parameters->launcher_args = CORBA::string_dup(cpp_batch_params.launcher_args.c_str());
// Create Job
try
return ret;
}
+// Get names or ids of hosts assigned to the job
+std::string
+BL::SALOMEServices::get_assigned_hostnames(BL::Job * job)
+{
+ std::string ret = "";
+
+ try
+ {
+ CORBA::String_var result = _salome_launcher->getAssignedHostnames(job->getSalomeLauncherId());
+ ret = result.in();
+ }
+ catch (const SALOME::SALOME_Exception & ex)
+ {
+ DEBMSG("SALOME Exception in get_assigned_hostnames !");
+ ret = ex.details.text.in();
+ }
+ catch (const CORBA::SystemException& ex)
+ {
+ DEBMSG("Receive SALOME System Exception: " << ex);
+ DEBMSG("Check SALOME servers...");
+ ret = "SALOME System Exception - see logs";
+ }
+ return ret;
+}
+
std::string
BL::SALOMEServices::save_jobs(const std::string & file_name)
{
job_return->setType(job_parameters->job_type.in());
job_return->setJobFile(job_parameters->job_file.in());
job_return->setEnvFile(job_parameters->env_file.in());
+ job_return->setPreCommand(job_parameters->pre_command.in());
job_return->setBatchQueue(job_parameters->queue.in());
+ job_return->setBatchPartition(job_parameters->partition.in());
+ job_return->setWCKey(job_parameters->wckey.in());
+ job_return->setExtraParams(job_parameters->extra_params.in());
BL::Job::FilesParam param;
param.result_directory = job_parameters->result_directory.in();
batch_param.batch_directory = job_parameters->work_directory.in();
batch_param.maximum_duration = job_parameters->maximum_duration.in();
batch_param.nb_proc = job_parameters->resource_required.nb_proc;
- std::ostringstream mem_stream;
- mem_stream << job_parameters->resource_required.mem_mb << "mb";
- batch_param.expected_memory = mem_stream.str();
+ batch_param.nb_node = job_parameters->resource_required.nb_node;
+ batch_param.exclusive = job_parameters->exclusive;
+
+ if (job_parameters->mem_per_cpu != 0)
+ {
+ batch_param.mem_limit = job_parameters->mem_per_cpu;
+ batch_param.mem_req_type = BL::Job::MEM_PER_CPU;
+ }
+ else
+ {
+ batch_param.mem_limit = job_parameters->resource_required.mem_mb;
+ batch_param.mem_req_type = BL::Job::MEM_PER_NODE;
+ }
+
+ // Parameters for COORM
+ batch_param.launcher_file = job_parameters->launcher_file.in();
+ batch_param.launcher_args = job_parameters->launcher_args.in();
+
job_return->setBatchParameters(batch_param);
job_return->setResource(job_parameters->resource_required.name.in());
iss >> value;
job_return->setDumpYACSState(value);
}
+ if (std::string(job_parameters->specific_parameters[i].name.in()) == "YACSDriverOptions")
+ {
+ std::string user_value = job_parameters->specific_parameters[i].value.in();
+ job_return->setYacsDriverOptions(user_value);
+ }
if (std::string(job_parameters->specific_parameters[i].name.in()) == "LoalLevelerJobType")
{
std::string user_value = job_parameters->specific_parameters[i].value.in();