zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(metrics): enable prometheus global switch when user enable devsever metrics (#3169)

Browse Source
This commit is contained in:
cong
2023-04-24 20:02:04 +08:00
committed by GitHub
parent 8c9b619199
commit ba9d510cdb
2 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
core/prometheus/agent.go
View File

@@ -21,6 +21,11 @@ func Enabled() bool {
return enabled.True() return enabled.True()
} }
// Enable enables prometheus.
func Enable() {
enabled.Set(true)
}
// StartAgent starts a prometheus agent. // StartAgent starts a prometheus agent.
func StartAgent(c Config) { func StartAgent(c Config) {
if len(c.Host) == 0 { if len(c.Host) == 0 {

3
internal/devserver/server.go
View File

@@ -10,6 +10,7 @@ import (
"github.com/felixge/fgprof" "github.com/felixge/fgprof"
"github.com/prometheus/client_golang/prometheus/promhttp" "github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/zeromicro/go-zero/core/logx" "github.com/zeromicro/go-zero/core/logx"
"github.com/zeromicro/go-zero/core/prometheus"
"github.com/zeromicro/go-zero/core/threading" "github.com/zeromicro/go-zero/core/threading"
"github.com/zeromicro/go-zero/internal/health" "github.com/zeromicro/go-zero/internal/health"
) )
@@ -42,6 +43,8 @@ func (s *Server) addRoutes() {
// metrics // metrics
if s.config.EnableMetrics { if s.config.EnableMetrics {
// enable prometheus global switch
prometheus.Enable()
s.handleFunc(s.config.MetricsPath, promhttp.Handler().ServeHTTP) s.handleFunc(s.config.MetricsPath, promhttp.Handler().ServeHTTP)
} }
// pprof // pprof