runner

type DelayedTask struct {
	Id      string `json:"id"`
	Task    string `json:"task"`
	Time    int64  `json:"time"`
	Payload any    `json:"payload"`
}

type FailedTaskHandler func(ctx context.Context, task TaskMessage, err error) error

type FlushFunc func(data []timingDto) error

type LongQueueHook func(Stats)

type Stats struct {
	PerTaskTiming     map[string]int64
	PredictedWaitTime float64
	AvgTiming         time.Duration
	AvgScheduleTiming int64
	// Estimated TPS, it's not actual tps
	TPS float64
}

type Task struct {
	Name     string
	MaxRetry int

	// When a task takes more than reservation time, woker will send a HeartBeat to queue, to prevent reClaim by
	ReservationTimeout time.Duration

	Action TaskAction `json:"-"`

	// If 'unique' is set to true, the task will not be dispatched if another instance of same task is already in the queue and has not finished processing.
	Unique bool

	// If 'unique' is set to true and a 'UniqueKey' is specified, the task with the same 'UniqueKey' will not be dispatched if another instance of it with the same 'UniqueKey' is already in the queue and has not finished processing.
	UniqueKey UniqueKeyFunc

	// If `UniqueFor` seconds have passed since a task with the same name and UniqueKey was enqueued and it has not finished processing yet, the new task with the same name and UniqueKey can now be dispatched.
	// this option is required when Unique is true
	UniqueFor int64
}

type TaskAction func(ctx context.Context, payload any) error

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

type TaskMessage struct {
	ID                 string `json:"-"`
	Unique             bool   `json:"unique"`
	UniqueFor          int64  `json:"unique_for"`
	UniqueKey          string `json:"unique_key"`
	UniqueLockValue    string `json:"unique_lock_value"`
	ReservationTimeout int64  `json:"reservation_timeout"`
	TaskName           string `json:"task_name"`
	Payload            any    `json:"payload"`
}

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

type TaskRunnerConfig struct {
	// Optional
	Host string

	BatchSize       int
	ConsumerGroup   string
	ConsumersPrefix string

	NumWorkers  int
	NumFetchers int

	// ReplicationFactor Number of pod replicas configured, affecting metric calculations
	// Let T_avg be the average execution time of task, Q_len be the length of the queue, and W_num be the number of workers
	// The total execution time for the queue is estimated as (T_avg * Q_len) / (W_num * ReplicationFactor).
	// Deprecated: taskrunner automaticaly handles
	ReplicationFactor int

	FailedTaskHandler FailedTaskHandler

	LongQueueHook      LongQueueHook
	LongQueueThreshold time.Duration

	BlockDuration time.Duration

	// MetricsResetInterval defines how often timing metrics (sum and count) are reset.
	// If zero or negative, a default of 24 hours is used.
	MetricsResetInterval time.Duration
}

type TaskRunnerError string

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

type UniqueKeyFunc func(payload any) string