fix isRetriableError (#159)

luanruisong · web-flow · commit 43900507c911 · 2024-09-16T12:27:23.000-07:00
fix issues-158
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

Original file line number	Diff line number	Diff line change
`@@ -299,10 +299,10 @@ func (c *Consumer) getShardIterator(ctx context.Context, streamName, shardID, se`
`299`	`299`	`}`
`300`	`300`
`301`	`301`	`func isRetriableError(err error) bool {`
`302`		`- switch err.(type) {`
`303`		`- case *types.ExpiredIteratorException:`
	`302`	`+ if oe := (*types.ExpiredIteratorException)(nil); errors.As(err, &oe) {`
`304`	`303`	`return true`
`305`		`- case *types.ProvisionedThroughputExceededException:`
	`304`	`+ }`
	`305`	`+ if oe := (*types.ProvisionedThroughputExceededException)(nil); errors.As(err, &oe) {`
`306`	`306`	`return true`
`307`	`307`	`}`
`308`	`308`	`return false`