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;


        /***********************/
        /***********************/

        static StorageService *SCRATCH;

        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> getNumCores();

        std::map<std::string, unsigned long> getNumIdleCores();

        unsigned long getTotalNumCores();

        unsigned long getTotalNumIdleCores();

        std::map<std::string, double> getMemoryCapacity();

        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;

        ComputeService(const std::string &hostname,
                       std::string service_name,
                       std::string mailbox_name_prefix,
                       double scratch_space_size);



    protected:

        ComputeService(const std::string &hostname,
                       std::string service_name,
                       std::string mailbox_name_prefix,
                       StorageService *scratch_space);

        StorageService *scratch_space_storage_service;
        std::shared_ptr<StorageService> scratch_space_storage_service_shared_ptr;

        /***********************/
        /***********************/

        /***********************/
        /***********************/

        StorageService *getScratch();
        std::shared_ptr<StorageService> getScratchSharedPtr();

        /***********************/
        /***********************/

    private:

        std::map<std::string, std::map<std::string, double>> getServiceResourceInformation();

    };


};

#endif //SIMULATION_COMPUTESERVICE_H