boot_options.go

package nacelle

import (
	"fmt"

	"github.com/efritz/nacelle/logging"
)

type (
	// LoggingInitFunc creates a factory from a config object.
	LoggingInitFunc func(Config) (Logger, error)

	// BootstrapperConfigFunc is a function used to configure an instance of
	// a Bootstrapper.
	BootstrapperConfigFunc func(*bootstrapperConfig)
)

// WithConfigSourcer sets the source that should be used for populating config structs.
func WithConfigSourcer(configSourcer ConfigSourcer) BootstrapperConfigFunc {
	return func(c *bootstrapperConfig) { c.configSourcer = configSourcer }
}

// WithConfigMaskedKeys sets the keys that are redacted when printed by the config logger.
func WithConfigMaskedKeys(configMaskedKeys []string) BootstrapperConfigFunc {
	return func(c *bootstrapperConfig) { c.configMaskedKeys = configMaskedKeys }
}

// WithLoggingInitFunc sets the function that initializes logging.
func WithLoggingInitFunc(loggingInitFunc LoggingInitFunc) BootstrapperConfigFunc {
	return func(c *bootstrapperConfig) { c.loggingInitFunc = loggingInitFunc }
}

// WithLoggingFields sets additional fields sent with every log message.
func WithLoggingFields(loggingFields LogFields) BootstrapperConfigFunc {
	return func(c *bootstrapperConfig) { c.loggingFields = loggingFields }
}

// WithRunnerOptions passes RunnerConfigFuncs to the runner created by Boot.
func WithRunnerOptions(configs ...RunnerConfigFunc) BootstrapperConfigFunc {
	return func(c *bootstrapperConfig) { c.runnerConfigFuncs = configs }
}

func defaultLogginInitFunc(config Config) (Logger, error) {
	c := &logging.Config{}
	if err := config.Load(c); err != nil {
		return nil, fmt.Errorf("could not load logging config (%s)", err.Error())
	}

	return logging.InitGomolShim(c)
}