Merge branch 'normj/lambda-ddbstreams' of https://github.com/aws/aws-lambda-dotnet into normj/lambda-ddbstreams

Author: normj

Tools/LambdaTestTool-v2/src/Amazon.Lambda.TestTool/Processes/DynamoDBStreamsEventSource/DynamoDBStreamsEventSourceBackgroundService.cs

@@ -217,13 +217,25 @@ private async Task PollStream(string streamArn, CancellationToken stoppingToken)
 
     private async Task<(string ShardId, GetRecordsResponse? Response)> PollShard(string shardId, string iterator, CancellationToken stoppingToken)
     {
-        var response = await _streamsClient.GetRecordsAsync(new GetRecordsRequest
+        try
         {
-            ShardIterator = iterator,
-            Limit = _config.BatchSize
-        }, stoppingToken);
+            var response = await _streamsClient.GetRecordsAsync(new GetRecordsRequest
+            {
+                ShardIterator = iterator,
+                Limit = _config.BatchSize
+            }, stoppingToken);
 
-        return (shardId, response);
+            return (shardId, response);
+        }
+        catch (OperationCanceledException) when (stoppingToken.IsCancellationRequested)
+        {
+            throw;
+        }
+        catch (Exception ex)
+        {
+            _logger.LogWarning(ex, "Failed to poll records for shard {shardId}", shardId);
+            return (shardId, null);
+        }
     }
 
     /// <summary>