diff --git a/main.go b/main.go index 90aa3aa..6b70e26 100644 --- a/main.go +++ b/main.go @@ -96,6 +96,12 @@ func LimitedHttp(rps float64, burst int) *http.Client { } } +var client = http.DefaultClient + +func WithClient(nclient *http.Client) { + client = nclient +} + func RequestCached[T any](req *http.Request, filename string) (T, error) { var zero T @@ -105,7 +111,7 @@ func RequestCached[T any](req *http.Request, filename string) (T, error) { return zero, fmt.Errorf("failed to read cache: %w", err) } - resp, httpErr := http.DefaultClient.Do(req) + resp, httpErr := client.Do(req) if httpErr != nil { return zero, fmt.Errorf("HTTP request failed: %w", httpErr) }