redisqueue

type BaseHandler struct{}

type ConsumerManager interface {
	Add(queueName string, consumer rmq.Consumer)
	StartConsuming(queueName string, replicas int, pollDuration time.Duration)
	StopConsuming(queueName string)
}

type ConsumerManagerMock struct {
	AddFn            func(queueName string, consumer rmq.Consumer)
	StartConsumingFn func(queueName string, replicas int, pollDuration time.Duration)
	StopConsumingFn  func(queueName string)
}

type FailHandler func(ctx context.Context, j Job, err error)

type HandleFunc func(context.Context, Job) error

type Handler interface {
	// Handle job return nil error mean job was processed successfully, otherwise job
	// was failed and method FailJob will be called
	Handle(ctx context.Context, job Job) error
	// Should retry on error
	ShouldRetryOnError(err error) bool
	// Should be move to rejected queue in case of fail
	ShouldRejectOnFailure(err error) bool
}

type HandlerFactory func(ctx context.Context) Handler

type HandlerMiddleWare func(next HandleFunc) HandleFunc

type HandlerMock struct {
	HandleFn                func(ctx context.Context, job Job) error
	ShouldRetryOnErrorFn    func(err error) bool
	ShouldRejectOnFailureFn func(err error) bool
}

type Job interface {
	// Get job ID
	GetID() string
	// Get user who triggered job
	GetUserID() string
	// Get trace id of request which triggered job
	GetTraceID() string
	// Set Queue
	OnQueue(queueName string)
	// Increase number of attempt times
	Attempt()
	// Get the number of job's attempt times
	GetAttempts() int
	// Mark job as failed
	Fail(err error)
	// Retry on error
	Retry(err error)
	// Determine if the job has been marked as a failure.
	HasFailed() bool
	// Get the number of times to attempt a job. Default is 1.
	GetMaxTries() int
	// Get job's Queue name
	GetQueue() string
	// Get delay time time in second before the job is retried again
	Delay() int
	// Return error string why job failed
	GetFailedError() string
}

type JobFactory func() Job

type Publisher interface {
	Publish(ctx context.Context, job Job) error
	PublishOnDelay(ctx context.Context, job Job) error
	PublishRejected(ctx context.Context, job Job) error
	UseMiddlewares(m ...PublisherHandlerMiddleWare)
}

type PublisherHandlerFunc func(ctx context.Context, job Job) error

type PublisherHandlerMiddleWare func(next PublisherHandlerFunc) PublisherHandlerFunc

type PublisherMock struct {
	PublishFn         func(queue string, job Job) error
	PublishOnDelayFn  func(queue string, job Job, delayAt time.Time) error
	PublishRejectedFn func(job Job) error
}

type RedisConfig struct {
	RedisMaster              string `envconfig:"REDIS_MASTER" required:"true"`
	SentinelHost             string `envconfig:"SENTINEL_HOST" required:"true"`
	SentinelPort             string `envconfig:"SENTINEL_PORT" required:"true"`
	RedisMaxActiveConnection int    `envconfig:"REDIS_MAX_ACTIVE" required:"false"`
	MaxIdle                  int    `envconfig:"REDIS_MAX_IDLE" required:"false"`
}

type RedisJob struct {
	ID       string
	UserID   string
	TraceID  string
	Queue    string
	Attempts int
	Failed   bool
	Error    string
}