fix isRetriableError (#159)

fix issues-158
harlow · Sep 16, 2024 · 4390050 · 4390050
1 parent 8d10ac8
commit 4390050
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/consumer.go b/consumer.go
@@ -299,10 +299,10 @@ func (c *Consumer) getShardIterator(ctx context.Context, streamName, shardID, se
 }
 
 func isRetriableError(err error) bool {
-	switch err.(type) {
-	case *types.ExpiredIteratorException:
+	if oe := (*types.ExpiredIteratorException)(nil); errors.As(err, &oe) {
 		return true
-	case *types.ProvisionedThroughputExceededException:
+	}
+	if oe := (*types.ProvisionedThroughputExceededException)(nil); errors.As(err, &oe) {
 		return true
 	}
 	return false