This commit is contained in:
kzkzzzz
2025-07-27 10:08:41 +08:00
parent eb6557e0d8
commit 781f79cf7c
2 changed files with 47 additions and 2 deletions

View File

@@ -1,6 +1,7 @@
package httpc
import (
"golang.org/x/time/rate"
"net/http"
"time"
)
@@ -12,6 +13,8 @@ type (
transport *http.Transport
redirectFn func(req *http.Request, via []*http.Request) error
noCheckStatus bool
qps int
qpsLimiter *rate.Limiter
}
ConfigOpt func(c *Config)
@@ -54,3 +57,15 @@ func WithNoCheckStatus(v bool) ConfigOpt {
c.noCheckStatus = v
}
}
func WithQps(v int) ConfigOpt {
return func(c *Config) {
c.qps = v
}
}
func WithQpsLimiter(v *rate.Limiter) ConfigOpt {
return func(c *Config) {
c.qpsLimiter = v
}
}