ComputeService.h

 
#ifndef SIMULATION_COMPUTESERVICE_H
#define SIMULATION_COMPUTESERVICE_H
 
#include <map>
 
#include <iostream>
#include <cfloat>
#include <climits>
 
#include "wrench/services/Service.h"
#include "wrench/workflow/job/WorkflowJob.h"
 
namespace wrench {
 
    class Simulation;
 
    class StandardJob;
 
    class PilotJob;
 
    class StorageService;
 
    class ComputeService : public Service {
 
        /***********************/
        /***********************/
 
        friend class StandardJobExecutorTest;
        friend class Simulation;
 
        /***********************/
        /***********************/
 
 
    public:
 
        static constexpr unsigned long ALL_CORES = ULONG_MAX;
 
        static constexpr double ALL_RAM = DBL_MAX;
 
        /***********************/
        /***********************/
 
        virtual ~ComputeService() {}
 
        void stop() override;
 
        void submitJob(WorkflowJob *job, std::map<std::string, std::string> = {});
 
        void terminateJob(WorkflowJob *job);
 
        bool supportsStandardJobs();
 
        bool supportsPilotJobs();
 
        bool hasScratch();
 
        unsigned long getNumHosts();
 
        std::map<std::string, unsigned long> getPerHostNumCores();
 
        unsigned long getTotalNumCores();
 
        std::map<std::string, unsigned long> getPerHostNumIdleCores();
 
        virtual unsigned long getTotalNumIdleCores();
 
        std::map<std::string, double> getMemoryCapacity();
 
        std::map<std::string, double> getPerHostAvailableMemoryCapacity();
 
        std::map<std::string, double> getCoreFlopRate();
 
        double getTTL();
 
        double getTotalScratchSpaceSize();
 
        double getFreeScratchSpaceSize();
 
 
        /***********************/
        /***********************/
 
        /***********************/
        /***********************/
 
        virtual void
        submitStandardJob(StandardJob *job, std::map<std::string, std::string> &service_specific_arguments) = 0;
 
        virtual void submitPilotJob(PilotJob *job, std::map<std::string, std::string> &service_specific_arguments) = 0;
 
        virtual void terminateStandardJob(StandardJob *job) = 0;
 
        virtual void terminatePilotJob(PilotJob *job) = 0;
 
        std::shared_ptr<StorageService> getScratch();
 
 
        ComputeService(const std::string &hostname,
                       std::string service_name,
                       std::string mailbox_name_prefix,
                       std::string scratch_space_mount_point);
 
    protected:
 
        ComputeService(const std::string &hostname,
                       std::string service_name,
                       std::string mailbox_name_prefix,
                       std::shared_ptr<StorageService> scratch_space);
 
        std::shared_ptr<StorageService> scratch_space_storage_service;
 
        /***********************/
        /***********************/
 
        /***********************/
        /***********************/
 
        std::shared_ptr<StorageService> getScratchSharedPtr();
 
        /***********************/
        /***********************/
 
    private:
 
        std::shared_ptr<StorageService> scratch_space_storage_service_shared_ptr;
 
        std::map<std::string, std::map<std::string, double>> getServiceResourceInformation();
 
    };
 
 
};
 
#endif //SIMULATION_COMPUTESERVICE_H