diff --git a/protocols/protocols.go b/protocols/protocols.go index 9aad2d8..f17330b 100644 --- a/protocols/protocols.go +++ b/protocols/protocols.go @@ -31,9 +31,15 @@ func BuildProtocols(config *viper.Viper) fx.Option { return fx.Options(options...) } -func SetupLifecycles(lifecycle fx.Lifecycle, protocols []registry.Protocol) { +type LifecyclesParams struct { + fx.In + + Protocols []registry.Protocol `group:"protocol"` +} + +func SetupLifecycles(lifecycle fx.Lifecycle, params LifecyclesParams) { for _, entry := range registry.GetRegistry() { - for _, protocol := range protocols { + for _, protocol := range params.Protocols { if protocol.Name() == entry.Key { lifecycle.Append(fx.Hook{ OnStart: func(ctx context.Context) error {