options.go

package redisdb

import (
	"context"
	"time"

	"github.com/golang-queue/queue"
	"github.com/golang-queue/queue/core"
)

// Option for queue system
type Option func(*options)

type options struct {
	runFunc          func(context.Context, core.QueuedMessage) error
	logger           queue.Logger
	addr             string
	db               int
	connectionString string
	password         string
	streamName       string
	cluster          bool
	group            string
	consumer         string
	maxLength        int64
	blockTime        time.Duration
}

// WithAddr setup the addr of redis
func WithAddr(addr string) Option {
	return func(w *options) {
		w.addr = addr
	}
}

// WithMaxLength setup the max length for publish messages
func WithMaxLength(m int64) Option {
	return func(w *options) {
		w.maxLength = m
	}
}

// WithBlockTime setup the block time for publish messages
// we use the block command to make sure if no entry is found we wait
// until an entry is found
func WithBlockTime(m time.Duration) Option {
	return func(w *options) {
		w.blockTime = m
	}
}

// WithPassword redis password
func WithDB(db int) Option {
	return func(w *options) {
		w.db = db
	}
}

// WithCluster redis cluster
func WithCluster(enable bool) Option {
	return func(w *options) {
		w.cluster = enable
	}
}

// WithStreamName Stream name
func WithStreamName(name string) Option {
	return func(w *options) {
		w.streamName = name
	}
}

// WithGroup group name
func WithGroup(name string) Option {
	return func(w *options) {
		w.group = name
	}
}

// WithConsumer consumer name
func WithConsumer(name string) Option {
	return func(w *options) {
		w.consumer = name
	}
}

// WithPassword redis password
func WithPassword(passwd string) Option {
	return func(w *options) {
		w.password = passwd
	}
}

// WithConnectionString redis connection string
func WithConnectionString(connectionString string) Option {
	return func(w *options) {
		w.connectionString = connectionString
	}
}

// WithRunFunc setup the run func of queue
func WithRunFunc(fn func(context.Context, core.QueuedMessage) error) Option {
	return func(w *options) {
		w.runFunc = fn
	}
}

// WithLogger set custom logger
func WithLogger(l queue.Logger) Option {
	return func(w *options) {
		w.logger = l
	}
}

func newOptions(opts ...Option) options {
	defaultOpts := options{
		addr:       "127.0.0.1:6379",
		streamName: "golang-queue",
		group:      "golang-queue",
		consumer:   "golang-queue",
		logger:     queue.NewLogger(),
		runFunc: func(context.Context, core.QueuedMessage) error {
			return nil
		},
		blockTime: 60 * time.Second,
	}

	// Loop through each option
	for _, opt := range opts {
		// Call the option giving the instantiated
		opt(&defaultOpts)
	}

	return defaultOpts
}