fix: update OfflinePushConsumerHandler.

openimsdk · Sep 10, 2024 · ad5c1c8 · ad5c1c8
1 parent c5ea3f5
commit ad5c1c8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
 	}

diff --git a/internal/push/push_handler.go b/internal/push/push_handler.go
@@ -257,7 +257,7 @@ func (c *ConsumerHandler) asyncOfflinePush(ctx context.Context, needOfflinePushU
 	if err := c.pushDatabase.MsgToOfflinePushMQ(ctx, conversationutil.GenConversationUniqueKeyForSingle(msg.SendID, msg.RecvID), needOfflinePushUserIDs, msg); err != nil {
 		log.ZError(ctx, "Msg To OfflinePush MQ error", err, "needOfflinePushUserIDs",
 			needOfflinePushUserIDs, "msg", msg)
-		// prommetrics.SingleChatMsgProcessFailedCounter.Inc()
+		prommetrics.SingleChatMsgProcessFailedCounter.Inc()
 		return
 	}
 }