]> SALOME platform Git repositories - modules/kernel.git/blob - src/Launcher/Launcher_Job.cxx
Salome HOME
if USER env variable is not defined, search for LOGNAME
[modules/kernel.git] / src / Launcher / Launcher_Job.cxx
1 // Copyright (C) 2009-2015  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
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, or (at your option) any later version.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 // Author: André RIBES - EDF R&D
21 //
22 //#define _DEBUG_
23 #include "Launcher_Job.hxx"
24 #include "Launcher.hxx"
25
26 #ifdef WITH_LIBBATCH
27 #include <libbatch/Constants.hxx>
28 #endif
29
30 using namespace std;
31
32 Launcher::Job::Job()
33 {
34   _number = -1;
35   _state = "CREATED";
36   _launch_date = getLaunchDate();
37
38   _env_file = "";
39   _job_name = "";
40   _job_file = "";
41   _job_file_name = "";
42   _job_file_name_complete = "";
43   _work_directory = "";
44   _local_directory = "";
45   _result_directory = "";
46   _maximum_duration = "";
47   _maximum_duration_in_second = -1;
48   _queue = "";
49   _job_type = "";
50   _exclusive = false;
51   _mem_per_cpu = 0;
52
53   // Parameters for COORM
54   _launcher_file = "";
55   _launcher_args = "";
56
57 #ifdef WITH_LIBBATCH
58   _batch_job = new Batch::Job();
59 #endif
60 }
61
62 Launcher::Job::~Job()
63 {
64   LAUNCHER_MESSAGE("Deleting job number: " << _number);
65 #ifdef WITH_LIBBATCH
66   if (_batch_job)
67     delete _batch_job;
68 #endif
69 }
70
71 void
72 Launcher::Job::stopJob()
73 {
74   LAUNCHER_MESSAGE("Stop resquested for job number: " << _number);
75   setState("FAILED");
76 #ifdef WITH_LIBBATCH
77   if (_batch_job_id.getReference() != "undefined")
78   {
79     try
80     {
81       _batch_job_id.deleteJob();
82     }
83     catch (const Batch::GenericException &ex)
84     {
85       LAUNCHER_INFOS("WARNING: exception when stopping the job: " << ex.message);
86     }
87   }
88 #endif
89 }
90
91
92 void
93 Launcher::Job::removeJob()
94 {
95   LAUNCHER_MESSAGE("Removing job number: " << _number);
96 #ifdef WITH_LIBBATCH
97   if (_batch_job_id.getReference() != "undefined")
98   {
99     try
100     {
101       _batch_job_id.deleteJob();
102     }
103     catch (const Batch::GenericException &ex)
104     {
105       LAUNCHER_INFOS("WARNING: exception when removing the job: " << ex.message);
106     }
107   }
108 #endif
109 }
110
111 std::string
112 Launcher::Job::getJobType() const
113 {
114   return _job_type;
115 }
116
117 void
118 Launcher::Job::setJobName(const std::string & job_name)
119 {
120   _job_name = job_name;
121 }
122
123 std::string
124 Launcher::Job::getJobName() const
125 {
126   return _job_name;
127 }
128
129 void
130 Launcher::Job::setState(const std::string & state)
131 {
132   // State of a Job: CREATED, QUEUED, RUNNING, FINISHED, FAILED
133   if (state != "CREATED" &&
134       state != "IN_PROCESS" &&
135       state != "QUEUED" &&
136       state != "RUNNING" &&
137       state != "PAUSED" &&
138       state != "FINISHED" &&
139       state != "FAILED" &&
140       state != "ERROR")
141   {
142     throw LauncherException("Bad state, this state does not exist: " + state);
143   }
144   _state = state;
145 }
146
147 std::string
148 Launcher::Job::getState() const
149 {
150   return _state;
151 }
152
153 // Get names or ids of hosts assigned to the job
154 std::string
155 Launcher::Job::getAssignedHostnames()
156 {
157   return _assigned_hostnames;
158 }
159
160 void
161 Launcher::Job::setNumber(const int & number)
162 {
163   if (_number != -1)
164     std::cerr << "Launcher::Job::setNumber -- Job number was already defined, before: " << _number << " now: " << number << std::endl;
165   _number = number;
166 }
167
168 int
169 Launcher::Job::getNumber()
170 {
171   return _number;
172 }
173
174 void
175 Launcher::Job::setResourceDefinition(const ParserResourcesType & resource_definition)
176 {
177   // Check machine_definition
178   std::string user_name = "";
179   if (resource_definition.UserName == "")
180   {
181     user_name = getenv("USER");
182     if (user_name == "")
183       user_name = getenv("LOGNAME");
184     if (user_name == "")
185     {
186       std::string mess = "You must define a user name: into your resource description or with one of env variables USER/LOGNAME";
187       throw LauncherException(mess);
188     }
189   }
190   else
191     user_name = resource_definition.UserName;
192
193   _resource_definition = resource_definition;
194   _resource_definition.UserName = user_name;
195 }
196
197 ParserResourcesType
198 Launcher::Job::getResourceDefinition() const
199 {
200   return _resource_definition;
201 }
202
203 void
204 Launcher::Job::setJobFile(const std::string & job_file)
205 {
206   // Check job file
207   if (job_file == "")
208   {
209     std::string mess = "Empty Job File is forbidden !";
210     throw LauncherException(mess);
211   }
212
213   _job_file = job_file;
214   std::string::size_type p1 = _job_file.find_last_of("/");
215   std::string::size_type p2 = _job_file.find_last_of(".");
216   _job_file_name_complete = _job_file.substr(p1+1);
217   _job_file_name = _job_file.substr(p1+1,p2-p1-1);
218 }
219
220 std::string
221 Launcher::Job::getJobFile() const
222 {
223   return _job_file;
224 }
225 void
226 Launcher::Job::setEnvFile(const std::string & env_file)
227 {
228   _env_file = env_file;
229 }
230
231 std::string
232 Launcher::Job::getEnvFile() const
233 {
234   return _env_file;
235 }
236
237 void
238 Launcher::Job::setWorkDirectory(const std::string & work_directory)
239 {
240   _work_directory = work_directory;
241 }
242
243 void
244 Launcher::Job::setLocalDirectory(const std::string & local_directory)
245 {
246   _local_directory = local_directory;
247 }
248
249 void
250 Launcher::Job::setResultDirectory(const std::string & result_directory)
251 {
252   _result_directory = result_directory;
253 }
254
255 void
256 Launcher::Job::add_in_file(const std::string & file)
257 {
258   std::list<std::string>::iterator it = std::find(_in_files.begin(), _in_files.end(), file);
259   if (it == _in_files.end())
260     _in_files.push_back(file);
261   else
262     std::cerr << "Launcher::Job::add_in_file -- Warning file was already entered in in_files: " << file << std::endl;
263 }
264
265 void
266 Launcher::Job::add_out_file(const std::string & file)
267 {
268   std::list<std::string>::iterator it = std::find(_out_files.begin(), _out_files.end(), file);
269   if (it == _out_files.end())
270     _out_files.push_back(file);
271   else
272     std::cerr << "Launcher::Job::add_out_file -- Warning file was already entered in out_files: " << file << std::endl;
273 }
274
275 void
276 Launcher::Job::setMaximumDuration(const std::string & maximum_duration)
277 {
278   checkMaximumDuration(maximum_duration);
279   _maximum_duration_in_second = convertMaximumDuration(maximum_duration);
280   _maximum_duration = maximum_duration;
281 }
282
283 // For COORM
284 void
285 Launcher::Job::setLauncherFile(const std::string & launcher_file)
286 {
287         _launcher_file = launcher_file;
288 }
289 void
290 Launcher::Job::setLauncherArgs(const std::string & launcher_args)
291 {
292         _launcher_args = launcher_args;
293 }
294
295 void
296 Launcher::Job::setResourceRequiredParams(const resourceParams & resource_required_params)
297 {
298   checkResourceRequiredParams(resource_required_params);
299   _resource_required_params = resource_required_params;
300 }
301
302 void
303 Launcher::Job::setQueue(const std::string & queue)
304 {
305   _queue = queue;
306 }
307
308 void
309 Launcher::Job::setExclusive(bool exclusive)
310 {
311   _exclusive = exclusive;
312 }
313
314 void
315 Launcher::Job::setExclusiveStr(const std::string & exclusiveStr)
316 {
317   if (exclusiveStr == "true")
318     _exclusive = true;
319   else if (exclusiveStr == "false")
320     _exclusive = false;
321   else
322     throw LauncherException(std::string("Invalid boolean value for exclusive: ") + exclusiveStr);
323 }
324
325 void
326 Launcher::Job::setMemPerCpu(unsigned long mem_per_cpu)
327 {
328   _mem_per_cpu = mem_per_cpu;
329 }
330
331 void
332 Launcher::Job::setWCKey(const std::string & wckey)
333 {
334   _wckey = wckey;
335 }
336
337 void
338 Launcher::Job::setExtraParams(const std::string & extra_params)
339 {
340   _extra_params = extra_params;
341 }
342
343 void
344 Launcher::Job::setReference(const std::string & reference)
345 {
346   _reference = reference;
347 }
348
349 std::string
350 Launcher::Job::getWorkDirectory() const
351 {
352   return _work_directory;
353 }
354
355 std::string
356 Launcher::Job::getLocalDirectory() const
357 {
358   return _local_directory;
359 }
360
361 std::string
362 Launcher::Job::getResultDirectory() const
363 {
364   return _result_directory;
365 }
366
367 const std::list<std::string> &
368 Launcher::Job::get_in_files() const
369 {
370   return _in_files;
371 }
372
373 const std::list<std::string> &
374 Launcher::Job::get_out_files() const
375 {
376   return _out_files;
377 }
378
379 std::string
380 Launcher::Job::getMaximumDuration() const
381 {
382   return _maximum_duration;
383 }
384
385 // For COORM
386 std::string
387 Launcher::Job::getLauncherFile() const
388 {
389         return _launcher_file;
390 }
391 std::string
392 Launcher::Job::getLauncherArgs() const
393 {
394         return _launcher_args;
395 }
396
397 resourceParams
398 Launcher::Job::getResourceRequiredParams() const
399 {
400   return _resource_required_params;
401 }
402
403 std::string
404 Launcher::Job::getQueue() const
405 {
406   return _queue;
407 }
408
409 bool
410 Launcher::Job::getExclusive() const
411 {
412   return _exclusive;
413 }
414
415 std::string
416 Launcher::Job::getExclusiveStr() const
417 {
418   return _exclusive ? "true" : "false";
419 }
420
421 unsigned long
422 Launcher::Job::getMemPerCpu() const
423 {
424   return _mem_per_cpu;
425 }
426
427 std::string
428 Launcher::Job::getWCKey() const
429 {
430   return _wckey;
431 }
432
433 std::string
434 Launcher::Job::getExtraParams() const
435 {
436   return _extra_params;
437 }
438
439 std::string
440 Launcher::Job::getReference() const
441 {
442   return _reference;
443 }
444
445 void
446 Launcher::Job::checkMaximumDuration(const std::string & maximum_duration)
447 {
448   std::string result("");
449   std::string edt_value = maximum_duration;
450   std::size_t pos = edt_value.find(":");
451
452   if (edt_value != "") {
453     if (pos == edt_value.npos) {
454       throw LauncherException("[Launcher::Job::checkMaximumDuration] Error on definition: " + edt_value);
455     }
456     std::string begin_edt_value = edt_value.substr(0, pos);
457     std::string mid_edt_value = edt_value.substr(pos, 1);
458     std::string end_edt_value = edt_value.substr(pos + 1, edt_value.npos);
459
460     long value;
461     std::istringstream iss(begin_edt_value);
462     if (!(iss >> value)) {
463       result = "[Launcher::Job::checkExpectedDuration] Error on definition ! : " + edt_value;
464     }
465     else if (value < 0) {
466       result = "[Launcher::Job::checkExpectedDuration] Error on definition time is negative ! : " + value;
467     }
468     std::istringstream iss_2(end_edt_value);
469     if (!(iss_2 >> value)) {
470       result = "[Launcher::Job::checkExpectedDuration] Error on definition ! : " + edt_value;
471     }
472     else if (value < 0) {
473       result = "[Launcher::Job::checkExpectedDuration] Error on definition time is negative ! : " + value;
474     }
475     if (mid_edt_value != ":") {
476       result = "[Launcher::Job::checkExpectedDuration] Error on definition ! :" + edt_value;
477     }
478   }
479   if (result != "")
480     throw LauncherException(result);
481 }
482
483 void
484 Launcher::Job::checkResourceRequiredParams(const resourceParams & resource_required_params)
485 {
486   // nb_proc has be to > 0
487   if (resource_required_params.nb_proc <= 0)
488   {
489     std::string message("[Launcher::Job::checkResourceRequiredParams] proc number is not > 0 ! ");
490     throw LauncherException(message);
491   }
492 }
493
494 long
495 Launcher::Job::convertMaximumDuration(const std::string & edt)
496 {
497   long hh, mm, ret;
498
499   if( edt.size() == 0 )
500     return -1;
501
502   std::string::size_type pos = edt.find(":");
503   std::string h = edt.substr(0,pos);
504   std::string m = edt.substr(pos+1,edt.size()-pos+1);
505   std::istringstream issh(h);
506   issh >> hh;
507   std::istringstream issm(m);
508   issm >> mm;
509   ret = hh*60 + mm;
510   ret = ret * 60;
511
512   return ret;
513 }
514
515 std::string
516 Launcher::Job::getLaunchDate() const
517 {
518   time_t rawtime;
519   time(&rawtime);
520   std::string launch_date = ctime(&rawtime);
521   int i = 0 ;
522   for (;i < launch_date.size(); i++)
523     if (launch_date[i] == '/' ||
524         launch_date[i] == '-' ||
525         launch_date[i] == ':' ||
526         launch_date[i] == ' ')
527       launch_date[i] = '_';
528   launch_date.erase(--launch_date.end()); // Last caracter is a \n
529
530   return launch_date;
531 }
532
533 std::string
534 Launcher::Job::updateJobState()
535 {
536
537   if (_state != "FINISHED" &&
538       _state != "ERROR"    &&
539       _state != "FAILED")
540   {
541 #ifdef WITH_LIBBATCH
542     if (_batch_job_id.getReference() != "undefined")
543     {
544       // A batch manager has been affected to the job
545       Batch::JobInfo job_info = _batch_job_id.queryJob();
546       Batch::Parametre par = job_info.getParametre();
547       _state = par[Batch::STATE].str();
548       _assigned_hostnames = (par.find(Batch::ASSIGNEDHOSTNAMES) == par.end())?
549                             "" : par[Batch::ASSIGNEDHOSTNAMES].str();
550       LAUNCHER_MESSAGE("State received is: " << par[Batch::STATE].str());
551     }
552 #endif
553   }
554   return _state;
555 }
556
557 #ifdef WITH_LIBBATCH
558 Batch::Job *
559 Launcher::Job::getBatchJob()
560 {
561   update_job();
562   return _batch_job;
563 }
564
565 Batch::Parametre
566 Launcher::Job::common_job_params()
567 {
568   Batch::Parametre params;
569
570   params[Batch::NAME] = getJobName();
571   params[Batch::NBPROC] = _resource_required_params.nb_proc;
572   params[Batch::NBPROCPERNODE] = _resource_required_params.nb_proc_per_node;
573
574   // Memory in megabytes
575   if (_resource_required_params.mem_mb > 0)
576   {
577     params[Batch::MAXRAMSIZE] = _resource_required_params.mem_mb;
578   }
579   else if (_mem_per_cpu > 0)
580   {
581     params[Batch::MEMPERCPU] = (long)_mem_per_cpu;
582   }
583
584   // We define a default directory
585   if (_work_directory == "")
586   {
587     const size_t BUFSIZE = 32;
588     char date[BUFSIZE];
589     time_t curtime = time(NULL);
590     strftime(date, BUFSIZE, "%Y_%m_%d__%H_%M_%S", localtime(&curtime));
591     if(!_resource_definition.working_directory.empty())
592     {
593       std::string job_dir = std::string("/job_") + date;
594       _work_directory = _resource_definition.working_directory + job_dir;
595     }
596     else
597     {
598       _work_directory = std::string("/$HOME/Batch/workdir_");
599       _work_directory += date;
600     }
601   }
602   params[Batch::WORKDIR] = _work_directory;
603
604   // Parameters for COORM
605   params[Batch::LAUNCHER_FILE] = _launcher_file;
606   params[Batch::LAUNCHER_ARGS] = _launcher_args;
607
608   // If result_directory is not defined, we use HOME environnement
609   if (_result_directory == "")
610     _result_directory = getenv("HOME");
611
612   // _in_files
613   std::list<std::string> in_files(_in_files);
614   in_files.push_back(_job_file);
615   if (_env_file != "")
616           in_files.push_back(_env_file);
617   for(std::list<std::string>::iterator it = in_files.begin(); it != in_files.end(); it++)
618   {
619     std::string file = *it;
620
621     // local file -> If file is not an absolute path, we apply _local_directory
622     std::string local_file;
623     if (file.substr(0, 1) == std::string("/"))
624       local_file = file;
625     else
626 #ifndef WIN32
627       local_file = _local_directory + "/" + file;
628 #else
629       local_file = file;
630 #endif
631
632     // remote file -> get only file name from in_files
633     size_t found = file.find_last_of("/");
634     std::string remote_file = _work_directory + "/" + file.substr(found+1);
635
636     params[Batch::INFILE] += Batch::Couple(local_file, remote_file);
637   }
638
639   // _out_files
640   for(std::list<std::string>::iterator it = _out_files.begin(); it != _out_files.end(); it++)
641   {
642     std::string file = *it;
643     // remote file -> If file is not an absolute path, we apply _work_directory
644     std::string remote_file;
645     std::string local_file;
646     if (file.substr(0, 1) == std::string("/"))
647     {
648       remote_file = file;
649       size_t found = file.find_last_of("/");
650       local_file = file.substr(found+1);
651     }
652     else
653     {
654       remote_file = _work_directory + "/" + file;
655       local_file = file;
656     }
657
658     params[Batch::OUTFILE] += Batch::Couple(local_file, remote_file);
659   }
660
661   // Time
662   if (_maximum_duration_in_second != -1)
663     params[Batch::MAXWALLTIME] = _maximum_duration_in_second / 60;
664
665   // Queue
666   if (_queue != "")
667     params[Batch::QUEUE] = _queue;
668
669   // Exclusive
670   if (getExclusive())
671     params[Batch::EXCLUSIVE] = true;
672
673   // WC Key
674   if (_wckey != "")
675     params[Batch::WCKEY] = _wckey;
676
677   // Extra params
678   if (_extra_params != "")
679     params[Batch::EXTRAPARAMS] = _extra_params;
680
681   // Specific parameters
682   std::map<std::string, std::string>::iterator it = _specific_parameters.find("LoalLevelerJobType");
683   if (it != _specific_parameters.end())
684     params["LL_JOBTYPE"] = it->second;
685   return params;
686 }
687
688 void
689 Launcher::Job::setBatchManagerJobId(Batch::JobId batch_manager_job_id)
690 {
691   _batch_job_id = batch_manager_job_id;
692 }
693
694 Batch::JobId
695 Launcher::Job::getBatchManagerJobId() const
696 {
697   return _batch_job_id;
698 }
699 #endif
700
701 void
702 Launcher::Job::addSpecificParameter(const std::string & name,
703                                       const std::string & value)
704 {
705   _specific_parameters[name] = value;
706 }
707
708 const std::map<std::string, std::string> &
709 Launcher::Job::getSpecificParameters() const
710 {
711   return _specific_parameters;
712 }
713
714 void
715 Launcher::Job::checkSpecificParameters()
716 {
717 }