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;

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


    protected:

        std::shared_ptr<StorageService> getScratch();

        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