From e1bdbc47529590939d4bb353d8306197f286e89d Mon Sep 17 00:00:00 2001
From: "lin.huang" <lin.huang@apulis.com>
Date: Mon, 6 Nov 2023 14:59:11 +0800
Subject: [PATCH] fix:prometheus enable function

---
 internal/msgtransfer/init.go | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/internal/msgtransfer/init.go b/internal/msgtransfer/init.go
index c3e5b8352..c18186fa8 100644
--- a/internal/msgtransfer/init.go
+++ b/internal/msgtransfer/init.go
@@ -118,13 +118,15 @@ func (m *MsgTransfer) Start(prometheusPort int) error {
 		return err
 	}*/
 	////////////////////////////
-	reg := prometheus.NewRegistry()
-	reg.MustRegister(
-		collectors.NewGoCollector(),
-	)
-	reg.MustRegister(prom_metrics.GetGrpcCusMetrics("Transfer")...)
-	http.Handle("/metrics", promhttp.HandlerFor(reg, promhttp.HandlerOpts{Registry: reg}))
-	log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", prometheusPort), nil))
+	if config.Config.Prometheus.Enable {
+		reg := prometheus.NewRegistry()
+		reg.MustRegister(
+			collectors.NewGoCollector(),
+		)
+		reg.MustRegister(prom_metrics.GetGrpcCusMetrics("Transfer")...)
+		http.Handle("/metrics", promhttp.HandlerFor(reg, promhttp.HandlerOpts{Registry: reg}))
+		log.Fatal(http.ListenAndServe(fmt.Sprintf(":%d", prometheusPort), nil))
+	}
 	////////////////////////////////////////
 	wg.Wait()
 	return nil