delayed

type Consumer struct {
	sync.Mutex // 互斥锁
	// contains filtered or unexported fields
}

type ConsumerOption func(*ConsumerOptions)

type ConsumerOptions struct {
	Sleep      int // 当就绪队列为空时, 休眠时间, 单位为秒
	ClosedWait int // 关闭队列时, 等待队列中的任务执行完成的时间, 单位为秒
	MaxRetries int // 任务失败后的重试次数
	WorkerNum  int // 任务处理器的数量
}

type DelayUnit uint8

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

type IConsumer interface {
	Run() error                  // 启动消费者
	JoinWorker(worker *Worker)   // 加入一个 worker
	Close(context.Context) error // 关闭消费者
}

type IProduct interface {
	Dispatch(topic string, payload interface{}) (string, error)                                      // 添加任务
	DispatchDelay(topic string, payload interface{}, delay int, delayUnit DelayUnit) (string, error) // 添加延迟任务
}

type Job struct {
	Id        string `msgpack:"1"` // 任务ID
	Topic     string `msgpack:"2"` // 任务主题
	Delay     int64  `msgpack:"3"` // 任务延迟执行的时间戳
	Payload   []byte `msgpack:"4"` // 任务内容
	Timestamp int64  `msgpack:"5"` // 任务投递时间
	Retries   int    `msgpack:"6"` // 任务重试次数
}

type JobHandler interface {
	Topic() string                          // 获取 Job 的 topic
	OnHandle(context.Context, []byte) error // 处理 Job
	OnError(context.Context, []byte) error  // 处理 Job 处理失败的情况
}

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

type Queue interface {
	Run() error                           // 启动队列
	AddTopic(string) error                // 添加一个 topic
	Push(*Job) error                      // 添加 Job 到队列中, 投递的方向为队列尾部
	Pop(string) (*Job, error)             // 从队列中获取 Job
	PushDelay(*Job) error                 // 将 Job 添加到延迟队列中
	Cancel(string, string) error          // 通过指定 topic 和 jobId 取消延迟任务
	FetchReadyJob(string) ([]*Job, error) // 获取队列中已就绪的 Job
	FetchDelayJob(string) ([]*Job, error) // 获取延迟队列中的所有 Job
	ReadyLen(string) (int, error)         // 获取队列中已就绪的 Job 数量
	DelayLen(string) (int, error)         // 获取延迟队列中的 Job 数量
	Exists(string, string) (bool, error)  // 判断 JobID 是否存在
	Close() error                         // 关闭队列
}

type RedisQueue struct {
	sync.RWMutex // 读写锁
	// contains filtered or unexported fields
}

type Worker struct {
	Topic       string              // 任务 topic
	Handler     JobHandler          // 任务处理器
	WorkerCount int                 // 并行任务数
	WorkerPool  *semaphore.Weighted // 通过信号量控制并发协程数
}