fix

Author: reakaleek

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

@@ -17,17 +17,13 @@
 const string bucketName = "elastic-docs-link-index";
 const string indexFile = "link-index-test.json";
 
-await LambdaBootstrapBuilder.Create<SQSEvent>(Handler, new SourceGeneratorLambdaJsonSerializer<LinkIndexUpdaterSerializerContext>())
+await LambdaBootstrapBuilder.Create<SQSEvent, SQSBatchResponse>(Handler, new SourceGeneratorLambdaJsonSerializer<LinkIndexUpdaterSerializerContext>())
 	.Build()
 	.RunAsync();
 
 return;
 
-// This function should return SQSBatchResponse but for some reason
-// it's not working as expected.
-// The partial batch response might not be serialized correctly.
-// However, serializing the response to JSON and returning it as string works.
-static async Task<string> Handler(SQSEvent ev, ILambdaContext context)
+static async Task<SQSBatchResponse> Handler(SQSEvent ev, ILambdaContext context)
 {
 	var s3Client = new AmazonS3Client();
 	var batchItemFailures = new List<SQSBatchResponse.BatchItemFailure>();
@@ -87,7 +83,8 @@ static async Task<string> Handler(SQSEvent ev, ILambdaContext context)
 			context.Logger.LogInformation($"Failed to process {batchItemFailures.Count}  messages. Returning them to the queue.");
 			context.Logger.LogInformation(jsonString);
 		}
-		return jsonString;
+
+		return response;
 	}
 	catch (Exception ex)
 	{
@@ -105,7 +102,7 @@ static async Task<string> Handler(SQSEvent ev, ILambdaContext context)
 		}).ToList());
 		var jsonString = JsonSerializer.Serialize(response, LinkIndexUpdaterSerializerContext.Default.SQSBatchResponse);
 		context.Logger.LogInformation(jsonString);
-		return jsonString;
+		return response;
 	}
 }
 

src/infra/docs-lambda-index-publisher/docs-lambda-index-publisher.csproj

@@ -31,4 +31,3 @@
     <ProjectReference Include="..\..\Elastic.Markdown\Elastic.Markdown.csproj"/>
   </ItemGroup>
 </Project>
-