JobManagerMessage.h

 
#ifndef WRENCH_JOBMANAGERMESSAGE_H
#define WRENCH_JOBMANAGERMESSAGE_H
 
#include "wrench/simulation/SimulationMessage.h"
#include "wrench/services/compute/ComputeService.h"
#include "wrench/job/StandardJob.h"
#include "wrench-dev.h"
 
namespace wrench {
 
 
    /***********************/
    /***********************/
 
    class JobManagerMessage : public SimulationMessage {
    protected:
        explicit JobManagerMessage();
    };
 
    class JobManagerWakeupMessage : public JobManagerMessage {
    public:
        JobManagerWakeupMessage();
    };
 
    class JobManagerCompoundJobCompletedMessage : public JobManagerMessage {
    public:
        JobManagerCompoundJobCompletedMessage(std::shared_ptr<CompoundJob> job, std::shared_ptr<ComputeService> compute_service);
 
        std::shared_ptr<CompoundJob> job;
        std::shared_ptr<ComputeService> compute_service;
    };
 
    class JobManagerCompoundJobFailedMessage : public JobManagerMessage {
    public:
        JobManagerCompoundJobFailedMessage(std::shared_ptr<CompoundJob> job,
                                           std::shared_ptr<ComputeService> compute_service,
                                           std::shared_ptr<FailureCause> cause);
 
 
        std::shared_ptr<CompoundJob> job;
        std::shared_ptr<ComputeService> compute_service;
        std::shared_ptr<FailureCause> cause;
    };
    
    
    class JobManagerStandardJobCompletedMessage : public JobManagerMessage {
    public:
        JobManagerStandardJobCompletedMessage(std::shared_ptr<StandardJob> job, std::shared_ptr<ComputeService> compute_service,
                                              std::map<std::shared_ptr<WorkflowTask>, WorkflowTask::State> necessary_state_changes);
 
 
        std::shared_ptr<StandardJob> job;
        std::shared_ptr<ComputeService> compute_service;
        std::map<std::shared_ptr<WorkflowTask>, WorkflowTask::State> necessary_state_changes;
    };
 
    class JobManagerStandardJobFailedMessage : public JobManagerMessage {
    public:
        JobManagerStandardJobFailedMessage(std::shared_ptr<StandardJob> job, std::shared_ptr<ComputeService> compute_service,
                                           std::map<std::shared_ptr<WorkflowTask>, WorkflowTask::State> necessary_state_changes,
                                           std::set<std::shared_ptr<WorkflowTask>> necessary_failure_count_increments,
                                           std::shared_ptr<FailureCause> cause);
 
 
        std::shared_ptr<StandardJob> job;
        std::shared_ptr<ComputeService> compute_service;
        std::map<std::shared_ptr<WorkflowTask>, WorkflowTask::State> necessary_state_changes;
        std::set<std::shared_ptr<WorkflowTask>> necessary_failure_count_increments;
        std::shared_ptr<FailureCause> cause;
    };
 
    class JobManagerPilotJobFailedMessage : public JobManagerMessage {
    public:
        JobManagerPilotJobFailedMessage(std::shared_ptr<PilotJob> job,
                                           std::shared_ptr<ComputeService> compute_service,
                                           std::shared_ptr<FailureCause> cause);
 
 
        std::shared_ptr<PilotJob> job;
        std::shared_ptr<ComputeService> compute_service;
        std::shared_ptr<FailureCause> cause;
    };
 
    /***********************/
    /***********************/
 
};
 
#endif //WRENCH_JOBMANAGERMESSAGE_H