External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-18 06:15:27 +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

14
lib/sdk_private.go
View File

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