add traffic limiter option for handler

This commit is contained in:
ginuerzh
2023-11-18 18:25:40 +08:00
parent a916f04016
commit 486f2cee61
2 changed files with 44 additions and 3 deletions

View File

@ -8,6 +8,7 @@ import (
"github.com/go-gost/core/bypass"
"github.com/go-gost/core/chain"
"github.com/go-gost/core/limiter/rate"
"github.com/go-gost/core/limiter/traffic"
"github.com/go-gost/core/logger"
"github.com/go-gost/core/metadata"
)
@ -18,6 +19,7 @@ type Options struct {
Auth *url.Userinfo
Auther auth.Authenticator
RateLimiter rate.RateLimiter
Limiter traffic.TrafficLimiter
TLSConfig *tls.Config
Logger logger.Logger
Service string
@ -55,6 +57,12 @@ func RateLimiterOption(limiter rate.RateLimiter) Option {
}
}
func TrafficLimiterOption(limiter traffic.TrafficLimiter) Option {
return func(opts *Options) {
opts.Limiter = limiter
}
}
func TLSConfigOption(tlsConfig *tls.Config) Option {
return func(opts *Options) {
opts.TLSConfig = tlsConfig