From 3a41f752e437e7e0db8466134544bd0338272b04 Mon Sep 17 00:00:00 2001 From: kiokuless Date: Wed, 13 Mar 2024 18:20:55 +0900 Subject: [PATCH] Use e.rateLimiter if it's not nil --- lib/sdk_private.go | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/lib/sdk_private.go b/lib/sdk_private.go index a9a2b4bef..9ce074d23 100644 --- a/lib/sdk_private.go +++ b/lib/sdk_private.go @@ -159,12 +159,14 @@ func (e *NucleiEngine) init() error { Browser: e.browserInstance, } - if e.opts.RateLimitMinute > 0 { - e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimitMinute), time.Minute) - } else if e.opts.RateLimit > 0 { - e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimit), time.Second) - } else { - e.executerOpts.RateLimiter = ratelimit.NewUnlimited(context.Background()) + if e.executerOpts.RateLimiter == nil { + if e.opts.RateLimitMinute > 0 { + e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimitMinute), time.Minute) + } else if e.opts.RateLimit > 0 { + e.executerOpts.RateLimiter = ratelimit.New(context.Background(), uint(e.opts.RateLimit), time.Second) + } else { + e.executerOpts.RateLimiter = ratelimit.NewUnlimited(context.Background()) + } } e.engine = core.New(e.opts)