ComputeService.h
1 
10 #ifndef SIMULATION_COMPUTESERVICE_H
11 #define SIMULATION_COMPUTESERVICE_H
12 
13 #include <map>
14 
15 #include <iostream>
16 #include <cfloat>
17 #include <climits>
18 
19 #include "wrench/services/Service.h"
20 #include "wrench/workflow/job/WorkflowJob.h"
21 #include "wrench/workflow/job/StandardJob.h"
22 #include "wrench/workflow/job/PilotJob.h"
23 
24 namespace wrench {
25 
26  class Simulation;
27 
28  class StorageService;
29 
33  class ComputeService : public Service {
34 
35  /***********************/
37  /***********************/
38 
39  friend class StandardJobExecutorTest;
40  friend class Simulation;
41 
42  /***********************/
44  /***********************/
45 
46 
47  public:
48 
52  static constexpr unsigned long ALL_CORES = ULONG_MAX;
53 
57  static constexpr double ALL_RAM = DBL_MAX;
58 
59  /***********************/
61  /***********************/
62 
63  virtual ~ComputeService() {}
64 
65  void stop() override;
66 
67 // void submitJob(WorkflowJob *job, const std::map<std::string, std::string>& = {});
68 
69  void terminateJob(std::shared_ptr<WorkflowJob> job);
70 
71  bool supportsStandardJobs();
72 
73  bool supportsPilotJobs();
74 
75  bool hasScratch();
76 
77  unsigned long getNumHosts();
78 
79  std::vector<std::string> getHosts();
80 
81  std::map<std::string, unsigned long> getPerHostNumCores();
82 
83  unsigned long getTotalNumCores();
84 
85  std::map<std::string, unsigned long> getPerHostNumIdleCores();
86 
87  virtual unsigned long getTotalNumIdleCores();
88 
89  std::map<std::string, double> getMemoryCapacity();
90 
91  std::map<std::string, double> getPerHostAvailableMemoryCapacity();
92 
93  std::map<std::string, double> getCoreFlopRate();
94 
95  double getTTL();
96 
97  double getTotalScratchSpaceSize();
98 
99  double getFreeScratchSpaceSize();
100 
101 
102  /***********************/
104  /***********************/
105 
106  /***********************/
108  /***********************/
109 
116  virtual void
117  submitStandardJob(std::shared_ptr<StandardJob> job, const std::map<std::string, std::string> &service_specific_arguments) = 0;
118 
125  virtual void submitPilotJob(std::shared_ptr<PilotJob> job, const std::map<std::string, std::string> &service_specific_arguments) = 0;
126 
131  virtual void terminateStandardJob(std::shared_ptr<StandardJob> job) = 0;
132 
137  virtual void terminatePilotJob(std::shared_ptr<PilotJob> job) = 0;
138 
143  std::shared_ptr<StorageService> getScratch();
144 
145 
146  ComputeService(const std::string &hostname,
147  std::string service_name,
148  std::string mailbox_name_prefix,
149  std::string scratch_space_mount_point);
150 
151  protected:
152 
153  friend class JobManager;
154 
155  void submitJob(std::shared_ptr<WorkflowJob> job, const std::map<std::string, std::string>& = {});
156 
157  ComputeService(const std::string &hostname,
158  std::string service_name,
159  std::string mailbox_name_prefix,
160  std::shared_ptr<StorageService> scratch_space);
161 
163  std::shared_ptr<StorageService> scratch_space_storage_service;
164 
165  /***********************/
167  /***********************/
168 
169  /***********************/
171  /***********************/
172 
173  std::shared_ptr<StorageService> getScratchSharedPtr();
174 
175  /***********************/
177  /***********************/
178 
179  private:
180 
181 
182  std::shared_ptr<StorageService> scratch_space_storage_service_shared_ptr;
183 
184  std::map<std::string, std::map<std::string, double>> getServiceResourceInformation();
185 
186  };
187 
188 
189 };
190 
191 #endif //SIMULATION_COMPUTESERVICE_H
double getTotalScratchSpaceSize()
Get the total capacity of the compute service's scratch storage space.
Definition: ComputeService.cpp:469
std::map< std::string, double > getPerHostAvailableMemoryCapacity()
Get ram availability for each of the compute service's host.
Definition: ComputeService.cpp:309
bool hasScratch()
Checks if the compute service has a scratch space.
Definition: ComputeService.cpp:503
std::map< std::string, unsigned long > getPerHostNumIdleCores()
Get idle core counts for each of the compute service's host.
Definition: ComputeService.cpp:282
double getTTL()
Get the time-to-live of the compute service.
Definition: ComputeService.cpp:412
virtual unsigned long getTotalNumIdleCores()
Get the total idle core count for all hosts of the compute service.
Definition: ComputeService.cpp:336
A helper daemon (co-located with and explicitly started by a WMS), which is used to handle all job ex...
Definition: JobManager.h:40
std::map< std::string, double > getCoreFlopRate()
Get the per-core flop rate of the compute service's hosts.
Definition: ComputeService.cpp:361
The compute service base class.
Definition: ComputeService.h:33
std::vector< std::string > getHosts()
Get the list of the compute service's compute host.
Definition: ComputeService.cpp:200
Definition: Alarm.cpp:20
std::map< std::string, unsigned long > getPerHostNumCores()
Get core counts for each of the compute service's host.
Definition: ComputeService.cpp:229
double getFreeScratchSpaceSize()
Get the free space on the compute service's scratch storage space.
Definition: ComputeService.cpp:478
std::shared_ptr< StorageService > getScratchSharedPtr()
Get a shared pointer to the compute service's scratch storage space.
Definition: ComputeService.cpp:495
void stop() override
Stop the compute service - must be called by the stop() method of derived classes.
Definition: ComputeService.cpp:33
unsigned long getTotalNumCores()
Get the total core counts for all hosts of the compute service.
Definition: ComputeService.cpp:256
static constexpr unsigned long ALL_CORES
A convenient constant to mean "use all cores of a physical host" whenever a number of cores is needed...
Definition: ComputeService.h:52
A class that provides basic simulation methods. Once the simulation object has been explicitly or imp...
Definition: Simulation.h:46
bool supportsStandardJobs()
Get whether the compute service supports standard jobs or not.
Definition: ComputeService.cpp:158
static constexpr double ALL_RAM
A convenient constant to mean "use all ram of a physical host" whenever a ram capacity is needed when...
Definition: ComputeService.h:57
A service that can be added to the simulation and that can be used by a WMS when executing a workflow...
Definition: Service.h:26
unsigned long getNumHosts()
Get the number of hosts that the compute service manages.
Definition: ComputeService.cpp:177
std::map< std::string, double > getMemoryCapacity()
Get the RAM capacities for each of the compute service's hosts.
Definition: ComputeService.cpp:386
void terminateJob(std::shared_ptr< WorkflowJob > job)
Terminate a previously-submitted job (which may or may not be running yet)
Definition: ComputeService.cpp:85
bool supportsPilotJobs()
Get whether the compute service supports pilot jobs or not.
Definition: ComputeService.cpp:166