options.go

package pgqueue

import (
	"log/slog"
	"time"
)

// QueueOption configures a Queue.
type QueueOption func(*queueConfig)

type queueConfig struct {
	logger *slog.Logger
}

// WithLogger sets the logger for the Queue and its Workers.
func WithLogger(l *slog.Logger) QueueOption {
	return func(c *queueConfig) { c.logger = l }
}

// EnqueueOption configures a single Enqueue call.
type EnqueueOption func(*enqueueConfig)

type enqueueConfig struct {
	runAt            *time.Time
	status           *TaskStatus
	ignoreDuplicates bool
}

// WithRunAt schedules the task for a specific time instead of now.
func WithRunAt(t time.Time) EnqueueOption {
	return func(c *enqueueConfig) { c.runAt = &t }
}

// WithIgnoreDuplicates silently ignores the enqueue if a task with the
// same queue+hash already exists (ON CONFLICT DO NOTHING).
func WithIgnoreDuplicates() EnqueueOption {
	return func(c *enqueueConfig) { c.ignoreDuplicates = true }
}

// WithStatus enqueues the task with a non-default status (e.g. Disabled).
func WithStatus(s TaskStatus) EnqueueOption {
	return func(c *enqueueConfig) { c.status = &s }
}