26 lines
669 B
Go
26 lines
669 B
Go
package flaresolverr
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"resty.dev/v3"
|
|
)
|
|
|
|
func NewRestyClient() *resty.Client {
|
|
restyClient := resty.New()
|
|
restyClient.SetRetryCount(3).SetRetryWaitTime(3 * time.Second).AddRetryConditions(
|
|
func(r *resty.Response, err error) bool {
|
|
return err != nil || r.StatusCode() == http.StatusTooManyRequests
|
|
},
|
|
)
|
|
restyClient.SetLogger(disableLogger{}).SetHeader("Accept-Charset", "utf-8")
|
|
return restyClient
|
|
}
|
|
|
|
type disableLogger struct{}
|
|
|
|
func (d disableLogger) Errorf(format string, v ...interface{}) {}
|
|
func (d disableLogger) Warnf(format string, v ...interface{}) {}
|
|
func (d disableLogger) Debugf(format string, v ...interface{}) {}
|