-// Copyright (C) 2009-2012 CEA/DEN, EDF R&D, OPEN CASCADE
+// Copyright (C) 2009-2021 CEA/DEN, EDF R&D, OPEN CASCADE
//
// 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
#include "Basics_DirUtils.hxx"
#ifdef WITH_LIBBATCH
-#include <Batch/Batch_Constants.hxx>
+#include <libbatch/Constants.hxx>
#endif
-#ifdef WNT
+#ifdef WIN32
#include <io.h>
#define _chmod chmod
+#include <process.h>
#endif
+#include <sys/stat.h>
+
+#include <sstream>
Launcher::Job_SALOME::Job_SALOME() {}
Launcher::Job_SALOME::~Job_SALOME() {}
-void
+void
Launcher::Job_SALOME::setResourceDefinition(const ParserResourcesType & resource_definition)
{
// Check resource_definition
#ifdef WITH_LIBBATCH
Batch::Parametre params = common_job_params();
params[Batch::EXECUTABLE] = buildSalomeScript(params);
- params[Batch::EXCLUSIVE] = true;
_batch_job->setParametre(params);
#endif
}
#ifdef WITH_LIBBATCH
-std::string
+std::string
Launcher::Job_SALOME::buildSalomeScript(Batch::Parametre params)
{
// parameters
std::string work_directory = params[Batch::WORKDIR].str();
-
- std::string launch_script = Kernel_Utils::GetTmpDir() + "runSalome_" + _job_file_name + "_" + _launch_date + ".sh";
+ std::ostringstream str_pid;
+#ifdef WIN32
+ str_pid << _getpid();
+#else
+ str_pid << ::getpid();
+#endif
+ struct stat statbuf;
+ // true if APPLI holds a salome file launcher (in state of an application directory)
+ bool is_launcher_file=stat(getenv("APPLI"), &statbuf) ==0 && S_ISREG(statbuf.st_mode);
+ std::string launch_tmp_dir = Kernel_Utils::GetTmpDir();
+ std::string launch_script = launch_tmp_dir + "runSalome_" + _job_file_name + "_" + _launch_date + "-" + str_pid.str() + ".sh";
std::ofstream launch_script_stream;
- launch_script_stream.open(launch_script.c_str(),
+ launch_script_stream.open(launch_script.c_str(),
std::ofstream::out
-#ifdef WIN32
+#ifdef WIN32
| std::ofstream::binary //rnv: to avoid CL+RF end of line on windows
#endif
);
-
+
// Begin of script
launch_script_stream << "#!/bin/sh -f" << std::endl;
launch_script_stream << "cd " << work_directory << std::endl;
+ // remove the exit code from any previous execution
+ launch_script_stream << "rm -f logs/exit_code.log" << std::endl;
+
launch_script_stream << "export PYTHONPATH=" << work_directory << ":$PYTHONPATH" << std::endl;
launch_script_stream << "export PATH=" << work_directory << ":$PATH" << std::endl;
if (_env_file != "")
launch_script_stream << "export SALOME_TMP_DIR=" << work_directory << "/logs" << std::endl;
// -- Generates Catalog Resources
- std::string resource_protocol = ParserResourcesType::protocolToString(_resource_definition.ClusterInternalProtocol);
- launch_script_stream << "if [ \"x$LIBBATCH_NODEFILE\" != \"x\" ]; then " << std::endl;
- launch_script_stream << "CATALOG_FILE=" << "CatalogResources_" << _launch_date << ".xml" << std::endl;
- launch_script_stream << "export USER_CATALOG_RESOURCES_FILE=" << "$CATALOG_FILE" << std::endl;
- launch_script_stream << "echo '<!DOCTYPE ResourcesCatalog>' > $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo '<resources>' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "cat $LIBBATCH_NODEFILE | sort | uniq -c | while read nbproc host" << std::endl;
- launch_script_stream << "do" << std::endl;
- launch_script_stream << "echo '<machine hostname='\\\"$host\\\" >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' protocol=\"" << resource_protocol << "\"' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' userName=\"" << _resource_definition.UserName << "\"' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' appliPath=\"" << _resource_definition.AppliPath << "\"' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' mpi=\"" << _resource_definition.PrintMpiImplType() << "\"' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' nbOfNodes='\\\"$nbproc\\\" >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo ' nbOfProcPerNode=\"1\"' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "echo '/>' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "done" << std::endl;
- launch_script_stream << "echo '</resources>' >> $CATALOG_FILE" << std::endl;
- launch_script_stream << "fi" << std::endl;
-
+ // do not build a new catalog if the job is launched on localhost
+ if(_resource_required_params.name != "localhost")
+ {
+ std::string resource_protocol = _resource_definition.getClusterInternalProtocolStr();
+ launch_script_stream << "if [ \"x$LIBBATCH_NODEFILE\" != \"x\" ]; then " << std::endl;
+ launch_script_stream << "CATALOG_FILE=" << "CatalogResources_" << _launch_date << ".xml" << std::endl;
+ launch_script_stream << "export USER_CATALOG_RESOURCES_FILE=" << "$CATALOG_FILE" << std::endl;
+ launch_script_stream << "{" << std::endl;
+ launch_script_stream << "echo '<!DOCTYPE ResourcesCatalog>'" << std::endl;
+ launch_script_stream << "echo '<resources>'" << std::endl;
+ launch_script_stream << "sort \"$LIBBATCH_NODEFILE\" | uniq -c | while read nbproc host" << std::endl;
+ launch_script_stream << "do" << std::endl;
+ // Full name doesn't work. eg: sagittaire-7 instead of sagittaire-7.lyon.grid5000.fr
+ launch_script_stream << "host_basename=$(echo \"$host\" | cut -f1 -d.)" << std::endl;
+ launch_script_stream << "echo '<machine name='\\\"\"$host_basename\"\\\"" << std::endl;
+ launch_script_stream << "echo ' hostname='\\\"\"$host_basename\"\\\"" << std::endl;
+ launch_script_stream << "echo ' type=\"single_machine\"'" << std::endl;
+ launch_script_stream << "echo ' protocol=\"" << resource_protocol << "\"'" << std::endl;
+ launch_script_stream << "echo ' userName=\"" << _resource_definition.UserName << "\"'" << std::endl;
+ launch_script_stream << "echo ' appliPath=\"" << _resource_definition.AppliPath << "\"'" << std::endl;
+ launch_script_stream << "echo ' mpi=\"" << _resource_definition.getMpiImplTypeStr() << "\"'" << std::endl;
+ launch_script_stream << "echo ' nbOfNodes='\\\"\"$nbproc\"\\\"" << std::endl;
+ launch_script_stream << "echo ' nbOfProcPerNode=\"1\"'" << std::endl;
+ launch_script_stream << "echo ' canRunContainers=\"true\"'" << std::endl;
+ launch_script_stream << "echo '/>'" << std::endl;
+ launch_script_stream << "done" << std::endl;
+ launch_script_stream << "echo '</resources>'" << std::endl;
+ launch_script_stream << "} > $CATALOG_FILE" << std::endl;
+ launch_script_stream << "fi" << std::endl;
+ }
// Create file for ns-port-log
- launch_script_stream << "NS_PORT_FILE_PATH=`mktemp " << _resource_definition.AppliPath << "/USERS/nsport_XXXXXX` &&\n";
- launch_script_stream << "NS_PORT_FILE_NAME=`basename $NS_PORT_FILE_PATH` &&\n";
+ if (is_launcher_file)
+ // for a salome application file, we write NS_PORT_FILE_PATH in working directory
+ launch_script_stream << "NS_PORT_FILE_PATH=$(mktemp " << work_directory << "/nsport_" << _launch_date << "_XXXXXX) &&\n";
+ else
+ launch_script_stream << "NS_PORT_FILE_PATH=$(mktemp " << _resource_definition.AppliPath << "/USERS/nsport_XXXXXX) &&\n";
// Launch SALOME with an appli
- launch_script_stream << _resource_definition.AppliPath << "/runAppli --terminal --ns-port-log=$NS_PORT_FILE_NAME --server-launch-mode=fork ";
+ if (is_launcher_file)
+ launch_script_stream << _resource_definition.AppliPath << " start --terminal --ns-port-log=\"$NS_PORT_FILE_PATH\" --server-launch-mode=fork ";
+ else
+ launch_script_stream << _resource_definition.AppliPath << "/salome start --terminal --ns-port-log=\"$NS_PORT_FILE_PATH\" --server-launch-mode=fork ";
launch_script_stream << "> logs/salome_" << _launch_date << ".log 2>&1 &&" << std::endl;
launch_script_stream << "current=0 &&\n"
<< "stop=20 &&\n"
- << "while ! test -s $NS_PORT_FILE_PATH\n"
+ << "while ! test -s \"$NS_PORT_FILE_PATH\"\n"
<< "do\n"
<< " sleep 2\n"
<< " current=$((current+1))\n"
<< " exit\n"
<< " fi\n"
<< "done &&\n"
- << "appli_port=`cat $NS_PORT_FILE_PATH` &&\n"
- << "rm $NS_PORT_FILE_PATH &&\n";
+ << "appli_port=$(cat \"$NS_PORT_FILE_PATH\") &&\n"
+ << "rm \"$NS_PORT_FILE_PATH\" &&\n";
// Call real job type
addJobTypeSpecificScript(launch_script_stream);
+ // log the exit code
+ launch_script_stream << "echo $? > logs/exit_code.log" << std::endl;
// End
- launch_script_stream << _resource_definition.AppliPath << "/runSession -p $appli_port shutdownSalome.py" << std::endl;
- launch_script_stream << "sleep 10" << std::endl;
+ if (is_launcher_file)
+ launch_script_stream << _resource_definition.AppliPath << " kill \"$appli_port\"" << std::endl;
+ else
+ launch_script_stream << _resource_definition.AppliPath << "/salome kill \"$appli_port\"" << std::endl;
// Return
launch_script_stream.flush();
return launch_script;
}
#endif
-