go-micro/events/natsjs/options.go

package natsjs

import (
	"crypto/tls"
	"time"

	"go-micro.dev/v5/logger"
)

// Options which are used to configure the nats stream.
type Options struct {
	ClusterID             string
	ClientID              string
	Address               string
	NkeyConfig            string
	TLSConfig             *tls.Config
	Logger                logger.Logger
	SyncPublish           bool
	Name                  string
	DisableDurableStreams bool
	Username              string
	Password              string
	RetentionPolicy       int
	MaxAge                time.Duration
	MaxMsgSize            int
}

// Option is a function which configures options.
type Option func(o *Options)

// ClusterID sets the cluster id for the nats connection.
func ClusterID(id string) Option {
	return func(o *Options) {
		o.ClusterID = id
	}
}

// ClientID sets the client id for the nats connection.
func ClientID(id string) Option {
	return func(o *Options) {
		o.ClientID = id
	}
}

// Address of the nats cluster.
func Address(addr string) Option {
	return func(o *Options) {
		o.Address = addr
	}
}

// TLSConfig to use when connecting to the cluster.
func TLSConfig(t *tls.Config) Option {
	return func(o *Options) {
		o.TLSConfig = t
	}
}

// NkeyConfig string to use when connecting to the cluster.
func NkeyConfig(nkey string) Option {
	return func(o *Options) {
		o.NkeyConfig = nkey
	}
}

// Logger sets the underlying logger.
func Logger(log logger.Logger) Option {
	return func(o *Options) {
		o.Logger = log
	}
}

// SynchronousPublish allows using a synchronous publishing instead of the default asynchronous.
func SynchronousPublish(sync bool) Option {
	return func(o *Options) {
		o.SyncPublish = sync
	}
}

// Name allows to add a name to the natsjs connection.
func Name(name string) Option {
	return func(o *Options) {
		o.Name = name
	}
}

// DisableDurableStreams will disable durable streams.
func DisableDurableStreams() Option {
	return func(o *Options) {
		o.DisableDurableStreams = true
	}
}

// Authenticate authenticates the connection with the given username and password.
func Authenticate(username, password string) Option {
	return func(o *Options) {
		o.Username = username
		o.Password = password
	}
}
func RetentionPolicy(rp int) Option {
	return func(o *Options) {
		o.RetentionPolicy = rp
	}
}

func MaxMsgSize(size int) Option {
	return func(o *Options) {
		o.MaxMsgSize = size
	}
}
func MaxAge(age time.Duration) Option {
	return func(o *Options) {
		o.MaxAge = age
	}
}