From 5c3fc2e4772dd7a23d6a0ea1bcb604b1e8e594fe Mon Sep 17 00:00:00 2001 From: Shiva Pentakota Date: Wed, 22 Mar 2023 20:46:24 -0700 Subject: [PATCH] fix: check for non-zero cooldown period Signed-off-by: Shiva Pentakota --- clientlibrary/worker/polling-shard-consumer.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/clientlibrary/worker/polling-shard-consumer.go b/clientlibrary/worker/polling-shard-consumer.go index 6e19f5b..642f431 100644 --- a/clientlibrary/worker/polling-shard-consumer.go +++ b/clientlibrary/worker/polling-shard-consumer.go @@ -187,8 +187,10 @@ func (sc *PollingShardConsumer) getRecords() error { continue } if err == maxBytesExceededError { - log.Infof("maxBytesExceededError so sleep for %+v seconds", coolDownPeriod) - time.Sleep(time.Duration(coolDownPeriod) * time.Second) + if coolDownPeriod != 0 { + log.Infof("maxBytesExceededError so sleep for %+v seconds", coolDownPeriod) + time.Sleep(time.Duration(coolDownPeriod) * time.Second) + } continue } if errors.As(err, &kmsThrottlingErr) {