luciano/kinesis-consumer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

#19 fixes retry on kinesis error

Browse source
This commit is contained in:
Alex Senger 2024-04-19 13:32:53 +02:00
parent 2d05606a38
commit ca7a109d1b
No known key found for this signature in database
GPG key ID: 0B4A96F8AF6934CF
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
consumer.go
View file

@ -276,10 +276,12 @@ func (c *Consumer) getShardIterator(ctx context.Context, streamName, shardID, se
}
func isRetriableError(err error) bool {
switch err.(type) {
case *types.ExpiredIteratorException:
var expiredIteratorException *types.ExpiredIteratorException
var provisionedThroughputExceededException *types.ProvisionedThroughputExceededException
switch {
case errors.As(err, &expiredIteratorException):
return true
case *types.ProvisionedThroughputExceededException:
case errors.As(err, &provisionedThroughputExceededException):
return true
}
return false