add rate limiter

2022-04-21 21:37:30 +08:00
parent a04c8b45f3
commit e23da0f319
8 changed files with 218 additions and 23 deletions
--- a/listener/udp/listener.go
+++ b/listener/udp/listener.go
@ -4,6 +4,7 @@ import (
 	"net"

 	"github.com/go-gost/core/common/net/udp"
+	limiter "github.com/go-gost/core/limiter/wrapper"
 	"github.com/go-gost/core/listener"
 	"github.com/go-gost/core/logger"
 	md "github.com/go-gost/core/metadata"
@ -54,6 +55,7 @@ func (l *udpListener) Init(md md.Metadata) (err error) {
 		return
 	}
 	conn = metrics.WrapPacketConn(l.options.Service, conn)
+	conn = limiter.WrapPacketConn(l.options.RateLimiter, conn)

 	l.ln = udp.NewListener(conn, &udp.ListenConfig{
 		Backlog:        l.md.backlog,