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

2
pkg/listener/http2/h2/listener.go
View File

@ -7,6 +7,7 @@ import (
"net/http"
"net/http/httputil"
"github.com/go-gost/gost/pkg/common/metrics"
"github.com/go-gost/gost/pkg/listener"
"github.com/go-gost/gost/pkg/logger"
md "github.com/go-gost/gost/pkg/metadata"
@ -68,6 +69,7 @@ func (l *h2Listener) Init(md md.Metadata) (err error) {
return err
}
l.addr = ln.Addr()
ln = metrics.WrapListener(l.options.Service, ln)
if l.h2c {
l.server.Handler = h2c.NewHandler(

7
pkg/listener/http2/listener.go
View File

@ -5,7 +5,7 @@ import (
"net"
"net/http"
"github.com/go-gost/gost/pkg/common/util"
"github.com/go-gost/gost/pkg/common/metrics"
http2_util "github.com/go-gost/gost/pkg/internal/util/http2"
"github.com/go-gost/gost/pkg/listener"
"github.com/go-gost/gost/pkg/logger"
@ -58,11 +58,10 @@ func (l *http2Listener) Init(md md.Metadata) (err error) {
return err
}
l.addr = ln.Addr()
ln = metrics.WrapListener(l.options.Service, ln)
ln = tls.NewListener(
&util.TCPKeepAliveListener{
TCPListener: ln.(*net.TCPListener),
},
ln,
l.options.TLSConfig,
)