WorkflowExecutionEvent.h

#ifndef WRENCH_WORKFLOWEXECUTIONEVENT_H
#define WRENCH_WORKFLOWEXECUTIONEVENT_H

#include <string>
#include "FailureCause.h"

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

namespace wrench {

    class WorkflowTask;

    class WorkflowFile;

    class StandardJob;

    class PilotJob;

    class ComputeService;

    class StorageService;

    class FileRegistryService;

    class FileRegistryService;

    class WorkflowExecutionEvent {

    public:

        enum EventType {
                    UNDEFINED,
                    STANDARD_JOB_COMPLETION,
                    STANDARD_JOB_FAILURE,
                    PILOT_JOB_START,
                    PILOT_JOB_EXPIRATION,
                    FILE_COPY_COMPLETION,
                    FILE_COPY_FAILURE,
        };

        WorkflowExecutionEvent::EventType type;


        /***********************/
        /***********************/
        static std::unique_ptr<WorkflowExecutionEvent> waitForNextExecutionEvent(std::string);
        virtual ~WorkflowExecutionEvent() = default;

    protected:

        WorkflowExecutionEvent(EventType type) : type(type) {}
        /***********************/
        /***********************/

    };


    class StandardJobCompletedEvent : public WorkflowExecutionEvent {


    private:

        friend class WorkflowExecutionEvent;
        StandardJobCompletedEvent(StandardJob *standard_job,
                                  ComputeService *compute_service)
                : WorkflowExecutionEvent(STANDARD_JOB_COMPLETION),
                  standard_job(standard_job), compute_service(compute_service) {}
    public:

        StandardJob *standard_job;
        ComputeService *compute_service;
    };

    class StandardJobFailedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;

        StandardJobFailedEvent(StandardJob *standard_job,
                               ComputeService *compute_service,
                               std::shared_ptr<FailureCause> failure_cause)
                : WorkflowExecutionEvent(STANDARD_JOB_FAILURE),
                  standard_job(standard_job),
                  compute_service(compute_service),
                  failure_cause(failure_cause) {}

    public:

        StandardJob *standard_job;
        ComputeService *compute_service;
        std::shared_ptr<FailureCause> failure_cause;
    };


    class PilotJobStartedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;

        PilotJobStartedEvent(PilotJob *pilot_job,
                             ComputeService *compute_service)
                : WorkflowExecutionEvent(PILOT_JOB_START),
                  pilot_job(pilot_job), compute_service(compute_service) {}

    public:
        PilotJob *pilot_job;
        ComputeService *compute_service;
    };

    class PilotJobExpiredEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        PilotJobExpiredEvent(PilotJob *pilot_job,
                             ComputeService *compute_service)
                : WorkflowExecutionEvent(PILOT_JOB_EXPIRATION),
                  pilot_job(pilot_job), compute_service(compute_service) {}

    public:

        PilotJob *pilot_job;
        ComputeService *compute_service;
    };

    class FileCopyCompletedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        FileCopyCompletedEvent(WorkflowFile *file,
                               StorageService *storage_service,
                               FileRegistryService *file_registry_service,
                               bool file_registry_service_updated)
                : WorkflowExecutionEvent(FILE_COPY_COMPLETION),
                  file(file), storage_service(storage_service),
                  file_registry_service(file_registry_service),
                  file_registry_service_updated(file_registry_service_updated) {}

    public:
        WorkflowFile *file;
        StorageService *storage_service;
        FileRegistryService *file_registry_service;
        bool file_registry_service_updated;
    };


    class FileCopyFailedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        FileCopyFailedEvent(WorkflowFile *file,
                            StorageService *storage_service,
                            std::shared_ptr<FailureCause> failure_cause
        )
                : WorkflowExecutionEvent(FILE_COPY_FAILURE),
                  file(file), storage_service(storage_service),
                  failure_cause(failure_cause) {}

    public:

        WorkflowFile *file;
        StorageService *storage_service;
        std::shared_ptr<FailureCause> failure_cause;

    };

};

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



#endif //WRENCH_WORKFLOWEXECUTIONEVENT_H