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

Merge pull request #4884 from kiokuless/fix-rate-limiter

Browse Source 
Fix overriding the predefined ratelimiter by WithGlobalRateLimit
This commit is contained in:
Sandeep Singh 2024-03-15 15:26:05 +05:30 committed by GitHub
commit aaf3cf9cb3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/sdk_private.go
View File

@ -186,6 +186,7 @@ func (e *NucleiEngine) init() error {
} }
} }
if e.executerOpts.RateLimiter == nil {
if e.opts.RateLimitMinute > 0 { if e.opts.RateLimitMinute > 0 {
e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimitMinute), time.Minute) e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimitMinute), time.Minute)
} else if e.opts.RateLimit > 0 { } else if e.opts.RateLimit > 0 {
@ -193,6 +194,7 @@ func (e *NucleiEngine) init() error {
} else { } else {
e.executerOpts.RateLimiter = ratelimit.NewUnlimited(context.Background()) e.executerOpts.RateLimiter = ratelimit.NewUnlimited(context.Background())
} }
}
e.engine = core.New(e.opts) e.engine = core.New(e.opts)
e.engine.SetExecuterOptions(e.executerOpts) e.engine.SetExecuterOptions(e.executerOpts)