queue

type Job struct {
	ID          string           `json:"id"`
	WorkflowID  string           `json:"workflow_id"`
	Workflow    *model.Workflow  `json:"workflow"`
	InputData   []model.DataItem `json:"input_data"`
	Priority    int              `json:"priority"` // Higher = more urgent
	Status      JobStatus        `json:"status"`
	RetryCount  int              `json:"retry_count"`
	MaxRetries  int              `json:"max_retries"`
	ResultData  []model.DataItem `json:"result_data,omitempty"`
	Error       string           `json:"error,omitempty"`
	CreatedAt   time.Time        `json:"created_at"`
	StartedAt   *time.Time       `json:"started_at,omitempty"`
	CompletedAt *time.Time       `json:"completed_at,omitempty"`
}

type JobQueue interface {
	// Enqueue adds a job to the queue
	Enqueue(job *Job) error

	// Dequeue retrieves and removes the next job from the queue
	// Blocks until a job is available or context is cancelled
	Dequeue(ctx context.Context) (*Job, error)

	// Ack marks a job as successfully completed
	Ack(jobID string, result *JobResult) error

	// Nack marks a job as failed, optionally requeueing for retry
	Nack(jobID string, err error) error

	// GetJob retrieves a job by ID
	GetJob(jobID string) (*Job, error)

	// GetPendingCount returns the number of pending jobs
	GetPendingCount() int

	// GetRunningCount returns the number of running jobs
	GetRunningCount() int

	// ListJobs returns jobs matching the given status (nil for all)
	ListJobs(status *JobStatus, limit int) ([]*Job, error)

	// Close gracefully shuts down the queue
	Close() error
}

type JobResult struct {
	Data  []model.DataItem
	Error error
}

type JobStatus string

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

type ResultHandler func(result *ResultMessage) error

type ResultMessage struct {
	Type        string           `json:"type"`         // "execution_result"
	ExecutionID string           `json:"execution_id"` // Execution ID
	Status      string           `json:"status"`       // "success", "failed", "timeout"
	ResultData  []model.DataItem `json:"result_data"`  // Output data
	Error       string           `json:"error"`        // Error message if failed
	DurationMs  int64            `json:"duration_ms"`  // Execution duration
	WorkerID    string           `json:"worker_id"`    // Which worker executed it
	Timestamp   time.Time        `json:"timestamp"`    // When result was generated
}

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

type WorkMessage struct {
	Type        string           `json:"type"`         // "execute_workflow"
	ExecutionID string           `json:"execution_id"` // Unique execution ID
	WorkflowID  string           `json:"workflow_id"`  // Workflow ID
	Workflow    *model.Workflow  `json:"workflow"`     // Full workflow definition
	InputData   []model.DataItem `json:"input_data"`   // Initial input data
	Priority    string           `json:"priority"`     // "high", "normal", "low"
	Timestamp   time.Time        `json:"timestamp"`    // When work was queued
	RetryCount  int              `json:"retry_count"`  // Number of retries
}

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

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

type WorkerRegistration struct {
	Type             string   `json:"type"`              // "register" or "heartbeat"
	WorkerID         string   `json:"worker_id"`         // Unique worker ID
	Capabilities     []string `json:"capabilities"`      // Supported node types
	MaxConcurrent    int      `json:"max_concurrent"`    // Max concurrent executions
	ActiveExecutions int      `json:"active_executions"` // Current active count
	UptimeSeconds    int64    `json:"uptime_seconds"`    // Worker uptime
}

type WorkerStatus struct {
	WorkerID         string    `json:"worker_id"`
	Status           string    `json:"status"` // "active", "offline"
	LastHeartbeat    time.Time `json:"last_heartbeat"`
	ActiveExecutions int       `json:"active_executions"`
	TotalExecutions  int64     `json:"total_executions"`
	MaxConcurrent    int       `json:"max_concurrent"`
	RegisteredAt     time.Time `json:"registered_at"`
}