From 706e803d27f4cfdc8ff8b51802c3cf15364ace87 Mon Sep 17 00:00:00 2001
From: Shiva Pentakota <spentakota@vmware.com>
Date: Tue, 24 Jan 2023 09:18:55 -0800
Subject: [PATCH] fix: add case for setting lastCheckTime in first iteration
 callGetRecordsAPI

Signed-off-by: Shiva Pentakota <spentakota@vmware.com>
---
 clientlibrary/worker/polling-shard-consumer.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/clientlibrary/worker/polling-shard-consumer.go b/clientlibrary/worker/polling-shard-consumer.go
index 6c494a6..a35f026 100644
--- a/clientlibrary/worker/polling-shard-consumer.go
+++ b/clientlibrary/worker/polling-shard-consumer.go
@@ -285,6 +285,9 @@ func (sc *PollingShardConsumer) callGetRecordsAPI(gri *kinesis.GetRecordsInput)
 	for _, record := range getResp.Records {
 		sc.bytesRead += float64(len(record.Data))
 	}
+	if sc.lastCheckTime.IsZero() {
+		sc.lastCheckTime = rateLimitTimeNow()
+	}
 
 	return getResp, err
 }