GuanM/hysteria
1
0
Fork 0
mirror of https://github.com/cedar2025/hysteria.git synced 2025-09-01 14:11:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: provide a TrafficCounter interface on server, remove prometheus client dependency from core

Browse Source
This commit is contained in:
tobyxdd
2023-01-07 13:59:30 -08:00
parent 29459d768d
commit f7dffd027f
6 changed files with 105 additions and 490 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
app/cmd/server.go
View File

@@ -192,9 +192,10 @@ func server(config *serverConfig) {
aclEngine.DefaultAction = acl.ActionDirect
}
// Prometheus
var promReg *prometheus.Registry
var trafficCounter cs.TrafficCounter
if len(config.PrometheusListen) > 0 {
promReg = prometheus.NewRegistry()
promReg := prometheus.NewRegistry()
trafficCounter = NewPrometheusTrafficCounter(promReg)
go func() {
http.Handle("/metrics", promhttp.HandlerFor(promReg, promhttp.HandlerOpts{}))
err := http.ListenAndServe(config.PrometheusListen, nil)
@@ -218,7 +219,7 @@ func server(config *serverConfig) {
up, down, _ := config.Speed()
server, err := cs.NewServer(tlsConfig, quicConfig, pktConn,
transport.DefaultServerTransport, up, down, config.DisableUDP, aclEngine,
connectFunc, disconnectFunc, tcpRequestFunc, tcpErrorFunc, udpRequestFunc, udpErrorFunc, promReg)
connectFunc, disconnectFunc, tcpRequestFunc, tcpErrorFunc, udpRequestFunc, udpErrorFunc, trafficCounter)
if err != nil {
logrus.WithField("error", err).Fatal("Failed to initialize server")
}