modelcontextprotocol/csharp-sdk
Watch
1
0
Fork
You've already forked csharp-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

Stop the host when the single session server service finishes (#226)

Browse source 
* Stop the host when the single session server service finishes

* Add test
This commit is contained in:
Stephen Toub 2025-04-07 12:45:17 -04:00 • committed by GitHub
commit 5d7c2a1a0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 40 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
src/ModelContextProtocol/Hosting/SingleSessionMcpServerHostedService.cs
View file

@ -6,8 +6,22 @@ namespace ModelContextProtocol.Hosting;
/// <summary>
/// Hosted service for a single-session (e.g. stdio) MCP server.
/// </summary>
internal sealed class SingleSessionMcpServerHostedService(IMcpServer session) : BackgroundService
/// <param name="session">The server representing the session being hosted.</param>
/// <param name="lifetime">
/// The host's application lifetime. If available, it will have termination requested when the session's run completes.
/// </param>
internal sealed class SingleSessionMcpServerHostedService(IMcpServer session, IHostApplicationLifetime? lifetime = null) : BackgroundService
{
/// <inheritdoc />
protected override Task ExecuteAsync(CancellationToken stoppingToken) => session.RunAsync(stoppingToken);
protected override async Task ExecuteAsync(CancellationToken stoppingToken)
{
try
{
await session.RunAsync(stoppingToken);
}
finally
{
lifetime?.StopApplication();
}
}
}