From 6a83403e57c4538b3d3d43a7be432cc484c9bb60 Mon Sep 17 00:00:00 2001 From: "lin.huang" Date: Thu, 2 Nov 2023 18:29:56 +0800 Subject: [PATCH] fix bug:change registerName logic in GetGrpcCusMetrics function --- pkg/common/prom_metrics/func.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/pkg/common/prom_metrics/func.go b/pkg/common/prom_metrics/func.go index a46ea28dd..e451c441b 100644 --- a/pkg/common/prom_metrics/func.go +++ b/pkg/common/prom_metrics/func.go @@ -2,6 +2,7 @@ package prom_metrics import ( grpc_prometheus "github.com/grpc-ecosystem/go-grpc-prometheus" + config2 "github.com/openimsdk/open-im-server/v3/pkg/common/config" "github.com/openimsdk/open-im-server/v3/pkg/common/ginPrometheus" "github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus/collectors" @@ -19,15 +20,15 @@ func NewGrpcPromObj(cusMetrics []prometheus.Collector) (*prometheus.Registry, *g func GetGrpcCusMetrics(registerName string) []prometheus.Collector { switch registerName { - case "MessageGateway": + case config2.Config.RpcRegisterName.OpenImMessageGatewayName: return []prometheus.Collector{OnlineUserGauge} - case "Msg": + case config2.Config.RpcRegisterName.OpenImMsgName: return []prometheus.Collector{SingleChatMsgProcessSuccessCounter, SingleChatMsgProcessFailedCounter, GroupChatMsgProcessSuccessCounter, GroupChatMsgProcessFailedCounter} case "Transfer": return []prometheus.Collector{MsgInsertRedisSuccessCounter, MsgInsertRedisFailedCounter, MsgInsertMongoSuccessCounter, MsgInsertMongoFailedCounter, SeqSetFailedCounter} - case "Push": + case config2.Config.RpcRegisterName.OpenImPushName: return []prometheus.Collector{MsgOfflinePushFailedCounter} - case "Auth": + case config2.Config.RpcRegisterName.OpenImAuthName: return []prometheus.Collector{UserLoginCounter} default: return nil