Pass through CancellationToken when writing OpenAPI response (#56096)

martincostello · web-flow · commit f36ba3968d6d · 2024-06-06T09:05:16.000-07:00
diff --git a/src/OpenApi/src/Extensions/OpenApiEndpointRouteBuilderExtensions.cs b/src/OpenApi/src/Extensions/OpenApiEndpointRouteBuilderExtensions.cs
@@ -50,8 +50,8 @@ public static IEndpointConventionBuilder MapOpenApi(this IEndpointRouteBuilder e
                     try
                     {
                         document.Serialize(new OpenApiJsonWriter(writer), documentOptions.OpenApiVersion);
-                        await context.Response.BodyWriter.WriteAsync(output.ToArray());
-                        await context.Response.BodyWriter.FlushAsync();
+                        await context.Response.BodyWriter.WriteAsync(output.ToArray(), context.RequestAborted);
+                        await context.Response.BodyWriter.FlushAsync(context.RequestAborted);
                     }
                     finally
                     {

Original file line number	Diff line number	Diff line change
`@@ -50,8 +50,8 @@ public static IEndpointConventionBuilder MapOpenApi(this IEndpointRouteBuilder e`
`50`	`50`	`try`
`51`	`51`	`{`
`52`	`52`	`document.Serialize(new OpenApiJsonWriter(writer), documentOptions.OpenApiVersion);`
`53`		`- await context.Response.BodyWriter.WriteAsync(output.ToArray());`
`54`		`- await context.Response.BodyWriter.FlushAsync();`
	`53`	`+ await context.Response.BodyWriter.WriteAsync(output.ToArray(), context.RequestAborted);`
	`54`	`+ await context.Response.BodyWriter.FlushAsync(context.RequestAborted);`
`55`	`55`	`}`
`56`	`56`	`finally`
`57`	`57`	`{`