re_configuration.go

package sentry

import (
	"github.com/D3rus/go-library/configurator"
	"github.com/D3rus/go-library/logger"
	logrusSentry "github.com/evalphobia/logrus_sentry"
	"github.com/sirupsen/logrus"
	"sync"
)

type ReConfiguration struct {
	sentryHook *logrusSentry.SentryHook
	rwMutex    *sync.RWMutex

	informer logger.Informer

	config *Config
}

func NewReConfigurationWithConfigurator(informer logger.Informer, config *Config, configurator configurator.Configurator) (*ReConfiguration, error) {
	return NewReConfiguration(informer, Configuration(config, configurator))
}

func NewReConfiguration(informer logger.Informer, config *Config) (*ReConfiguration, error) {
	reConfiguration := &ReConfiguration{informer: informer, config: config, rwMutex: &sync.RWMutex{}}

	return reConfiguration, reConfiguration.init(config)
}

func (reConfiguration *ReConfiguration) Levels() []logrus.Level {
	sentryHook := reConfiguration.SentryHook()
	if sentryHook == nil {
		return nil
	}

	return sentryHook.Levels()
}

func (reConfiguration *ReConfiguration) Fire(entry *logrus.Entry) error {
	sentryHook := reConfiguration.SentryHook()
	if sentryHook == nil {
		return nil
	}

	return sentryHook.Fire(entry)
}

func (reConfiguration *ReConfiguration) ReConfiguration(configurator configurator.Configurator) error {
	isConfigChange := false

	config := Configuration(NewConfig(), configurator)

	if config.DSN != reConfiguration.config.DSN {
		reConfiguration.informer.Infof("logger.sentry: old dsn '%s', new dsn '%s'", reConfiguration.config.DSN, config.DSN)

		reConfiguration.config.DSN = config.DSN
		isConfigChange = true
	}

	if isConfigChange {
		reConfiguration.informer.Info("logger.sentry: config change detected")

		return reConfiguration.init(reConfiguration.config)
	}

	return nil
}

func (reConfiguration *ReConfiguration) init(config *Config) error {
	reConfiguration.rwMutex.Lock()
	defer reConfiguration.rwMutex.Unlock()

	reConfiguration.sentryHook = nil

	if config.DSN == "" {
		return nil
	}

	sentryHook, err := New(config)
	if err != nil {
		return err
	}

	reConfiguration.sentryHook = sentryHook

	return nil
}

func (reConfiguration *ReConfiguration) SentryHook() *logrusSentry.SentryHook {
	reConfiguration.rwMutex.RLock()
	defer reConfiguration.rwMutex.RUnlock()

	return reConfiguration.sentryHook
}