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

add admission option for listener

Browse Source
This commit is contained in:
ginuerzh
2022-03-13 00:08:16 +08:00
parent 4007e80811
commit bfc1f8472c
22 changed files with 348 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,6 +6,7 @@ import (
"net/http"
"net/http/httputil"
"github.com/go-gost/gost/pkg/common/admission"
"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"
@ -84,6 +85,7 @@ func (l *wsListener) Init(md md.Metadata) (err error) {
return
}
ln = metrics.WrapListener(l.options.Service, ln)
ln = admission.WrapListener(l.options.Admission, ln)
if l.tlsEnabled {
ln = tls.NewListener(ln, l.options.TLSConfig)

2
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/admission"
"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"
@ -89,6 +90,7 @@ func (l *mwsListener) Init(md md.Metadata) (err error) {
return
}
ln = metrics.WrapListener(l.options.Service, ln)
ln = admission.WrapListener(l.options.Admission, ln)
if l.tlsEnabled {
ln = tls.NewListener(ln, l.options.TLSConfig)