longrunning

type Checkpoint struct {
	ID        string         `json:"id"`
	Step      int            `json:"step"`
	State     any            `json:"state"`
	Timestamp time.Time      `json:"timestamp"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type CheckpointStore interface {
	SaveCheckpoint(ctx context.Context, exec *Execution) error
	LoadCheckpoint(ctx context.Context, execID string) (*Execution, error)
	ListCheckpoints(ctx context.Context) ([]*Execution, error)
	DeleteCheckpoint(ctx context.Context, execID string) error
}

type EventType string

type Execution struct {
	ID          string         `json:"id"`
	Name        string         `json:"name"`
	State       ExecutionState `json:"state"`
	Progress    float64        `json:"progress"`
	CurrentStep int            `json:"current_step"`
	TotalSteps  int            `json:"total_steps"`
	StepNames   []string       `json:"step_names,omitempty"`
	StartTime   time.Time      `json:"start_time"`
	LastUpdate  time.Time      `json:"last_update"`
	EndTime     *time.Time     `json:"end_time,omitempty"`
	Checkpoints []Checkpoint   `json:"checkpoints"`
	Metadata    map[string]any `json:"metadata,omitempty"`
	Error       string         `json:"error,omitempty"`
	// contains filtered or unexported fields
}

type ExecutionEvent struct {
	Type      EventType
	ExecID    string
	Step      int
	Timestamp time.Time
	Error     error
	State     any
}

type ExecutionState string

type Executor struct {
	OnEvent func(ExecutionEvent)
	// contains filtered or unexported fields
}

type ExecutorConfig struct {
	CheckpointInterval time.Duration `json:"checkpoint_interval"`
	CheckpointDir      string        `json:"checkpoint_dir"`
	MaxRetries         int           `json:"max_retries"`
	HeartbeatInterval  time.Duration `json:"heartbeat_interval"`
	AutoResume         bool          `json:"auto_resume"`
}

type ExecutorOption func(*Executor)

type FileCheckpointStore struct {
	// contains filtered or unexported fields
}

type NamedStep struct {
	Name string
	Func StepFunc
}

type PersistentCheckpointStore struct {
	// contains filtered or unexported fields
}

type StepFunc func(ctx context.Context, state any) (any, error)

type StepRegistry struct {
	// contains filtered or unexported fields
}

type TaskRecord struct {
	ID       string
	Status   string
	Progress float64
	Data     []byte // JSON-serialized Execution
	Metadata map[string]string
}

type TaskStoreAdapter interface {
	SaveTask(ctx context.Context, task *TaskRecord) error
	GetTask(ctx context.Context, taskID string) (*TaskRecord, error)
	ListTasks(ctx context.Context) ([]*TaskRecord, error)
	DeleteTask(ctx context.Context, taskID string) error
	UpdateProgress(ctx context.Context, taskID string, progress float64) error
	UpdateStatus(ctx context.Context, taskID string, status string) error
}

type TaskStoreBridge struct {
	// contains filtered or unexported fields
}