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:

        /***********************/
        /***********************/
        static std::shared_ptr<WorkflowExecutionEvent> waitForNextExecutionEvent(std::string);
        static std::shared_ptr<WorkflowExecutionEvent> waitForNextExecutionEvent(std::string, double timeout);

        virtual std::string toString() { return "Generic WorkflowExecutionEvent"; }

        virtual ~WorkflowExecutionEvent() = default;

    protected:
        WorkflowExecutionEvent() = default;

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



    };


    class StandardJobCompletedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;

        StandardJobCompletedEvent(StandardJob *standard_job,
                                  std::shared_ptr<ComputeService>  compute_service)
                : standard_job(standard_job), compute_service(compute_service) {}
    public:

        StandardJob *standard_job;
        std::shared_ptr<ComputeService>  compute_service;

        std::string toString() override { return "StandardJobCompletedEvent (job: " + this->standard_job->getName() + "; cs = " + this->compute_service->getName() + ")";}
    };

    class StandardJobFailedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;

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

    public:

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

        std::string toString() override { return "StandardJobFailedEvent (job: " + this->standard_job->getName() + "; cs = " +
                                                 this->compute_service->getName() + "; cause: " + this->failure_cause->toString() + ")";}

    };


    class PilotJobStartedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;

        PilotJobStartedEvent(PilotJob *pilot_job,
                             std::shared_ptr<ComputeService>  compute_service)
                : pilot_job(pilot_job), compute_service(compute_service) {}

    public:
        PilotJob *pilot_job;
        std::shared_ptr<ComputeService>  compute_service;

        std::string toString() override { return "PilotJobStartedEvent (cs = " + this->compute_service->getName() + ")";}

    };

    class PilotJobExpiredEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        PilotJobExpiredEvent(PilotJob *pilot_job,
                             std::shared_ptr<ComputeService>  compute_service)
                : pilot_job(pilot_job), compute_service(compute_service) {}

    public:

        PilotJob *pilot_job;
        std::shared_ptr<ComputeService>  compute_service;

        std::string toString() override { return "PilotJobExpiredEvent (cs = " + this->compute_service->getName() + ")";}

    };

    class FileCopyCompletedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        FileCopyCompletedEvent(WorkflowFile *file,
                               std::shared_ptr<FileLocation> src,
                               std::shared_ptr<FileLocation> dst,
                               std::shared_ptr<FileRegistryService> file_registry_service,
                               bool file_registry_service_updated)
                : file(file), src(src), dst(dst),
                  file_registry_service(file_registry_service),
                  file_registry_service_updated(file_registry_service_updated) {}

    public:
        WorkflowFile *file;
        std::shared_ptr<FileLocation> src;
        std::shared_ptr<FileLocation> dst;
        std::shared_ptr<FileRegistryService> file_registry_service;
        bool file_registry_service_updated;

        std::string toString() override {
            return "FileCopyCompletedEvent (file: " + this->file->getID() +
                   "; src = " + this->src->toString() +
                   "; dst = "+ this->dst->toString();
        }

    };


    class FileCopyFailedEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        FileCopyFailedEvent(WorkflowFile *file,
                            std::shared_ptr<FileLocation> src,
                            std::shared_ptr<FileLocation> dst,
                            std::shared_ptr<FailureCause> failure_cause
        )
                : file(file), src(src), dst(dst),
                  failure_cause(failure_cause) {}

    public:

        WorkflowFile *file;
        std::shared_ptr<FileLocation> src;
        std::shared_ptr<FileLocation> dst;
        std::shared_ptr<FailureCause> failure_cause;

        std::string toString() override {
            return "FileCopyFailedEvent (file: " + this->file->getID() +
                   "; src = " + this->src->toString() +
                   "; dst = " + this->dst->toString() +
                   "; cause: " + this->failure_cause->toString() + ")";}

    };

    class TimerEvent : public WorkflowExecutionEvent {

    private:

        friend class WorkflowExecutionEvent;
        TimerEvent(std::string message)
                : message(message) {}

    public:

        std::string message;

        std::string toString() override { return "TimerEvent (message: " + this->message + ")";}

    };

};

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



#endif //WRENCH_WORKFLOWEXECUTIONEVENT_H