Call IMcpServer.RunAsync in AspNetCoreSseServer

halter73 · halter73 · commit fbe1dbe5e452 · 2025-03-28T09:37:23.000-07:00
diff --git a/samples/AspNetCoreSseServer/McpEndpointRouteBuilderExtensions.cs b/samples/AspNetCoreSseServer/McpEndpointRouteBuilderExtensions.cs
@@ -26,7 +26,9 @@ public static IEndpointConventionBuilder MapMcpSse(this IEndpointRouteBuilder en
 
             try
             {
+                var serverTask = server.RunAsync(cancellationToken: requestAborted);
                 await transport.RunAsync(cancellationToken: requestAborted);
+                await serverTask;
             }
             catch (OperationCanceledException) when (requestAborted.IsCancellationRequested)
             {

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,9 @@ public static IEndpointConventionBuilder MapMcpSse(this IEndpointRouteBuilder en`
`26`	`26`
`27`	`27`	`try`
`28`	`28`	`{`
	`29`	`+ var serverTask = server.RunAsync(cancellationToken: requestAborted);`
`29`	`30`	`await transport.RunAsync(cancellationToken: requestAborted);`
	`31`	`+ await serverTask;`
`30`	`32`	`}`
`31`	`33`	`catch (OperationCanceledException) when (requestAborted.IsCancellationRequested)`
`32`	`34`	`{`