From 5ce0cf1aad6c0ed56b644094746e9ae8383cfa24 Mon Sep 17 00:00:00 2001
From: wangchuxiao <wangchuxiao97@outlook.com>
Date: Wed, 14 Sep 2022 18:16:02 +0800
Subject: [PATCH] prome

---
 internal/rpc/auth/auth.go         | 22 ++++------------------
 pkg/common/prometheus/constant.go | 26 ++++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 18 deletions(-)
 create mode 100644 pkg/common/prometheus/constant.go

diff --git a/internal/rpc/auth/auth.go b/internal/rpc/auth/auth.go
index 0947cb4f9..3f6a957db 100644
--- a/internal/rpc/auth/auth.go
+++ b/internal/rpc/auth/auth.go
@@ -17,28 +17,14 @@ import (
 	"strconv"
 	"strings"
 
-	"github.com/prometheus/client_golang/prometheus"
-	"github.com/prometheus/client_golang/prometheus/promauto"
-
 	"Open_IM/pkg/common/config"
 
 	"google.golang.org/grpc"
 )
 
-var (
-	userLoginCounter    prometheus.Counter
-	userRegisterCounter prometheus.Counter
-)
-
 func (rpc *rpcAuth) initPrometheus() {
-	userLoginCounter = promauto.NewCounter(prometheus.CounterOpts{
-		Name: "user_login",
-		Help: "The number of user login",
-	})
-	userRegisterCounter = promauto.NewCounter(prometheus.CounterOpts{
-		Name: "user_register",
-		Help: "The number of user register",
-	})
+	promePkg.NewUserLoginCounter()
+	promePkg.NewUserRegisterCounter()
 }
 
 func (rpc *rpcAuth) UserRegister(_ context.Context, req *pbAuth.UserRegisterReq) (*pbAuth.UserRegisterResp, error) {
@@ -55,7 +41,7 @@ func (rpc *rpcAuth) UserRegister(_ context.Context, req *pbAuth.UserRegisterReq)
 		log.NewError(req.OperationID, errMsg, user)
 		return &pbAuth.UserRegisterResp{CommonResp: &pbAuth.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: errMsg}}, nil
 	}
-	promePkg.PromeInc(userRegisterCounter)
+	promePkg.PromeInc(promePkg.UserRegisterCounter)
 	log.NewInfo(req.OperationID, utils.GetSelfFuncName(), " rpc return ", pbAuth.UserRegisterResp{CommonResp: &pbAuth.CommonResp{}})
 	return &pbAuth.UserRegisterResp{CommonResp: &pbAuth.CommonResp{}}, nil
 }
@@ -68,7 +54,7 @@ func (rpc *rpcAuth) UserToken(_ context.Context, req *pbAuth.UserTokenReq) (*pbA
 		log.NewError(req.OperationID, errMsg)
 		return &pbAuth.UserTokenResp{CommonResp: &pbAuth.CommonResp{ErrCode: constant.ErrDB.ErrCode, ErrMsg: errMsg}}, nil
 	}
-	promePkg.PromeInc(userLoginCounter)
+	promePkg.PromeInc(promePkg.UserLoginCounter)
 	log.NewInfo(req.OperationID, utils.GetSelfFuncName(), " rpc return ", pbAuth.UserTokenResp{CommonResp: &pbAuth.CommonResp{}, Token: tokens, ExpiredTime: expTime})
 	return &pbAuth.UserTokenResp{CommonResp: &pbAuth.CommonResp{}, Token: tokens, ExpiredTime: expTime}, nil
 }
diff --git a/pkg/common/prometheus/constant.go b/pkg/common/prometheus/constant.go
new file mode 100644
index 000000000..9ced5908d
--- /dev/null
+++ b/pkg/common/prometheus/constant.go
@@ -0,0 +1,26 @@
+package prometheus
+
+import (
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promauto"
+)
+
+var (
+	// user rpc
+	UserLoginCounter    prometheus.Counter
+	UserRegisterCounter prometheus.Counter
+)
+
+func NewUserLoginCounter() {
+	UserLoginCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Name: "user_login",
+		Help: "The number of user login",
+	})
+}
+
+func NewUserRegisterCounter() {
+	UserRegisterCounter = promauto.NewCounter(prometheus.CounterOpts{
+		Name: "user_register",
+		Help: "The number of user register",
+	})
+}