send group message split topic

internal/msg_transfer/logic/init.go

@@ -2,20 +2,36 @@ package logic
 
 import (
 	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
 	"Open_IM/pkg/common/kafka"
+	"Open_IM/pkg/statistics"
 	"fmt"
+	"sync"
 )
 
+const OnlineTopicBusy = "Busy"
+const OnlineTopicVacancy = "Vacancy"
+
 var (
-	persistentCH PersistentConsumerHandler
-	historyCH    HistoryConsumerHandler
-	producer     *kafka.Producer
+	persistentCH          PersistentConsumerHandler
+	historyCH             OnlineHistoryConsumerHandler
+	offlineHistoryCH      OfflineHistoryConsumerHandler
+	producer              *kafka.Producer
+	cmdCh                 chan Cmd2Value
+	onlineTopicStatus     string
+	w                     *sync.Mutex
+	singleMsgSuccessCount uint64
+	groupMsgCount         uint64
+	singleMsgFailedCount  uint64
 )
 
 func Init() {
-
+	cmdCh = make(chan Cmd2Value, 10000)
 	persistentCH.Init()
-	historyCH.Init()
+	historyCH.Init(cmdCh)
+	offlineHistoryCH.Init(cmdCh)
+	statistics.NewStatistics(&singleMsgSuccessCount, config.Config.ModuleName.MsgTransferName, fmt.Sprintf("%d second singleMsgCount insert to mongo", constant.StatisticsTimeInterval), constant.StatisticsTimeInterval)
+	statistics.NewStatistics(&groupMsgCount, config.Config.ModuleName.MsgTransferName, fmt.Sprintf("%d second groupMsgCount insert to mongo", constant.StatisticsTimeInterval), constant.StatisticsTimeInterval)
 	producer = kafka.NewKafkaProducer(config.Config.Kafka.Ms2pschat.Addr, config.Config.Kafka.Ms2pschat.Topic)
 }
 func Run() {
@@ -26,4 +42,15 @@ func Run() {
 		fmt.Println("not start mysql consumer")
 	}
 	go historyCH.historyConsumerGroup.RegisterHandleAndConsumer(&historyCH)
+	go offlineHistoryCH.historyConsumerGroup.RegisterHandleAndConsumer(&offlineHistoryCH)
+}
+func SetOnlineTopicStatus(status string) {
+	w.Lock()
+	defer w.Unlock()
+	onlineTopicStatus = status
+}
+func GetOnlineTopicStatus() string {
+	w.Lock()
+	defer w.Unlock()
+	return onlineTopicStatus
 }