diff --git a/internal/push/offlinepush_handler.go b/internal/push/offlinepush_handler.go index 500335565..bf69aed3e 100644 --- a/internal/push/offlinepush_handler.go +++ b/internal/push/offlinepush_handler.go @@ -22,9 +22,10 @@ type OfflinePushConsumerHandler struct { offlinePusher offlinepush.OfflinePusher } -func NewOfflinePushConsumerHandler(config *Config) (*OfflinePushConsumerHandler, error) { +func NewOfflinePushConsumerHandler(config *Config, offlinePusher offlinepush.OfflinePusher) (*OfflinePushConsumerHandler, error) { var offlinePushConsumerHandler OfflinePushConsumerHandler var err error + offlinePushConsumerHandler.offlinePusher = offlinePusher offlinePushConsumerHandler.OfflinePushConsumerGroup, err = kafka.NewMConsumerGroup(config.KafkaConfig.Build(), config.KafkaConfig.ToOfflineGroupID, []string{config.KafkaConfig.ToOfflinePushTopic}, true) if err != nil { diff --git a/internal/push/push.go b/internal/push/push.go index 152c4ea96..850f91d22 100644 --- a/internal/push/push.go +++ b/internal/push/push.go @@ -64,7 +64,7 @@ func Start(ctx context.Context, config *Config, client discovery.SvcDiscoveryReg return err } - offlinePushConsumer, err := NewOfflinePushConsumerHandler(config) + offlinePushConsumer, err := NewOfflinePushConsumerHandler(config, offlinePusher) if err != nil { return err }