update

myhttp/fasthttpc/option.go

@@ -0,0 +1,84 @@
+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
+	}
+}