JobManager.h

#ifndef WRENCH_PILOTJOBMANAGER_H
#define WRENCH_PILOTJOBMANAGER_H

#include <vector>
#include <set>

#include "wrench/services/Service.h"

namespace wrench {

    class WMS;
        class Workflow;
        class WorkflowTask;
        class WorkflowFile;
        class WorkflowJob;
        class PilotJob;
        class StandardJob;
        class ComputeService;
        class StorageService;

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


        class JobManager : public Service {

        public:


                void stop();

                void kill();

                StandardJob *createStandardJob(std::vector<WorkflowTask *> tasks,
                                               std::map<WorkflowFile *, StorageService *> file_locations,
                                               std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> pre_file_copies,
                                               std::set<std::tuple<WorkflowFile *, StorageService *, StorageService *>> post_file_copies,
                                               std::set<std::tuple<WorkflowFile *, StorageService *>> cleanup_file_deletions);

                StandardJob *createStandardJob(std::vector<WorkflowTask *> tasks,
                                               std::map<WorkflowFile *,
                                                               StorageService *> file_locations);

                StandardJob *createStandardJob(WorkflowTask *task,
                                               std::map<WorkflowFile *,
                                                               StorageService *> file_locations);

                PilotJob *createPilotJob();

                void submitJob(WorkflowJob *job, ComputeService *compute_service, std::map<std::string, std::string> service_specific_args = {});

                void terminateJob(WorkflowJob *);

                void forgetJob(WorkflowJob *job);

                std::set<PilotJob *> getPendingPilotJobs();

                std::set<PilotJob *> getRunningPilotJobs();
                /***********************/
                /***********************/

                ~JobManager();

        protected:

                friend class WMS;

                JobManager(WMS *wms);

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

        private:

                int main();

                // Relevant WMS
                WMS *wms;

                // Job map
                std::map<WorkflowJob*, std::unique_ptr<WorkflowJob>> jobs;

                // Job lists
                std::set<StandardJob *> pending_standard_jobs;
                std::set<StandardJob *> running_standard_jobs;
                std::set<StandardJob *> completed_standard_jobs;
                std::set<StandardJob *> failed_standard_jobs;

                std::set<PilotJob *> pending_pilot_jobs;
                std::set<PilotJob *> running_pilot_jobs;
                std::set<PilotJob *> completed_pilot_jobs;

        };

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

};

#endif //WRENCH_PILOTJOBMANAGER_H