f1af251c32* Allow user-defined JsonSerializerOptions for user-defined and inputs and hardcode everything else to the source generator. * Add validation for user-defined JsonSerializerOptions. * Remove out of date comment. * Address feedback and add tests
38 lines
1.7 KiB
C#
38 lines
1.7 KiB
C#
using ModelContextProtocol.Protocol.Messages;
|
|
using ModelContextProtocol.Utils;
|
|
using System.Text.Json;
|
|
using System.Text.Json.Nodes;
|
|
using System.Text.Json.Serialization.Metadata;
|
|
|
|
namespace ModelContextProtocol.Shared;
|
|
|
|
internal sealed class RequestHandlers : Dictionary<string, Func<JsonRpcRequest, CancellationToken, Task<JsonNode?>>>
|
|
{
|
|
/// <summary>
|
|
/// Registers a handler for incoming requests of a specific method.
|
|
/// </summary>
|
|
/// <typeparam name="TRequest">Type of request payload</typeparam>
|
|
/// <typeparam name="TResponse">Type of response payload (not full RPC response</typeparam>
|
|
/// <param name="method">Method identifier to register for</param>
|
|
/// <param name="handler">Handler to be called when a request with specified method identifier is received</param>
|
|
/// <param name="requestTypeInfo">The JSON contract governing request serialization.</param>
|
|
/// <param name="responseTypeInfo">The JSON contract governing response serialization.</param>
|
|
public void Set<TRequest, TResponse>(
|
|
string method,
|
|
Func<TRequest?, CancellationToken, Task<TResponse>> handler,
|
|
JsonTypeInfo<TRequest> requestTypeInfo,
|
|
JsonTypeInfo<TResponse> responseTypeInfo)
|
|
{
|
|
Throw.IfNull(method);
|
|
Throw.IfNull(handler);
|
|
Throw.IfNull(requestTypeInfo);
|
|
Throw.IfNull(responseTypeInfo);
|
|
|
|
this[method] = async (request, cancellationToken) =>
|
|
{
|
|
TRequest? typedRequest = JsonSerializer.Deserialize(request.Params, requestTypeInfo);
|
|
object? result = await handler(typedRequest, cancellationToken).ConfigureAwait(false);
|
|
return JsonSerializer.SerializeToNode(result, responseTypeInfo);
|
|
};
|
|
}
|
|
}
|