add admission option for listener

2022-03-13 00:08:16 +08:00
parent 4007e80811
commit bfc1f8472c
22 changed files with 348 additions and 5 deletions
--- a/pkg/listener/grpc/listener.go
+++ b/pkg/listener/grpc/listener.go
@ -3,6 +3,7 @@ package grpc
 import (
 	"net"

+	"github.com/go-gost/gost/pkg/common/admission"
 	"github.com/go-gost/gost/pkg/common/metrics"
 	pb "github.com/go-gost/gost/pkg/common/util/grpc/proto"
 	"github.com/go-gost/gost/pkg/listener"
@ -48,6 +49,7 @@ func (l *grpcListener) Init(md md.Metadata) (err error) {
 		return
 	}
 	ln = metrics.WrapListener(l.options.Service, ln)
+	ln = admission.WrapListener(l.options.Admission, ln)

 	var opts []grpc.ServerOption
 	if !l.md.insecure {