luciano/vmware-go-kcl-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: add case for setting lastCheckTime in first iteration callGetRecordsAPI

Browse source 
Signed-off-by: Shiva Pentakota <spentakota@vmware.com>
This commit is contained in:
Shiva Pentakota 2023-01-24 09:18:55 -08:00
parent ae3763e478
commit 706e803d27
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
clientlibrary/worker/polling-shard-consumer.go
View file

@ -285,6 +285,9 @@ func (sc *PollingShardConsumer) callGetRecordsAPI(gri *kinesis.GetRecordsInput)
for _, record := range getResp.Records { for _, record := range getResp.Records {
sc.bytesRead += float64(len(record.Data)) sc.bytesRead += float64(len(record.Data))
} }
if sc.lastCheckTime.IsZero() {
sc.lastCheckTime = rateLimitTimeNow()
}
return getResp, err return getResp, err
} }