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

add metrics for service

Browse Source
This commit is contained in:
ginuerzh
2022-03-05 00:28:13 +08:00
parent 8d8785f534
commit e587b4df7c
34 changed files with 548 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pkg/listener/ws/listener.go
View File

@ -6,6 +6,7 @@ import (
"net/http"
"net/http/httputil"
"github.com/go-gost/gost/pkg/common/metrics"
ws_util "github.com/go-gost/gost/pkg/internal/util/ws"
"github.com/go-gost/gost/pkg/listener"
"github.com/go-gost/gost/pkg/logger"
@ -82,6 +83,8 @@ func (l *wsListener) Init(md md.Metadata) (err error) {
if err != nil {
return
}
ln = metrics.WrapListener(l.options.Service, ln)
if l.tlsEnabled {
ln = tls.NewListener(ln, l.options.TLSConfig)
}

3
pkg/listener/ws/mux/listener.go
View File

@ -6,6 +6,7 @@ import (
"net/http"
"net/http/httputil"
"github.com/go-gost/gost/pkg/common/metrics"
ws_util "github.com/go-gost/gost/pkg/internal/util/ws"
"github.com/go-gost/gost/pkg/listener"
"github.com/go-gost/gost/pkg/logger"
@ -87,6 +88,8 @@ func (l *mwsListener) Init(md md.Metadata) (err error) {
if err != nil {
return
}
ln = metrics.WrapListener(l.options.Service, ln)
if l.tlsEnabled {
ln = tls.NewListener(ln, l.options.TLSConfig)
}