123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293 |
- package aggregate
- import (
- "github.com/fatedier/frp/pkg/metrics/mem"
- "github.com/fatedier/frp/pkg/metrics/prometheus"
- "github.com/fatedier/frp/server/metrics"
- )
- func EnableMem() {
- sm.Add(mem.ServerMetrics)
- }
- func EnablePrometheus() {
- sm.Add(prometheus.ServerMetrics)
- }
- var sm = &serverMetrics{}
- func init() {
- metrics.Register(sm)
- }
- type serverMetrics struct {
- ms []metrics.ServerMetrics
- }
- func (m *serverMetrics) Add(sm metrics.ServerMetrics) {
- m.ms = append(m.ms, sm)
- }
- func (m *serverMetrics) NewClient() {
- for _, v := range m.ms {
- v.NewClient()
- }
- }
- func (m *serverMetrics) CloseClient() {
- for _, v := range m.ms {
- v.CloseClient()
- }
- }
- func (m *serverMetrics) NewProxy(name string, proxyType string) {
- for _, v := range m.ms {
- v.NewProxy(name, proxyType)
- }
- }
- func (m *serverMetrics) CloseProxy(name string, proxyType string) {
- for _, v := range m.ms {
- v.CloseProxy(name, proxyType)
- }
- }
- func (m *serverMetrics) OpenConnection(name string, proxyType string) {
- for _, v := range m.ms {
- v.OpenConnection(name, proxyType)
- }
- }
- func (m *serverMetrics) CloseConnection(name string, proxyType string) {
- for _, v := range m.ms {
- v.CloseConnection(name, proxyType)
- }
- }
- func (m *serverMetrics) AddTrafficIn(name string, proxyType string, trafficBytes int64) {
- for _, v := range m.ms {
- v.AddTrafficIn(name, proxyType, trafficBytes)
- }
- }
- func (m *serverMetrics) AddTrafficOut(name string, proxyType string, trafficBytes int64) {
- for _, v := range m.ms {
- v.AddTrafficOut(name, proxyType, trafficBytes)
- }
- }
|