contract

type Consumer interface {
	// Subscribe 订阅消息
	Subscribe(ctx context.Context, topic string, handler message.Handler) error
	// Unsubscribe 取消订阅
	Unsubscribe(topic string) error
	// Close 关闭消费者
	Close() error
}

type DelayQueue interface {
	// Push 推送延时消息
	Push(ctx context.Context, msg *message.Message, delay time.Duration) error
	// Pop 弹出到期消息
	Pop(ctx context.Context) (*message.Message, error)
	// Remove 移除消息
	Remove(ctx context.Context, msgID string) error
	// Size 获取队列大小
	Size(ctx context.Context) (int64, error)
}

type ErrorCode int

type MQ interface {
	Producer() Producer
	Consumer() Consumer
	DelayQueue() DelayQueue
	HealthCheck() error
	Close() error
}

type MQError struct {
	Code    ErrorCode `json:"code"`
	Message string    `json:"message"`
	Cause   error     `json:"cause,omitempty"`
}

type Middleware func(next message.Handler) message.Handler

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

type Producer interface {
	// Send 发送消息
	Send(ctx context.Context, msg *message.Message) error
	// SendDelay 发送延时消息
	SendDelay(ctx context.Context, msg *message.Message, delay time.Duration) error
	// SendBatch 批量发送消息
	SendBatch(ctx context.Context, msgs []*message.Message) error
	// Close 关闭生产者
	Close() error
}