stack/libs/go-libs/publish/circuit_breaker/module.go

package circuitbreaker

import (
	"context"
	"time"

	"github.com/formancehq/stack/libs/go-libs/logging"
	"github.com/formancehq/stack/libs/go-libs/publish/circuit_breaker/storage"
	topicmapper "github.com/formancehq/stack/libs/go-libs/publish/topic_mapper"
	"go.uber.org/fx"
)

func Module(schema string, openIntervalDuration time.Duration, storageLimit int) fx.Option {
	options := make([]fx.Option, 0)

	options = append(options,
		fx.Provide(func(
			logger logging.Logger,
			topicMapper *topicmapper.TopicMapperPublisherDecorator,
			store storage.Store,
			lc fx.Lifecycle,
		) *CircuitBreaker {
			cb := newCircuitBreaker(logger, topicMapper, store, openIntervalDuration)

			lc.Append(fx.Hook{
				OnStart: func(ctx context.Context) error {
					go cb.loop()
					return nil
				},
				OnStop: func(ctx context.Context) error {
					return cb.Close()
				},
			})
			return cb
		}),
	)

	options = append(options, storage.Module(schema, storageLimit))

	return fx.Options(options...)
}