From 0c28a12ab47450b3e8be2afef4451acdf388faea Mon Sep 17 00:00:00 2001 From: "Bo-Yi.Wu" Date: Sun, 22 Jan 2023 08:20:20 +0800 Subject: [PATCH 1/2] chore(options): update default capacity as zero Signed-off-by: Bo-Yi.Wu --- options.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/options.go b/options.go index bdbf981..b1373de 100644 --- a/options.go +++ b/options.go @@ -8,7 +8,7 @@ import ( ) var ( - defaultQueueSize = 4096 + defaultCapacity = 0 defaultWorkerCount = runtime.NumCPU() defaultNewLogger = NewLogger() defaultFn = func(context.Context, core.QueuedMessage) error { return nil } @@ -87,7 +87,7 @@ type Options struct { func NewOptions(opts ...Option) *Options { o := &Options{ workerCount: defaultWorkerCount, - queueSize: defaultQueueSize, + queueSize: defaultCapacity, logger: defaultNewLogger, worker: nil, fn: defaultFn, From dda7b8acc8a4ce774812a11ea71b75a41da3b69c Mon Sep 17 00:00:00 2001 From: "Bo-Yi.Wu" Date: Sun, 22 Jan 2023 08:26:10 +0800 Subject: [PATCH 2/2] chore: update default consumer. Signed-off-by: Bo-Yi.Wu --- consumer.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/consumer.go b/consumer.go index 3afb41e..f5db326 100644 --- a/consumer.go +++ b/consumer.go @@ -52,7 +52,7 @@ func (s *Consumer) Queue(task core.QueuedMessage) error { //nolint:stylecheck if atomic.LoadInt32(&s.stopFlag) == 1 { return ErrQueueShutdown } - if s.count >= s.capacity { + if s.capacity > 0 && s.count >= s.capacity { return errMaxCapacity }