HTCondorComputeService.h
1 
10 #ifndef WRENCH_HTCONDOR_H
11 #define WRENCH_HTCONDOR_H
12 
13 #include "wrench/services/compute/ComputeServiceMessage.h"
14 #include "wrench/services/compute/ComputeService.h"
15 #include "wrench/services/compute/htcondor/HTCondorCentralManagerService.h"
16 #include "wrench/services/compute/htcondor/HTCondorComputeServiceProperty.h"
17 #include "wrench/services/compute/htcondor/HTCondorComputeServiceMessagePayload.h"
18 #include "wrench/workflow/job/PilotJob.h"
19 #include "wrench/workflow/job/StandardJob.h"
20 
21 namespace wrench {
22 
28  private:
29  std::map<std::string, std::string> default_property_values = {
32  };
33 
34  std::map<std::string, double> default_messagepayload_values = {
43  };
44 
45  public:
46  HTCondorComputeService(const std::string &hostname,
47  const std::string &pool_name,
48  std::set<ComputeService *> compute_resources,
49  std::map<std::string, std::string> property_list = {},
50  std::map<std::string, double> messagepayload_list = {});
51 
52  /***********************/
54  /***********************/
55 
57  const std::map<std::string, std::string> &service_specific_arguments) override;
58 
59  void submitPilotJob(PilotJob *job, const std::map<std::string, std::string> &service_specific_arguments) override;
60 
61  std::shared_ptr<StorageService> getLocalStorageService() const;
62 
63  void setLocalStorageService(std::shared_ptr<StorageService> local_storage_service);
64 
65  /***********************/
67  /***********************/
68 
69 
70  /***********************/
72  /***********************/
73 
74  ~HTCondorComputeService() override;
75 
76  void terminateStandardJob(StandardJob *job) override;
77 
78  void terminatePilotJob(PilotJob *job) override;
79 
80  /***********************/
82  /***********************/
83 
84  private:
85  int main() override;
86 
87  bool processNextMessage();
88 
89  void processSubmitStandardJob(const std::string &answer_mailbox, StandardJob *job,
90  const std::map<std::string, std::string> &service_specific_args);
91 
92  void processSubmitPilotJob(const std::string &answer_mailbox, PilotJob *job,
93  const std::map<std::string, std::string> &service_specific_args);
94 
95  void terminate();
96 
97  std::string pool_name;
98  std::shared_ptr<StorageService> local_storage_service;
99  std::shared_ptr<HTCondorCentralManagerService> central_manager;
100  };
101 }
102 
103 #endif //WRENCH_HTCONDOR_H
static const std::string SUBMIT_STANDARD_JOB_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to submit a standard job.
Definition: ComputeServiceMessagePayload.h:26
A workload management framework compute service.
Definition: HTCondorComputeService.h:27
static const std::string DAEMON_STOPPED_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to confirm it has terminated.
Definition: ServiceMessagePayload.h:33
static const std::string RESOURCE_DESCRIPTION_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to ask it for information on its resour...
Definition: ComputeServiceMessagePayload.h:56
The compute service base class.
Definition: ComputeService.h:35
void submitStandardJob(StandardJob *job, const std::map< std::string, std::string > &service_specific_arguments) override
Submit a standard job to the HTCondor service.
Definition: HTCondorComputeService.cpp:117
A standard (i.e., non-pilot) workflow job that can be submitted to a ComputeService by a WMS (via a J...
Definition: StandardJob.h:37
Definition: Alarm.cpp:20
void setLocalStorageService(std::shared_ptr< StorageService > local_storage_service)
Set the service's local storage service.
Definition: HTCondorComputeService.cpp:233
static const std::string SUBMIT_PILOT_JOB_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent from the daemon to acknowledge a pilot job submission...
Definition: ComputeServiceMessagePayload.h:40
HTCondorComputeService(const std::string &hostname, const std::string &pool_name, std::set< ComputeService * > compute_resources, std::map< std::string, std::string > property_list={}, std::map< std::string, double > messagepayload_list={})
Constructor.
Definition: HTCondorComputeService.cpp:36
std::map< std::string, double > messagepayload_list
The service's messagepayload list.
Definition: Service.h:112
static const std::string RESOURCE_DESCRIPTION_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to state information on its resources.
Definition: ComputeServiceMessagePayload.h:58
void submitPilotJob(PilotJob *job, const std::map< std::string, std::string > &service_specific_arguments) override
Asynchronously submit a pilot job to the cloud service.
Definition: HTCondorComputeService.cpp:164
static const std::string STOP_DAEMON_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to terminate it.
Definition: ServiceMessagePayload.h:31
std::string hostname
The name of the host on which the daemon is running.
Definition: S4U_Daemon.h:51
~HTCondorComputeService() override
Destructor.
Definition: HTCondorComputeService.cpp:102
static const std::string SUBMIT_STANDARD_JOB_ANSWER_MESSAGE_PAYLOAD
The number of bytes in the control message sent by the daemon to acknowledge a standard job submissio...
Definition: ComputeServiceMessagePayload.h:28
static const std::string SUBMIT_PILOT_JOB_REQUEST_MESSAGE_PAYLOAD
The number of bytes in the control message sent to the daemon to submit a pilot job.
Definition: ComputeServiceMessagePayload.h:38
static const std::string SUPPORTS_PILOT_JOBS
Whether the compute service supports pilot jobs (true or false)
Definition: ComputeServiceProperty.h:26
A pilot (i.e., non-standard) workflow job that can be submitted to a ComputeService by a WMS (via a J...
Definition: PilotJob.h:29
void terminateStandardJob(StandardJob *job) override
Terminate a standard job to the compute service (virtual)
Definition: HTCondorComputeService.cpp:207
std::shared_ptr< StorageService > getLocalStorageService() const
Get the service's local storage service.
Definition: HTCondorComputeService.cpp:225
std::map< std::string, std::string > property_list
The service's property list.
Definition: Service.h:109
static const std::string SUPPORTS_STANDARD_JOBS
Whether the compute service supports standard jobs (true or false)
Definition: ComputeServiceProperty.h:24
void terminatePilotJob(PilotJob *job) override
Terminate a pilot job to the compute service.
Definition: HTCondorComputeService.cpp:217