errcode

2026-05-10 20:15:59 +08:00 · 2023-02-20 10:13:29 +08:00
parent b6d19dba00
commit 5be441f4fd
37 changed files with 366 additions and 672 deletions
@@ -0,0 +1,64 @@
+/*
+** description("").
+** copyright('Open_IM,www.Open_IM.io').
+** author("fg,Gordon@tuoyun.net").
+** time(2021/5/13 10:33).
+ */
+package push
+
+import (
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
+	kfk "Open_IM/pkg/common/kafka"
+	"Open_IM/pkg/common/log"
+	pbChat "Open_IM/pkg/proto/msg"
+	pbPush "Open_IM/pkg/proto/push"
+	"Open_IM/pkg/utils"
+	"github.com/Shopify/sarama"
+	"github.com/golang/protobuf/proto"
+)
+
+type ConsumerHandler struct {
+	pushConsumerGroup *kfk.MConsumerGroup
+}
+
+func (c *ConsumerHandler) Init() {
+	c.pushConsumerGroup = kfk.NewMConsumerGroup(&kfk.MConsumerGroupConfig{KafkaVersion: sarama.V2_0_0_0,
+		OffsetsInitial: sarama.OffsetNewest, IsReturnErr: false}, []string{config.Config.Kafka.Ms2pschat.Topic}, config.Config.Kafka.Ms2pschat.Addr,
+		config.Config.Kafka.ConsumerGroupID.MsgToPush)
+}
+func (c *ConsumerHandler) handleMs2PsChat(msg []byte) {
+	log.NewDebug("", "msg come from kafka  And push!!!", "msg", string(msg))
+	msgFromMQ := pbChat.PushMsgDataToMQ{}
+	if err := proto.Unmarshal(msg, &msgFromMQ); err != nil {
+		log.Error("", "push Unmarshal msg err", "msg", string(msg), "err", err.Error())
+		return
+	}
+	pbData := &pbPush.PushMsgReq{
+		OperationID:  msgFromMQ.OperationID,
+		MsgData:      msgFromMQ.MsgData,
+		PushToUserID: msgFromMQ.PushToUserID,
+	}
+	sec := msgFromMQ.MsgData.SendTime / 1000
+	nowSec := utils.GetCurrentTimestampBySecond()
+	if nowSec-sec > 10 {
+		return
+	}
+	switch msgFromMQ.MsgData.SessionType {
+	case constant.SuperGroupChatType:
+		MsgToSuperGroupUser(pbData)
+	default:
+		MsgToUser(pbData)
+	}
+}
+func (ConsumerHandler) Setup(_ sarama.ConsumerGroupSession) error   { return nil }
+func (ConsumerHandler) Cleanup(_ sarama.ConsumerGroupSession) error { return nil }
+func (c *ConsumerHandler) ConsumeClaim(sess sarama.ConsumerGroupSession,
+	claim sarama.ConsumerGroupClaim) error {
+	for msg := range claim.Messages() {
+		log.NewDebug("", "kafka get info to mysql", "msgTopic", msg.Topic, "msgPartition", msg.Partition, "msg", string(msg.Value))
+		c.handleMs2PsChat(msg.Value)
+		sess.MarkMessage(msg, "")
+	}
+	return nil
+}