errcode

internal/push/init.go

@@ -0,0 +1,60 @@
+/*
+** description("").
+** copyright('open-im,www.open-im.io').
+** author("fg,Gordon@open-im.io").
+** time(2021/3/22 15:33).
+ */
+package push
+
+import (
+	fcm "Open_IM/internal/push/fcm"
+	"Open_IM/internal/push/getui"
+	jpush "Open_IM/internal/push/jpush"
+	"Open_IM/pkg/common/config"
+	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/prome"
+	"Open_IM/pkg/statistics"
+	"fmt"
+)
+
+var (
+	rpcServer     RPCServer
+	pushCh        ConsumerHandler
+	offlinePusher OfflinePusher
+	successCount  uint64
+)
+
+func Init(rpcPort int) {
+	rpcServer.Init(rpcPort)
+	pushCh.Init()
+
+}
+func init() {
+	statistics.NewStatistics(&successCount, config.Config.ModuleName.PushName, fmt.Sprintf("%d second push to msg_gateway count", constant.StatisticsTimeInterval), constant.StatisticsTimeInterval)
+	if *config.Config.Push.Getui.Enable {
+		offlinePusher = getui.GetuiClient
+	}
+	if config.Config.Push.Jpns.Enable {
+		offlinePusher = jpush.JPushClient
+	}
+
+	if config.Config.Push.Fcm.Enable {
+		offlinePusher = fcm.NewFcm()
+	}
+}
+
+func initPrometheus() {
+	prome.NewMsgOfflinePushSuccessCounter()
+	prome.NewMsgOfflinePushFailedCounter()
+}
+
+func Run(promethuesPort int) {
+	go rpcServer.run()
+	go pushCh.ConsumerGroup.RegisterHandleAndConsumer(&pushCh)
+	go func() {
+		err := prome.StartPromeSrv(promethuesPort)
+		if err != nil {
+			panic(err)
+		}
+	}()
+}