diff --git a/pkg/config/appconfig/events.go b/pkg/config/appconfig/events.go index aaafc7839..1f79af151 100644 --- a/pkg/config/appconfig/events.go +++ b/pkg/config/appconfig/events.go @@ -18,7 +18,7 @@ const ( UpdateSubject events.Type = "update" ) -func (c *Config) registerSubscriptions(ctx context.Context) error { +func (c *Config) registerSubscriptions(ctx context.Context, bc context.Context) error { cfg := jetstream.StreamConfig{ Name: "APPCONFIG", Description: "AppConfig update events", @@ -46,10 +46,10 @@ func (c *Config) registerSubscriptions(ctx context.Context) error { c.jsCons = nil } - c.jsCons, err = consumer.Consume(c.handleMessageFunc(ctx), + c.jsCons, err = consumer.Consume(c.handleMessageFunc(bc), c.js.ConsumeErrHandlerWithRestart(context.Background(), c.logger, func(_ context.Context, ctx context.Context) error { - return c.registerSubscriptions(ctx) + return c.registerSubscriptions(bc, bc) })) if err != nil { return err diff --git a/pkg/config/appconfig/load.go b/pkg/config/appconfig/load.go index 5a5e32f9d..0080a9e6d 100644 --- a/pkg/config/appconfig/load.go +++ b/pkg/config/appconfig/load.go @@ -86,7 +86,7 @@ func New(p Params) (IConfig, error) { return err } - return cfg.registerSubscriptions(c) + return cfg.registerSubscriptions(c, ctx) })) p.LC.Append(fx.StopHook(func(ctx context.Context) error {