85 lines
1.4 KiB
Go
85 lines
1.4 KiB
Go
package fasthttpc
|
|
|
|
import (
|
|
"github.com/valyala/fasthttp"
|
|
"golang.org/x/time/rate"
|
|
"time"
|
|
)
|
|
|
|
type (
|
|
Config struct {
|
|
timeout time.Duration
|
|
client *fasthttp.Client
|
|
noCheckStatus bool
|
|
qps int
|
|
qpsLimiter *rate.Limiter
|
|
maxConnPerHost int
|
|
fasthttpTimeout time.Duration
|
|
useCtxTimeout bool
|
|
}
|
|
|
|
ConfigOpt func(c *Config)
|
|
)
|
|
|
|
func WithTimout(v time.Duration) ConfigOpt {
|
|
return func(c *Config) {
|
|
c.timeout = v
|
|
}
|
|
}
|
|
|
|
func WithUseCtxTimeout(v bool) ConfigOpt {
|
|
return func(c *Config) {
|
|
c.useCtxTimeout = v
|
|
}
|
|
}
|
|
|
|
func WithFasthttpimout(v time.Duration) ConfigOpt {
|
|
return func(c *Config) {
|
|
c.fasthttpTimeout = v
|
|
}
|
|
}
|
|
|
|
func WithClient(v *fasthttp.Client) ConfigOpt {
|
|
return func(c *Config) {
|
|
c.client = v
|
|
}
|
|
}
|
|
|
|
func WithMaxConnPerHost(v int) ConfigOpt {
|
|
return func(c *Config) {
|
|
c.maxConnPerHost = v
|
|
}
|
|
}
|
|
|
|
//func WithRedirectFn(v func(req *http.Request, via []*http.Request) error) ConfigOpt {
|
|
// return func(c *Config) {
|
|
// c.redirectFn = v
|
|
// }
|
|
//}
|
|
//
|
|
//func WithNoRedirect() ConfigOpt {
|
|
// return func(c *Config) {
|
|
// c.redirectFn = func(req *http.Request, via []*http.Request) error {
|
|
// return http.ErrUseLastResponse
|
|
// }
|
|
// }
|
|
//}
|
|
|
|
func WithNoCheckStatus(v bool) ConfigOpt {
|
|
return func(c *Config) {
|
|
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
|
|
}
|
|
}
|