ytask

type BaseTask struct {
	ID                string
	Type              string
	Priority          int
	Status            TaskStatus
	FailureCount      int
	NextExecutionTime time.Time
}

type CPUUsageFunc func(prevTime *time.Time, prevUsage *float64) float64

type ExampleTask struct {
	BaseTask
	// Data holds the task's data
	Data string
	// Duration is how long the task will take to execute
	Duration time.Duration
	// ShouldFail indicates if the task should fail
	ShouldFail bool
}

type FailureHandler func(task Task, err error) (time.Time, bool)

type MemoryUsageFunc func() float64

type Middleware func(next func(ctx context.Context, task Task) error) func(ctx context.Context, task Task) error

type PriorityMode int

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

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

type Task interface {
	// Execute runs the task and returns any error
	Execute(ctx context.Context) error

	// GetID returns the unique identifier of the task
	GetID() string

	// GetType returns the type of the task
	GetType() string

	// GetPriority returns the priority of the task
	GetPriority() int

	// GetStatus returns the current status of the task
	GetStatus() TaskStatus

	// SetStatus updates the status of the task
	SetStatus(status TaskStatus)

	// GetFailureCount returns the number of times this task has failed
	GetFailureCount() int

	// IncrementFailureCount increments the failure count
	IncrementFailureCount()

	// GetNextExecutionTime returns the next time this task should be executed
	GetNextExecutionTime() time.Time

	// SetNextExecutionTime sets the next time this task should be executed
	SetNextExecutionTime(time time.Time)

	// IsDuplicate checks if this task is a duplicate of another task
	IsDuplicate(other Task) bool
}

type TaskStats struct {
	Queued    int64
	Executing int64
	Completed int64
	Failed    int64
	Paused    int64
	Cancelled int64
}

type TaskStatus string

type TaskTypeConfig struct {
	// MaxConcurrency is the maximum number of concurrent tasks of this type
	MaxConcurrency int
	// PriorityLevels is the number of priority levels for this task type
	PriorityLevels int
	// PriorityMode determines how priorities are handled
	PriorityMode PriorityMode
	// PriorityPercentages defines the percentage of tasks to execute at each priority level
	// Only used when PriorityMode is PriorityModePercentage
	PriorityPercentages []int
	// FailureHandler is called when a task fails
	FailureHandler FailureHandler
	// FilterDuplicates determines whether to filter duplicate tasks
	FilterDuplicates bool
}