ok

Author: reakaleek

src/infra/docs-lambda-index-publisher/Models.cs

@@ -10,7 +10,8 @@ namespace Elastic.Documentation.Lambda.LinkIndexUploader;
 
 [JsonSerializable(typeof(SQSEvent))]
 [JsonSerializable(typeof(S3EventNotification))]
-public partial class SQSEventSerializerContext : JsonSerializerContext;
+[JsonSerializable(typeof(SQSBatchResponse))]
+public partial class LinkIndexUpdaterSerializerContext : JsonSerializerContext;
 
 public class S3EventNotification
 {
@@ -30,7 +31,7 @@ public class S3EventRecord
 public class S3Event
 {
 	[JsonPropertyName("object")]
-	public S3Object S3Object { get; set; } = new();
+	public S3Object Obj { get; set; } = new();
 
 	[JsonPropertyName("bucket")]
 	public S3Bucket Bucket { get; set; } = new();

src/infra/docs-lambda-index-publisher/Program.cs

@@ -17,7 +17,7 @@
 const string bucketName = "elastic-docs-link-index";
 const string indexFile = "link-index-test.json";
 
-await LambdaBootstrapBuilder.Create<SQSEvent>(Handler, new SourceGeneratorLambdaJsonSerializer<SQSEventSerializerContext>())
+await LambdaBootstrapBuilder.Create<SQSEvent>(Handler, new SourceGeneratorLambdaJsonSerializer<LinkIndexUpdaterSerializerContext>())
 	.Build()
 	.RunAsync();
 
@@ -82,8 +82,8 @@ static async Task<SQSBatchResponse> Handler(SQSEvent ev, ILambdaContext context)
 	catch (Exception ex)
 	{
 		// if we fail to update the object, we need to return all the messages
-		context.Logger.LogError($"Failed to update {bucketName}/{indexFile}. Returning all messages to the queue.");
-		context.Logger.LogError(ex.Message);
+		context.Logger.LogError($"Failed to update {bucketName}/{indexFile}. Returning all {ev.Records.Count} messages to the queue.");
+		context.Logger.LogError(ex, ex.Message);
 		return new SQSBatchResponse(ev.Records.Select(r => new SQSBatchResponse.BatchItemFailure
 		{
 			ItemIdentifier = r.MessageId
@@ -96,14 +96,14 @@ static async Task<SQSBatchResponse> Handler(SQSEvent ev, ILambdaContext context)
 	if (string.IsNullOrEmpty(message.Body))
 		throw new Exception("No Body in SQS Message.");
 	context.Logger.LogInformation($"Received message {message.Body}");
-	var s3Event = JsonSerializer.Deserialize<S3EventNotification>(message.Body, SQSEventSerializerContext.Default.S3EventNotification);
+	var s3Event = JsonSerializer.Deserialize<S3EventNotification>(message.Body, LinkIndexUpdaterSerializerContext.Default.S3EventNotification);
 	if (s3Event?.Records == null || s3Event.Records.Count == 0)
 		throw new Exception("Invalid S3 event message format");
 	var linkReferences = new ConcurrentBag<(S3EventRecord, LinkReference)>();
 	await Parallel.ForEachAsync(s3Event.Records, async (record, ctx) =>
 	{
 		var s3Bucket = record.S3.Bucket;
-		var s3Object = record.S3.S3Object;
+		var s3Object = record.S3.Obj;
 		context.Logger.LogInformation($"Get object {s3Object.Key} from bucket {s3Bucket.Name}");
 		var getObjectResponse = await s3Client.GetObjectAsync(s3Bucket.Name, s3Object.Key, ctx);
 		await using var stream = getObjectResponse.ResponseStream;
@@ -118,7 +118,7 @@ await Parallel.ForEachAsync(s3Event.Records, async (record, ctx) =>
 
 static void UpdateLinkIndex(LinkIndex linkIndex, LinkReference linkReference, S3EventRecord s3EventRecord, ILambdaContext context)
 {
-	var s3Object = s3EventRecord.S3.S3Object;
+	var s3Object = s3EventRecord.S3.Obj;
 	var keyTokens = s3Object.Key.Split('/');
 	var repository = keyTokens[1];
 	var branch = keyTokens[2];