add conn limiter

This commit is contained in:
ginuerzh
2022-09-14 00:15:19 +08:00
parent 999707db19
commit 91c12882f5
48 changed files with 1041 additions and 244 deletions

View File

@ -10,7 +10,8 @@ import (
md "github.com/go-gost/core/metadata"
admission "github.com/go-gost/x/admission/wrapper"
xnet "github.com/go-gost/x/internal/net"
limiter "github.com/go-gost/x/limiter/wrapper"
climiter "github.com/go-gost/x/limiter/conn/wrapper"
limiter "github.com/go-gost/x/limiter/traffic/wrapper"
metrics "github.com/go-gost/x/metrics/wrapper"
"github.com/go-gost/x/registry"
)
@ -79,7 +80,8 @@ func (l *rtcpListener) Accept() (conn net.Conn, err error) {
}
l.ln = metrics.WrapListener(l.options.Service, l.ln)
l.ln = admission.WrapListener(l.options.Admission, l.ln)
l.ln = limiter.WrapListener(l.options.RateLimiter, l.ln)
l.ln = limiter.WrapListener(l.options.TrafficLimiter, l.ln)
l.ln = climiter.WrapListener(l.options.ConnLimiter, l.ln)
}
conn, err = l.ln.Accept()
if err != nil {