External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-22 20:05:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

retry with 504 status code (timeouts) (#2852)

Browse Source 
* Incase of 504/timeouts, do retry

* Update the version and pass the custom policy from outside
This commit is contained in:
Jaideep Khandelwal 2022-12-12 16:43:21 +05:30 committed by GitHub
parent 8a75cb2574
commit 19b56570b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
v2/internal/runner/nucleicloud/cloud.go
View File

@ -2,6 +2,7 @@ package nucleicloud
import ( import (
"bytes" "bytes"
"context"
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
@ -27,10 +28,21 @@ const (
resultSize = 100 resultSize = 100
) )
// HTTPErrorRetryPolicy is to retry for HTTPCodes >= 500.
func HTTPErrorRetryPolicy() func(ctx context.Context, resp *http.Response, err error) (bool, error) {
return func(ctx context.Context, resp *http.Response, err error) (bool, error) {
if resp.StatusCode >= http.StatusInternalServerError {
return true, errors.New(resp.Status)
}
return retryablehttp.CheckRecoverableErrors(ctx, resp, err)
}
}
// New returns a nuclei-cloud API client // New returns a nuclei-cloud API client
func New(baseURL, apiKey string) *Client { func New(baseURL, apiKey string) *Client {
options := retryablehttp.DefaultOptionsSingle options := retryablehttp.DefaultOptionsSingle
options.Timeout = 15 * time.Second options.Timeout = 15 * time.Second
options.CheckRetry = HTTPErrorRetryPolicy()
client := retryablehttp.NewClient(options) client := retryablehttp.NewClient(options)
baseAppURL := baseURL baseAppURL := baseURL