checkpoint

type Checkpoint struct {
	StepNumber    int
	WorkflowInfo  WorkflowInfo
	RunnerData    RunnerStateData
	StateData     *hashmap.Map[workflow.ScopeKey, workflow.PortableValue]
	EdgeStateData map[string]workflow.PortableValue
	Parent        *workflow.CheckpointInfo
}

type CheckpointingHandle interface {
	IsCheckpointingEnabled() bool
	Checkpoints() []workflow.CheckpointInfo
	RestoreCheckpoint(context.Context, workflow.CheckpointInfo) error
}

type Manager interface {
	Commit(ctx context.Context, sessionID string, checkpoint *Checkpoint) (workflow.CheckpointInfo, error)
	Lookup(ctx context.Context, sessionID string, checkpointInfo workflow.CheckpointInfo) (*Checkpoint, error)
	RetrieveIndex(ctx context.Context, sessionID string, withParent *workflow.CheckpointInfo) ([]workflow.CheckpointInfo, error)
}

type PortableMessageEnvelope struct {
	MessageType  workflow.TypeID
	Message      workflow.PortableValue
	SourceID     string
	TargetID     string
	TraceContext map[string]string
}

type RunnerStateData struct {
	InstantiatedExecutors map[string]struct{}
	QueuedMessages        map[string][]*PortableMessageEnvelope
	OutstandingRequests   []*workflow.ExternalRequest
	RequestOwners         map[string]string
	ResponsePortOwners    map[string]string
}

type SessionCache[T any] struct {
	CheckpointIndex []workflow.CheckpointInfo
	Cache           map[workflow.CheckpointInfo]T
}

type WorkflowInfo struct {
	Executors       map[string]executorInfo
	Edges           map[string][]workflow.EdgeInfo
	RequestPorts    map[string]workflow.RequestPortInfo
	StartExecutorID string
	OutputExecutors map[string][]workflow.OutputTag
}