BatschedBatchScheduler.h

 
#ifndef WRENCH_BATSCHEDBATCHSCHEDULER_H
#define WRENCH_BATSCHEDBATCHSCHEDULER_H
 
#include "wrench/services/compute/batch/batch_schedulers/BatchScheduler.h"
 
namespace wrench {
 
/***********************/
/***********************/
 
    class BatschedBatchScheduler : public BatchScheduler {
 
    public:
 
        explicit BatschedBatchScheduler(BatchComputeService *cs) : BatchScheduler(cs) {}
 
        void init() override;
 
        void launch() override;
 
        void shutdown() override;
 
        void processQueuedJobs() override;
 
        void processJobSubmission(std::shared_ptr<BatchJob> batch_job) override;
 
        void processJobFailure(std::shared_ptr<BatchJob> batch_job) override;
 
        void processJobCompletion(std::shared_ptr<BatchJob> batch_job) override;
 
        void processJobTermination(std::shared_ptr<BatchJob> batch_job) override;
 
        void processUnknownJobTermination(std::string job_id) override;
 
        std::map<std::string, double> getStartTimeEstimates(std::set<std::tuple<std::string, unsigned long, unsigned long, double>> set_of_jobs) override;
 
    private:
#ifdef ENABLE_BATSCHED
 
        void notifyJobEventsToBatSched(std::string job_id, std::string status, std::string job_state,
                                       std::string kill_reason, std::string event_type);
 
        void appendJobInfoToCSVOutputFile(BatchJob *batch_job, std::string status);
 
        void startBatschedNetworkListener();
 
 
        pid_t pid;
 
        unsigned long batsched_port;
 
 
#endif
 
    };
 
/***********************/
/***********************/
 
}
 
 
#endif //WRENCH_BATSCHEDBATCHSCHEDULER_H