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

add rate limiter

Browse Source
This commit is contained in:
ginuerzh
2022-04-21 21:37:30 +08:00
parent a04c8b45f3
commit e23da0f319
8 changed files with 218 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
listener/tcp/listener.go
View File

@ -3,6 +3,7 @@ package tcp
import (
"net"
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"
@ -47,7 +48,8 @@ func (l *tcpListener) Init(md md.Metadata) (err error) {
return
}
l.ln = metrics.WrapListener(l.options.Service, ln)
ln = metrics.WrapListener(l.options.Service, ln)
l.ln = limiter.WrapListener(l.options.RateLimiter, ln)
return
}

2
listener/udp/listener.go
View File

@ -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,