cyonjan/frp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

prometheus: add proxy status

Browse Source
This commit is contained in:
Nguyendream 2021-05-28 16:41:44 +08:00
parent 9ae322cccf
commit c275529b5c
No known key found for this signature in database
GPG Key ID: 6E0C68F41A97EE47
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/metrics/prometheus/server.go
View File

@ -19,6 +19,7 @@ type serverMetrics struct {
connectionCount *prometheus.GaugeVec connectionCount *prometheus.GaugeVec
trafficIn *prometheus.CounterVec trafficIn *prometheus.CounterVec
trafficOut *prometheus.CounterVec trafficOut *prometheus.CounterVec
status *prometheus.GaugeVec
} }
func (m *serverMetrics) NewClient() { func (m *serverMetrics) NewClient() {
@ -31,10 +32,12 @@ func (m *serverMetrics) CloseClient() {
func (m *serverMetrics) NewProxy(name string, proxyType string) { func (m *serverMetrics) NewProxy(name string, proxyType string) {
m.proxyCount.WithLabelValues(proxyType).Inc() m.proxyCount.WithLabelValues(proxyType).Inc()
m.status.WithLabelValues(name, proxyType).Set(1)
} }
func (m *serverMetrics) CloseProxy(name string, proxyType string) { func (m *serverMetrics) CloseProxy(name string, proxyType string) {
m.proxyCount.WithLabelValues(proxyType).Dec() m.proxyCount.WithLabelValues(proxyType).Dec()
m.status.WithLabelValues(name, proxyType).Set(0)
} }
func (m *serverMetrics) OpenConnection(name string, proxyType string) { func (m *serverMetrics) OpenConnection(name string, proxyType string) {
@ -85,11 +88,18 @@ func newServerMetrics() *serverMetrics {
Name: "traffic_out", Name: "traffic_out",
Help: "The total out traffic", Help: "The total out traffic",
}, []string{"name", "type"}), }, []string{"name", "type"}),
status: prometheus.NewGaugeVec(prometheus.GaugeOpts{
Namespace: namespace,
Subsystem: serverSubsystem,
Name: "status",
Help: "The proxy status",
}, []string{"name", "type"}),
} }
prometheus.MustRegister(m.clientCount) prometheus.MustRegister(m.clientCount)
prometheus.MustRegister(m.proxyCount) prometheus.MustRegister(m.proxyCount)
prometheus.MustRegister(m.connectionCount) prometheus.MustRegister(m.connectionCount)
prometheus.MustRegister(m.trafficIn) prometheus.MustRegister(m.trafficIn)
prometheus.MustRegister(m.trafficOut) prometheus.MustRegister(m.trafficOut)
prometheus.MustRegister(m.status)
return m return m
} }