1 // Copyright (C) 2009 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 // Author: André RIBES - EDF R&D
21 #include "Launcher_Job_SALOME.hxx"
23 Launcher::Job_SALOME::Job_SALOME() {}
25 Launcher::Job_SALOME::~Job_SALOME() {}
28 Launcher::Job_SALOME::setResourceDefinition(const ParserResourcesType & resource_definition)
30 // Check resource_definition
31 if (resource_definition.AppliPath == "")
33 std::string mess = "Resource definition must define an application path !, resource name is: " + resource_definition.Name;
34 throw LauncherException(mess);
36 Launcher::Job::setResourceDefinition(resource_definition);
40 Launcher::Job_SALOME::update_job()
43 Batch::Parametre params = common_job_params();
44 params[Batch::EXECUTABLE] = buildSalomeScript(params);
45 _batch_job->setParametre(params);
51 Launcher::Job_SALOME::buildSalomeScript(Batch::Parametre params)
54 std::string work_directory = params[Batch::WORKDIR].str();
56 std::string launch_date_port_file = _launch_date;
57 std::string launch_script = "/tmp/runSalome_" + _job_file_name + "_" + _launch_date + ".sh";
58 std::ofstream launch_script_stream;
59 launch_script_stream.open(launch_script.c_str(), std::ofstream::out);
62 launch_script_stream << "#! /bin/sh -f" << std::endl;
63 launch_script_stream << "cd " << work_directory << std::endl;
64 launch_script_stream << "export PYTHONPATH=" << work_directory << ":$PYTHONPATH" << std::endl;
65 launch_script_stream << "export PATH=" << work_directory << ":$PATH" << std::endl;
68 std::string::size_type last = _env_file.find_last_of("/");
69 launch_script_stream << "source ./" << _env_file.substr(last+1) << std::endl;
71 launch_script_stream << "export SALOME_TMP_DIR=" << work_directory << "/logs" << std::endl;
73 // -- Generates Catalog Resources
74 std::string resource_protocol = "ssh";
75 if (_resource_definition.ClusterInternalProtocol == rsh)
76 resource_protocol = "rsh";
78 launch_script_stream << "if [ \"x$LIBBATCH_NODEFILE\" != \"x\" ]; then " << std::endl;
79 launch_script_stream << "CATALOG_FILE=" << work_directory << "/CatalogResources_" << _launch_date << ".xml" << std::endl;
80 launch_script_stream << "export USER_CATALOG_RESOURCES_FILE=" << "$CATALOG_FILE" << std::endl;
81 launch_script_stream << "echo '<!DOCTYPE ResourcesCatalog>' > $CATALOG_FILE" << std::endl;
82 launch_script_stream << "echo '<resources>' >> $CATALOG_FILE" << std::endl;
83 launch_script_stream << "cat $LIBBATCH_NODEFILE | sort -u | while read host" << std::endl;
84 launch_script_stream << "do" << std::endl;
85 launch_script_stream << "echo '<machine hostname='\\\"$host\\\" >> $CATALOG_FILE" << std::endl;
86 launch_script_stream << "echo ' protocol=\"" << resource_protocol << "\"' >> $CATALOG_FILE" << std::endl;
87 launch_script_stream << "echo ' userName=\"" << _resource_definition.UserName << "\"' >> $CATALOG_FILE" << std::endl;
88 launch_script_stream << "echo ' appliPath=\"" << _resource_definition.AppliPath << "\"' >> $CATALOG_FILE" << std::endl;
89 launch_script_stream << "echo '/>' >> $CATALOG_FILE" << std::endl;
90 launch_script_stream << "done" << std::endl;
91 launch_script_stream << "echo '</resources>' >> $CATALOG_FILE" << std::endl;
92 launch_script_stream << "fi" << std::endl;
94 // Launch SALOME with an appli
95 launch_script_stream << _resource_definition.AppliPath << "/runAppli --terminal --ns-port-log=" << launch_date_port_file << " > logs/salome_" << _launch_date << ".log 2>&1" << std::endl;
96 launch_script_stream << "current=0\n"
98 << "while ! test -f " << _resource_definition.AppliPath << "/" << launch_date_port_file << "\n"
101 << " let current=current+1\n"
102 << " if [ \"$current\" -eq \"$stop\" ] ; then\n"
103 << " echo Error Naming Service failed ! >&2\n"
107 << "appli_port=`cat " << _resource_definition.AppliPath << "/" << launch_date_port_file << "`\n";
109 // Call real job type
110 addJobTypeSpecificScript(launch_script_stream);
113 launch_script_stream << _resource_definition.AppliPath << "/runSession -p $appli_port shutdownSalome.py" << std::endl;
114 launch_script_stream << "sleep 10" << std::endl;
117 launch_script_stream.flush();
118 launch_script_stream.close();
119 chmod(launch_script.c_str(), 0x1ED);
120 return launch_script;