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

use sync.Once

Browse Source
This commit is contained in:
Doğan Can Bakır 2024-08-21 11:26:17 +03:00
parent 50023428d0
commit 46782ff90c
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/protocols/http/httpclientpool/clientpool.go
View File

@ -29,10 +29,10 @@ import (
var ( var (
rawHttpClient *rawhttp.Client rawHttpClient *rawhttp.Client
rawHttpClientOnce sync.Once
forceMaxRedirects int forceMaxRedirects int
normalClient *retryablehttp.Client normalClient *retryablehttp.Client
clientPool *mapsutil.SyncLockMap[string, *retryablehttp.Client] clientPool *mapsutil.SyncLockMap[string, *retryablehttp.Client]
rawHttpClientMu sync.Mutex
) )
// Init initializes the clientpool implementation // Init initializes the clientpool implementation
@ -148,10 +148,7 @@ func (c *Configuration) HasStandardOptions() bool {
// GetRawHTTP returns the rawhttp request client // GetRawHTTP returns the rawhttp request client
func GetRawHTTP(options *protocols.ExecutorOptions) *rawhttp.Client { func GetRawHTTP(options *protocols.ExecutorOptions) *rawhttp.Client {
rawHttpClientMu.Lock() rawHttpClientOnce.Do(func() {
defer rawHttpClientMu.Unlock()
if rawHttpClient == nil {
rawHttpOptions := rawhttp.DefaultOptions rawHttpOptions := rawhttp.DefaultOptions
if types.ProxyURL != "" { if types.ProxyURL != "" {
rawHttpOptions.Proxy = types.ProxyURL rawHttpOptions.Proxy = types.ProxyURL
@ -162,7 +159,7 @@ func GetRawHTTP(options *protocols.ExecutorOptions) *rawhttp.Client {
} }
rawHttpOptions.Timeout = options.Options.GetTimeouts().HttpTimeout rawHttpOptions.Timeout = options.Options.GetTimeouts().HttpTimeout
rawHttpClient = rawhttp.NewClient(rawHttpOptions) rawHttpClient = rawhttp.NewClient(rawHttpOptions)
} })
return rawHttpClient return rawHttpClient
} }