WorkflowTask.h

 
#ifndef WRENCH_WORKFLOWTASK_H
#define WRENCH_WORKFLOWTASK_H
 
#include <map>
#include <stack>
#include <set>
 
#include "wrench/workflow/job/WorkflowJob.h"
#include "wrench/workflow/WorkflowFile.h"
#include "wrench/workflow/parallel_model/ParallelModel.h"
#include "wrench/workflow/parallel_model/AmdahlParallelModel.h"
#include "wrench/workflow/parallel_model/ConstantEfficiencyParallelModel.h"
#include "wrench/workflow/parallel_model/CustomParallelModel.h"
 
#include <boost/graph/adjacency_list.hpp>
 
namespace wrench {
 
    class WorkflowTask {
 
    public:
        std::string getID() const;
 
        double getFlops() const;
 
        unsigned long getMinNumCores() const;
 
        unsigned long getMaxNumCores() const;
 
        std::shared_ptr<ParallelModel> getParallelModel();
 
        void setParallelModel(std::shared_ptr<ParallelModel> model);
 
        double getMemoryRequirement() const;
 
        unsigned long getNumberOfChildren() const;
 
        std::vector<WorkflowTask *> getChildren() const;
 
        unsigned long getNumberOfParents() const;
 
        std::vector<WorkflowTask *> getParents() const;
 
        void addInputFile(WorkflowFile *file);
 
        void addOutputFile(WorkflowFile *file);
 
        unsigned int getFailureCount();
 
 
        /***********************/
        /***********************/
 
        enum State {
            NOT_READY,
            READY,
            PENDING,
            COMPLETED,
            UNKNOWN
        };
 
        static std::string stateToString(WorkflowTask::State state);
 
        WorkflowJob *getJob() const;
 
        Workflow *getWorkflow() const;
 
        std::string getClusterID() const;
 
        void setClusterID(std::string);
 
        void setPriority(long);
 
        unsigned long getPriority() const;
 
        void setAverageCPU(double);
 
        double getAverageCPU() const;
 
        void setBytesRead(unsigned long);
 
        unsigned long getBytesRead() const;
 
        void setBytesWritten(unsigned long);
 
        unsigned long getBytesWritten() const;
 
        std::vector<WorkflowFile *> getInputFiles();
 
        std::vector<WorkflowFile *> getOutputFiles();
 
        unsigned long getTopLevel();
 
        double getStartDate();
 
        double getEndDate();
 
        double getFailureDate();
 
        double getTerminationDate();
 
        double getReadInputStartDate();
 
        double getReadInputEndDate();
 
        double getComputationStartDate();
 
        double getComputationEndDate();
 
        double getWriteOutputStartDate();
 
        double getWriteOutputEndDate();
 
        unsigned long getNumCoresAllocated();
 
        struct WorkflowTaskExecution;
 
        std::stack<WorkflowTaskExecution> getExecutionHistory();
 
        std::string getExecutionHost();
 
        WorkflowTask::State getState() const;
 
        std::string getColor();
 
        void setColor(std::string);
 
        /***********************/
        /***********************/
 
 
        /***********************/
        /***********************/
 
        enum InternalState {
            TASK_NOT_READY,
            TASK_READY,
            TASK_RUNNING,
            TASK_COMPLETED,
            TASK_FAILED
        };
 
        static std::string stateToString(WorkflowTask::InternalState state);
 
        unsigned long updateTopLevel();
 
        void setInternalState(WorkflowTask::InternalState);
 
        void setState(WorkflowTask::State);
 
        void setUpcomingState(WorkflowTask::State);
 
        WorkflowTask::State getUpcomingState() const;
 
        WorkflowTask::InternalState getInternalState() const;
 
        void setJob(WorkflowJob *job);
 
        void setStartDate(double date);
 
        void setEndDate(double date);
 
        void setReadInputStartDate(double date);
 
        void setReadInputEndDate(double date);
 
        void setComputationStartDate(double date);
 
        void setComputationEndDate(double date);
 
        void setWriteOutputStartDate(double date);
 
        void setWriteOutputEndDate(double date);
 
        void setFailureDate(double date);
 
        void setTerminationDate(double date);
 
        void incrementFailureCount();
 
        void setExecutionHost(std::string hostname);
 
        void setNumCoresAllocated(unsigned long num_cores);
 
        struct WorkflowTaskExecution {
            double task_start = -1.0;
            double read_input_start = -1.0;
            double read_input_end = -1.0;
            double computation_start = -1.0;
            double computation_end = -1.0;
            double write_output_start = -1.0;
            double write_output_end = -1.0;
            double task_end = -1.0;
            double task_failed = -1.0;
            double task_terminated = -1.0;
 
            std::string execution_host = "";
            unsigned long num_cores_allocated = 0;
 
            WorkflowTaskExecution(double task_start) : task_start(task_start) {}
 
 
        };
 
 
        /***********************/
        /***********************/
 
    private:
        friend class Workflow;
 
        std::string id;                    // Task ID
        std::string cluster_id;            // ID for clustered task
        std::string color;                 // A RGB color formatted as "#rrggbb"
        double flops;                      // Number of flops
        double average_cpu = -1;           // Average CPU utilization
        unsigned long bytes_read = -1;     // Total bytes read in KB
        unsigned long bytes_written = -1;  // Total bytes written in KB
        unsigned long min_num_cores;
        unsigned long max_num_cores;
        std::shared_ptr<ParallelModel> parallel_model;
        double memory_requirement;
        unsigned long priority = 0;        // Task priority
        unsigned long toplevel;            // 0 if entry task
        unsigned int failure_count = 0;    // Number of times the tasks has failed
        std::string execution_host;        // Host on which the task executed ("" if not executed successfully - yet)
        State visible_state;               // To be exposed to developer level
        State upcoming_visible_state;      // A visible state that will become active once a WMS has process a previously sent workflow execution event
        InternalState internal_state;      // Not to be exposed to developer level
 
        Workflow *workflow;                // Containing workflow
 
        std::map<std::string, WorkflowFile *> output_files;   // List of output files
        std::map<std::string, WorkflowFile *> input_files;    // List of input files
 
        // Private constructor (called by Workflow)
        WorkflowTask(std::string id,
                     double t,
                     unsigned long min_num_cores,
                     unsigned long max_num_cores,
                     double memory_requirement);
 
        // Containing job
        WorkflowJob *job;
 
        std::stack<WorkflowTaskExecution> execution_history;
 
        friend class DagOfTasks;
    };
};
 
#endif //WRENCH_WORKFLOWTASK_H