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

Add Trace-level logging for JSON-RPC payloads in transports (#1191)

Browse source
This commit is contained in:
Copilot 2026-01-29 11:26:33 -08:00 • committed by GitHub
commit ee8c485d41
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 90 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/ModelContextProtocol.Core/Client/SseClientSessionTransport.cs
View file

@ -80,6 +80,8 @@ internal sealed partial class SseClientSessionTransport : TransportBase
messageId = messageWithId.Id.ToString();
}
LogTransportSendingMessageSensitive(message);
using var httpRequestMessage = new HttpRequestMessage(HttpMethod.Post, _messageEndpoint);
StreamableHttpClientSessionTransport.CopyAdditionalHeaders(httpRequestMessage.Headers, _options.AdditionalHeaders, sessionId: null, protocolVersion: null);
var response = await _httpClient.SendAsync(httpRequestMessage, message, cancellationToken).ConfigureAwait(false);

2
src/ModelContextProtocol.Core/Client/StreamClientSessionTransport.cs
View file

@ -105,6 +105,8 @@ internal class StreamClientSessionTransport : TransportBase
var json = JsonSerializer.Serialize(message, McpJsonUtilities.JsonContext.Default.JsonRpcMessage);
LogTransportSendingMessageSensitive(Name, json);
using var _ = await _sendLock.LockAsync(cancellationToken).ConfigureAwait(false);
try
{

2
src/ModelContextProtocol.Core/Client/StreamableHttpClientSessionTransport.cs
View file

@ -75,6 +75,8 @@ internal sealed partial class StreamableHttpClientSessionTransport : TransportBa
$"Call {nameof(McpClient)}.{nameof(McpClient.ResumeSessionAsync)} to resume existing sessions.");
}
LogTransportSendingMessageSensitive(message);
using var sendCts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken, _connectionCts.Token);
cancellationToken = sendCts.Token;

16
src/ModelContextProtocol.Core/Protocol/TransportBase.cs
View file

@ -1,6 +1,7 @@
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Abstractions;
using System.Diagnostics;
using System.Text.Json;
using System.Threading.Channels;
namespace ModelContextProtocol.Protocol;
@ -166,6 +167,21 @@ public abstract partial class TransportBase : ITransport
[LoggerMessage(Level = LogLevel.Error, Message = "{EndpointName} transport send failed for message ID '{MessageId}'.")]
private protected partial void LogTransportSendFailed(string endpointName, string messageId, Exception exception);
[LoggerMessage(Level = LogLevel.Trace, Message = "{EndpointName} transport sending message. Message: '{Message}'.")]
private protected partial void LogTransportSendingMessageSensitive(string endpointName, string message);
/// <summary>
/// Logs a sending message at Trace level if trace logging is enabled.
/// </summary>
/// <param name="message">The JSON-RPC message to log.</param>
private protected void LogTransportSendingMessageSensitive(JsonRpcMessage message)
{
if (_logger.IsEnabled(LogLevel.Trace))
{
LogTransportSendingMessageSensitive(Name, JsonSerializer.Serialize(message, McpJsonUtilities.JsonContext.Default.JsonRpcMessage));
}
}
[LoggerMessage(Level = LogLevel.Information, Message = "{EndpointName} transport reading messages.")]
private protected partial void LogTransportEnteringReadMessagesLoop(string endpointName);

4
src/ModelContextProtocol.Core/Server/StreamServerTransport.cs
View file

@ -74,7 +74,9 @@ public class StreamServerTransport : TransportBase
try
{
await JsonSerializer.SerializeAsync(_outputStream, message, McpJsonUtilities.DefaultOptions.GetTypeInfo(typeof(JsonRpcMessage)), cancellationToken).ConfigureAwait(false);
var json = JsonSerializer.Serialize(message, McpJsonUtilities.JsonContext.Default.JsonRpcMessage);
LogTransportSendingMessageSensitive(Name, json);
await _outputStream.WriteAsync(Encoding.UTF8.GetBytes(json), cancellationToken).ConfigureAwait(false);
await _outputStream.WriteAsync(s_newlineBytes, cancellationToken).ConfigureAwait(false);
await _outputStream.FlushAsync(cancellationToken).ConfigureAwait(false);
}