go-honeybee/config.go

package honeybee

import (
	"git.wisehodl.dev/jay/go-honeybee/transport"
	"git.wisehodl.dev/jay/go-honeybee/types"
	"time"
)

// ----------------------------------------------------------------------------
// Pool Config
// ----------------------------------------------------------------------------

// Types

type PoolConfig struct {
	InboxBufferSize  int
	EventsBufferSize int
	ConnectionConfig transport.ConnectionConfig
	WorkerFactory    WorkerFactory
	WorkerConfig     WorkerConfig
	Dialer           types.Dialer
}

type PoolOption func(*PoolConfig) error

// Constructor

func NewPoolConfig(options ...PoolOption) (*PoolConfig, error) {
	conf := GetDefaultPoolConfig()
	if err := applyPoolOptions(conf, options...); err != nil {
		return nil, err
	}
	if err := ValidatePoolConfig(conf); err != nil {
		return nil, err
	}
	return conf, nil
}

func GetDefaultPoolConfig() *PoolConfig {
	return &PoolConfig{
		InboxBufferSize:  256,
		EventsBufferSize: 10,
		ConnectionConfig: *transport.GetDefaultConnectionConfig(),
		WorkerFactory:    nil,
		WorkerConfig:     *GetDefaultWorkerConfig(),
	}
}

func applyPoolOptions(config *PoolConfig, options ...PoolOption) error {
	for _, option := range options {
		if err := option(config); err != nil {
			return err
		}
	}
	return nil
}

// Validation

func ValidatePoolConfig(config *PoolConfig) error {
	var err error

	err = transport.ValidateConnectionConfig(&config.ConnectionConfig)
	if err != nil {
		return err
	}

	err = ValidateWorkerConfig(&config.WorkerConfig)
	if err != nil {
		return err
	}

	return nil
}

func validateBufferSize(value int) error {
	if value < 1 {
		return InvalidBufferSize
	}
	return nil
}

// Options

func WithInboxBufferSize(value int) PoolOption {
	return func(c *PoolConfig) error {
		if err := validateBufferSize(value); err != nil {
			return err
		}
		c.InboxBufferSize = value
		return nil
	}
}

func WithEventsBufferSize(value int) PoolOption {
	return func(c *PoolConfig) error {
		if err := validateBufferSize(value); err != nil {
			return err
		}
		c.EventsBufferSize = value
		return nil
	}
}

func WithConnectionConfig(cc transport.ConnectionConfig) PoolOption {
	return func(c *PoolConfig) error {
		err := transport.ValidateConnectionConfig(&cc)
		if err != nil {
			return err
		}
		c.ConnectionConfig = cc
		return nil
	}
}

func WithWorkerConfig(wc WorkerConfig) PoolOption {
	return func(c *PoolConfig) error {
		err := ValidateWorkerConfig(&wc)
		if err != nil {
			return err
		}
		c.WorkerConfig = wc
		return nil
	}
}

func WithPoolDialer(d types.Dialer) PoolOption {
	return func(c *PoolConfig) error {
		c.Dialer = d
		return nil
	}
}

func WithWorkerFactory(wf WorkerFactory) PoolOption {
	return func(c *PoolConfig) error {
		c.WorkerFactory = wf
		return nil
	}
}

// ----------------------------------------------------------------------------
// Worker Config
// ----------------------------------------------------------------------------

// Types

type WorkerConfig struct {
	KeepaliveTimeout time.Duration
	ReconnectDelay   time.Duration
}

type WorkerOption func(*WorkerConfig) error

// Constructor

func NewWorkerConfig(options ...WorkerOption) (*WorkerConfig, error) {
	conf := GetDefaultWorkerConfig()
	if err := applyWorkerOptions(conf, options...); err != nil {
		return nil, err
	}
	if err := ValidateWorkerConfig(conf); err != nil {
		return nil, err
	}
	return conf, nil
}

func GetDefaultWorkerConfig() *WorkerConfig {
	return &WorkerConfig{
		KeepaliveTimeout: 60 * time.Second,
		ReconnectDelay:   2 * time.Second,
	}
}

func applyWorkerOptions(config *WorkerConfig, options ...WorkerOption) error {
	for _, option := range options {
		if err := option(config); err != nil {
			return err
		}
	}
	return nil
}

// Validation

func ValidateWorkerConfig(config *WorkerConfig) error {
	err := validateKeepaliveTimeout(config.KeepaliveTimeout)
	if err != nil {
		return err
	}

	return nil
}

func validateKeepaliveTimeout(value time.Duration) error {
	if value < 0 {
		return InvalidKeepaliveTimeout
	}
	return nil
}

func validateReconnectDelay(value time.Duration) error {
	if value < 0 {
		return InvalidReconnectDelay
	}
	return nil
}

// Options

// When KeepaliveTimeout is set to zero, keepalive timeouts are disabled.
func WithKeepaliveTimeout(value time.Duration) WorkerOption {
	return func(c *WorkerConfig) error {
		err := validateKeepaliveTimeout(value)
		if err != nil {
			return err
		}
		c.KeepaliveTimeout = value
		return nil
	}
}

func WithReconnectDelay(value time.Duration) WorkerOption {
	return func(c *WorkerConfig) error {
		err := validateReconnectDelay(value)
		if err != nil {
			return err
		}
		c.ReconnectDelay = value
		return nil
	}
}