scheduler.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/graph/types.h>
#include <map>
#include <optional>
#include <set>
#include <vector>
 
namespace neograph::graph {
 
class GraphState;
 
struct StepRouting {
    std::string node_name;
    std::optional<std::string> command_goto;
};
 
struct NextStepPlan {
    std::vector<std::string> ready;
    bool hit_end = false;
    std::optional<std::string> winning_command_goto;
};
 
using BarrierState = std::map<std::string, std::set<std::string>>;
 
using BarrierSpecs = std::map<std::string, std::set<std::string>>;
 
class NEOGRAPH_API Scheduler {
public:
    Scheduler(const std::vector<Edge>& edges,
              const std::vector<ConditionalEdge>& conditional_edges,
              BarrierSpecs barrier_specs = {});
 
    std::vector<std::string> resolve_next_nodes(
        const std::string& current, const GraphState& state) const;
 
    std::vector<std::string> plan_start_step() const;
 
    NextStepPlan plan_next_step(const std::vector<StepRouting>& routings,
                                const GraphState& state) const;
 
    NextStepPlan plan_next_step(const std::vector<std::string>& just_ran,
                                const std::vector<NodeResult>& results,
                                const GraphState& state) const;
 
    NextStepPlan plan_next_step(const std::vector<StepRouting>& routings,
                                const GraphState& state,
                                BarrierState& barrier_state) const;
 
    NextStepPlan plan_next_step(const std::vector<std::string>& just_ran,
                                const std::vector<NodeResult>& results,
                                const GraphState& state,
                                BarrierState& barrier_state) const;
 
    const BarrierSpecs& barrier_specs() const { return barrier_specs_; }
 
private:
    const std::vector<Edge>& edges_;
    const std::vector<ConditionalEdge>& conditional_edges_;
    BarrierSpecs barrier_specs_;
};
 
} // namespace neograph::graph