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/udp/listener.go
+++ b/pkg/listener/udp/listener.go
@ -43,13 +43,15 @@ func (l *udpListener) Init(md md.Metadata) (err error) {
 		return
 	}

-	conn, err := net.ListenUDP("udp", laddr)
+	var conn net.PacketConn
+	conn, err = net.ListenUDP("udp", laddr)
 	if err != nil {
 		return
 	}
+	conn = metrics.WrapPacketConn(l.options.Service, conn)

 	l.Listener = udp.NewListener(
-		metrics.WrapPacketConn(l.options.Service, conn),
+		conn,
 		laddr,
 		l.md.backlog,
 		l.md.readQueueSize, l.md.readBufferSize,