Skip to content

Commit ef12e8b

Browse files
normjCopilot
andauthored
Potential fix for pull request finding
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
1 parent 83c1986 commit ef12e8b

1 file changed

Lines changed: 17 additions & 5 deletions

File tree

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

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -217,13 +217,25 @@ private async Task PollStream(string streamArn, CancellationToken stoppingToken)
217217

218218
private async Task<(string ShardId, GetRecordsResponse? Response)> PollShard(string shardId, string iterator, CancellationToken stoppingToken)
219219
{
220-
var response = await _streamsClient.GetRecordsAsync(new GetRecordsRequest
220+
try
221221
{
222-
ShardIterator = iterator,
223-
Limit = _config.BatchSize
224-
}, stoppingToken);
222+
var response = await _streamsClient.GetRecordsAsync(new GetRecordsRequest
223+
{
224+
ShardIterator = iterator,
225+
Limit = _config.BatchSize
226+
}, stoppingToken);
225227

226-
return (shardId, response);
228+
return (shardId, response);
229+
}
230+
catch (OperationCanceledException) when (stoppingToken.IsCancellationRequested)
231+
{
232+
throw;
233+
}
234+
catch (Exception ex)
235+
{
236+
_logger.LogWarning(ex, "Failed to poll records for shard {shardId}", shardId);
237+
return (shardId, null);
238+
}
227239
}
228240

229241
/// <summary>

0 commit comments

Comments
 (0)