src/Launcher/Launcher_Job.hxx

   1 // Copyright (C) 2009-2014  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 #ifndef _LAUNCHER_JOB_HXX_
  23 #define _LAUNCHER_JOB_HXX_
  24
  25 #include "Launcher_Utils.hxx"
  26 #include "ResourcesManager.hxx"
  27
  28 #include <stdlib.h>
  29 #include <time.h>
  30 #include <sys/stat.h>
  31
  32 #include <string>
  33 #include <list>
  34 #include <map>
  35 #include <iostream>
  36 #include <sstream>
  37 #include <algorithm>
  38 #include <exception>
  39
  40 #ifdef WITH_LIBBATCH
  41 #include <libbatch/Job.hxx>
  42 #include <libbatch/JobId.hxx>
  43 #include <libbatch/GenericException.hxx>
  44 #endif
  45
  46 namespace Launcher
  47 {
  48   class LAUNCHER_EXPORT Job
  49   {
  50     public:
  51       Job();
  52       virtual ~Job();
  53
  54       // Launcher managing parameters
  55       // State of a Job: CREATED, IN_PROCESS, QUEUED, RUNNING, PAUSED, FINISHED, ERROR
  56       void setState(const std::string & state);
  57       std::string getState() const;
  58           // Get names or ids of hosts assigned to the job
  59       std::string getAssignedHostnames();
  60
  61       void setNumber(const int & number);
  62       int getNumber();
  63
  64       virtual void setResourceDefinition(const ParserResourcesType & resource_definition);
  65       ParserResourcesType getResourceDefinition() const;
  66
  67       // Common parameters
  68       void setJobName(const std::string & job_name);
  69       virtual void setJobFile(const std::string & job_file);
  70       void setWorkDirectory(const std::string & work_directory);
  71       void setLocalDirectory(const std::string & local_directory);
  72       void setResultDirectory(const std::string & result_directory);
  73       void add_in_file(const std::string & file);
  74       void add_out_file(const std::string & file);
  75       void setMaximumDuration(const std::string & maximum_duration);
  76       void setResourceRequiredParams(const resourceParams & resource_required_params);
  77       void setQueue(const std::string & queue);
  78       void setEnvFile(const std::string & env_file);
  79       void setExclusive(bool exclusive);
  80       void setExclusiveStr(const std::string & exclusiveStr);
  81       void setMemPerCpu(unsigned long mem_per_cpu);
  82       void setWCKey(const std::string & wckey);
  83       void setReference(const std::string & reference);
  84           // For COORM
  85           void setLauncherFile(const std::string & launcher_file);
  86           void setLauncherArgs(const std::string & launcher_args);
  87
  88       std::string getJobName() const;
  89       std::string getJobFile() const;
  90       std::string getWorkDirectory() const;
  91       std::string getLocalDirectory() const;
  92       std::string getResultDirectory() const;
  93       const std::list<std::string> & get_in_files() const;
  94       const std::list<std::string> & get_out_files() const;
  95       std::string getMaximumDuration() const;
  96       resourceParams getResourceRequiredParams() const;
  97       std::string getQueue() const;
  98       std::string getEnvFile() const;
  99       std::string getJobType() const;
 100       bool getExclusive() const;
 101       std::string getExclusiveStr() const;
 102       unsigned long getMemPerCpu() const;
 103       std::string getWCKey() const;
 104       std::string getReference() const;
 105
 106           // For COORM
 107           std::string getLauncherFile() const;
 108           std::string getLauncherArgs() const;
 109
 110       std::string updateJobState();
 111
 112       void addSpecificParameter(const std::string & name,
 113                                   const std::string & value);
 114       const std::map<std::string, std::string> & getSpecificParameters() const;
 115       virtual void checkSpecificParameters();
 116
 117       // Checks
 118       void checkMaximumDuration(const std::string & maximum_duration);
 119       void checkResourceRequiredParams(const resourceParams & resource_required_params);
 120
 121       // Helps
 122       long convertMaximumDuration(const std::string & maximum_duration);
 123       std::string getLaunchDate() const;
 124
 125       void stopJob();
 126       void removeJob();
 127
 128       // Abstract class
 129       virtual void update_job() = 0;
 130
 131     protected:
 132       int _number;
 133
 134       std::string _job_type;
 135
 136       std::string _state;
 137       std::string _assigned_hostnames; // Assigned hostnames
 138       std::string _launch_date;
 139       std::string _env_file;
 140
 141       ParserResourcesType _resource_definition;
 142
 143       std::string _job_name;
 144       std::string _job_file;
 145       std::string _job_file_name;
 146       std::string _job_file_name_complete;
 147
 148       std::string _work_directory;
 149       std::string _local_directory;
 150       std::string _result_directory;
 151       std::list<std::string> _in_files;
 152       std::list<std::string> _out_files;
 153       std::map<std::string, std::string> _specific_parameters;
 154       std::string _maximum_duration;
 155       long _maximum_duration_in_second;
 156       resourceParams _resource_required_params;
 157       std::string _queue;
 158       bool _exclusive;
 159       unsigned long _mem_per_cpu;
 160       std::string _wckey;
 161       std::string _reference; //! Reference of the job for the batch manager
 162
 163           // Parameters for COORM
 164           std::string _launcher_file;
 165           std::string _launcher_args;
 166
 167 #ifdef WITH_LIBBATCH
 168     // Connection with LIBBATCH
 169     public:
 170       Batch::Job * getBatchJob();
 171       Batch::Parametre common_job_params();
 172       void setBatchManagerJobId(Batch::JobId batch_manager_job_id);
 173       Batch::JobId getBatchManagerJobId() const;
 174
 175     protected:
 176       Batch::Job * _batch_job;
 177       Batch::JobId _batch_job_id;
 178 #endif
 179   };
 180 }
 181
 182 #endif