.NET 开源项目 StreamJsonRpc 介绍[下篇]

阅读本文大概需要 9 分钟。

大家好，这是 .NET 开源项目 StreamJsonRpc 介绍的最后一篇。上篇介绍了一些预备知识，包括 JSON-RPC 协议介绍，StreamJsonRpc 是一个实现了 JSON-RPC 协议的库，它基于 Stream、WebSocket 和自定义的全双工管道传输。中篇通过示例讲解了 StreamJsonRpc 如何使用全双工的 Stream 作为传输管道实现 RPC 通讯。本篇（下篇）将继续通过示例讲解如何基于 WebSocket 传输管道实现 RPC 通讯。

准备工作

为了示例的完整性，本文示例继续在中篇创建的示例基础上进行。该示例的 GitHub 地址为：

github.com/liamwang/StreamJsonRpcSamples

我们继续添加三个项目，一个是名为 WebSocketSample.Client 的 Console 应用，一个是名为 WebSocketSample.Server 的 ASP.NET Core 应用，还有一个名为 Contract 的契约类库（和 gRPC 类似）。

你可以直接复制并执行下面的命令一键完成大部分准备工作：

dotnet new console -n WebSocketSample.Client # 建新客户端应用
dotnet new webapi -n WebSocketSample.Server # 新建服务端应用
dotnet new classlib -n Contract # 新建契约类库
dotnet sln add WebSocketSample.Client WebSocketSample.Server Contract # 将项目添加到解决方案
dotnet add WebSocketSample.Client package StreamJsonRpc # 为客户端安装 StreamJsonRpc 包
dotnet add WebSocketSample.Server package StreamJsonRpc # 为服务端安装 StreamJsonRpc 包
dotnet add WebSocketSample.Client reference Contract # 添加客户端引用 Common 引用
dotnet add WebSocketSample.Server reference Contract # 添加服务端引用 Common 引用

为了把重点放在实现上，这次我们依然以一个简单的功能作为示例。该示例实现客户端向服务端发送一个问候数据，然后服务端响应一个消息。为了更贴合实际的场景，这次使用强类型进行操作。为此，我们在 Contract 项目中添加三个类用来约定客户端和服务端通讯的数据结构和接口。

用于客户端发送的数据的 HelloRequest 类：

public class HelloRequest
{
    public string Name { get; set; }
}

用于服务端响应的数据的 HelloResponse 类：

public class HelloResponse
{
    public string Message { get; set; }
}

用于约定服务端和客户端行为的 IGreeter 接口：

public interface IGreeter
{
    Task SayHelloAsync(HelloRequest request);
}

接下来和中篇一样，通过建立连接、发送请求、接收请求、断开连接这四个步骤演示和讲解一个完整的基于 WebSocket 的 RPC 通讯示例。

建立连接

上一篇讲到要实现 JSON-RPC 协议的通讯，要求传输管道必须是全双工的。而 WebSocket 就是标准的全双工通讯，所以自然可以用来实现 JSON-RPC 协议的通讯。.NET 本身就有现成的 WebSocket 实现，所以在建立连接阶段和 StreamJsonRpc 没有关系。我们只需要把 WebSocket 通讯管道架设好，然后再使用 StreamJsonRpc 来发送和接收请求即可。

客户端使用 WebSocket 建立连接比较简单，使用 ClientWebSocket 来实现，代码如下：

using (var webSocket = new ClientWebSocket())
{
    Console.WriteLine("正在与服务端建立连接...");
    var uri = new Uri("ws://localhost:5000/rpc/greeter");
    await webSocket.ConnectAsync(uri, CancellationToken.None);
    Console.WriteLine("已建立连接");
}

服务端建立 WebSocket 连接最简单的方法就是使用 ASP.NET Core，借助 Kestrel 和 ASP.NET Core 的中间件机制可以轻松搭建基于 WebSocket 的 RPC 服务。只要简单的封装还可以实现同一套代码同时提供 RPC 服务和 Web API 服务。

首先在服务端项目的 Startup.cs 类的 Configure 方法中引入 WebSocket 中间件：

public void Configure(IApplicationBuilder app, IWebHostEnvironment env)
{
    app.UseRouting();

    app.UseWebSockets(); // 增加此行，引入 WebSocket 中间件

    app.UseEndpoints(endpoints =>
    {
        endpoints.MapControllers();
    });
}

再新建一个 Controller 并定义一个 Action 用来路由映射 WebSocket 请求：

public class RpcController : ControllerBase
{
    ...
    [Route("/rpc/greeter")]
    public async Task Greeter()
    {
        if (!HttpContext.WebSockets.IsWebSocketRequest)
        {
            return new BadRequestResult();
        }

        var socket = await HttpContext.WebSockets.AcceptWebSocketAsync();

        ...
    }
}

这里的 Greeter 提供的服务既能接收 HTTP 请求也能接收 WebSocket 请求。HttpContext 中的 WebSockets 属性是一个 WebSocketManager 对象，它可以用来判断当前请求是否为一个 WebSocket 请求，也可以用来等待和接收 WebSocket 连接，即上面代码中的 AcceptWebSocketAsync 方法。另外客户端的 WebSocket 的 Uri 路径需要与 Router 指定的路径对应。

连接已经建立，现在到了 StreamJsonRpc 发挥作用的时候了。

发送请求

客户端通过 WebSocket 发送请求的方式和前一篇讲的 Stream 方式是一样的。还记得前一篇讲到的 JsonRpc 类的 Attach 静态方法吗？它告诉 StreamJsonRpc 如何传输数据，并返回一个用于调用 RPC 的客户端，它除了可以接收 Stream 参数外还有多个重载方法。比如：

public static T Attach(Stream stream);
public static T Attach(IJsonRpcMessageHandler handler);

第二个重载方法可以实现更灵活的 Attach 方式，你可以 Attach 一个交由 WebSocket 传输数据的管道，也可以 Attach 给一个自定义实现的 TCP 全双工传输管道（此方式本文不讲，但文末会直接给出示例）。现在我们需要一个实现了 IJsonRpcMessageHandler 接口的处理程序，StreamJsonRpc 已经实现好了，它是 WebSocketMessageHandler 类。通过 Attach 该实例，可以拿到一个用于调用 RPC 服务的对象。代码示例如下：

Console.WriteLine("开始向服务端发送消息...");
var messageHandler = new WebSocketMessageHandler(webSocket);
var greeterClient = JsonRpc.Attach(messageHandler);
var request = new HelloRequest { Name = "精致码农" };
var response = await greeterClient.SayHelloAsync(request);
Console.WriteLine($"收到来自服务端的响应：{response.Message}");

你会发现，定义客户端和服务端契约的好处是可以实现强类型编程。接下来看服务端如何接收并处理客户端发送的消息。

接收请求

和前一篇一样，我们先定义一个 GreeterServer 类用来处理接收到的客户端消息。

public class GreeterServer : IGreeter
{
    private readonly ILogger _logger;
    public GreeterServer(ILogger logger)
    {
        _logger = logger;
    }

    public Task SayHelloAsync(HelloRequest request)
    {
        _logger.LogInformation("收到并回复了客户端消息");
        return Task.FromResult(new HelloResponse
        {
            Message = $"您好， {request.Name}！"
        });
    }
}

同样，WebSocket 服务端也需要使用 Attach 来告诉 StreamJsonRpc 数据如何通讯，而且使用的也是 WebSocketMessageHandler 类，方法与客户端类似。在前一篇中，我们 Attach 一个 Stream 调用的方法是：

public static JsonRpc Attach(Stream stream, object? target = null);

同理，我们推测应该也有一个这样的静态重载方法：

public static JsonRpc Attach(IJsonRpcMessageHandler handler, object? target = null);

可惜，StreamJsonRpc 并没有提供这个静态方法。既然 Attach 方法返回的是一个 JsonRpc 对象，那我们是否可以直接实例化该对象呢？查看该类的定义，我们发现是可以的，而且有我们需要的构造函数：

public JsonRpc(IJsonRpcMessageHandler messageHandler, object? target);

接下来就简单了，一切和前一篇的 Stream 示例都差不多。在 RpcController 的 Greeter Action 中实例化一个 JsonRpc，然后开启消息监听。

public class RpcController : ControllerBase
{
    private readonly ILogger _logger;
    private readonly GreeterServer _greeterServer;

    public RpcController(ILogger logger, GreeterServer greeterServer)
    {
        _logger = logger;
        _greeterServer = greeterServer;
    }

    [Route("/rpc/greeter")]
    public async Task Greeter()
    {
        if (!HttpContext.WebSockets.IsWebSocketRequest)
        {
            return new BadRequestResult();
        }

        _logger.LogInformation("等待客户端连接...");
        var socket = await HttpContext.WebSockets.AcceptWebSocketAsync();
        _logger.LogInformation("已与客户端建立连接");

        var handler = new WebSocketMessageHandler(socket);

        using (var jsonRpc = new JsonRpc(handler, _greeterServer))
        {
            _logger.LogInformation("开始监听客户端消息...");
            jsonRpc.StartListening();
            await jsonRpc.Completion;
            _logger.LogInformation("客户端断开了连接");
        }

        return new EmptyResult();
    }
}

看起来和我们平时写 Web API 差不多，区别仅仅是对请求的处理方式。但需要注意的是，WebSocket 是长连接，如果客户端没有事情可以处理了，最好主动断开与服务端的连接。如果客户客户没有断开连接，执行的上下文就会停在 await jsonRpc.Completion 处。

断开连接

通常断开连接是由客户端主动发起的，所以服务端不需要做什么处理。服务端响应完消息后，只需使用 jsonRpc.Completion 等待客户端断开连接即可，上一节的代码示例中已经包含了这部分代码，就不再累述了。如果特殊情况下服务端需要断开连接，调用 JsonRpc 对象的 Dispose 方法即可。

不管是 Stream 还是 WebSocket，其客户端对象都提供了 Close 或 Dispose 方法，连接会随着对象的释放自动断开。但最好还是主动调用 Close 方法断开连接，以确保服务端收到断开的请求。对于 ClientWebSocket，需要调用 CloseAsync 方法。客户端完整示例代码如下：

static async Task Main(string[] args)
{
    using (var webSocket = new ClientWebSocket())
    {
        Console.WriteLine("正在与服务端建立连接...");
        var uri = new Uri("ws://localhost:5000/rpc/greeter");
        await webSocket.ConnectAsync(uri, CancellationToken.None);
        Console.WriteLine("已建立连接");

        Console.WriteLine("开始向服务端发送消息...");
        var messageHandler = new WebSocketMessageHandler(webSocket);
        var greeterClient = JsonRpc.Attach(messageHandler);
        var request = new HelloRequest { Name = "精致码农" };
        var response = await greeterClient.SayHelloAsync(request);
        Console.WriteLine($"收到来自服务端的响应：{response.Message}");

        Console.WriteLine("正在断开连接...");
        await webSocket.CloseAsync(WebSocketCloseStatus.NormalClosure, "断开连接", CancellationToken.None);
        Console.WriteLine("已断开连接");
    }

    Console.ReadKey();
}

在实际项目中可能还需要因异常而断开连接的情况做处理，比如网络不稳定可能导致连接中断，这种情况可能需要加入重试机制。

运行示例

由于服务端使用的是 ASP.NET Core 模板，VS 默认使用 IIS Express 启动，启动后会自动打开网页，这样看不到 Console 的日志信息。所以需要把服务端项目 WebSocketSample.Server 的启动方式改成自启动。

另外，为了更方便地同时运行客户端和服务端应用，可以把解决方案设置成多启动。右键解决方案，选择“Properties”，把对应的项目设置“Start”即可。

如果你用的是 VS Code，也是支持多启动调试的，具体方法你自行 Google。如果你用的是 dotnet run 命令运行项目可忽略以上设置。

项目运行后的截图如下：

你也可以自定义实现 TCP 全双工通讯管道，但比较复杂而且也很少这么做，所以就略过不讲了。但我在 GitHub 的示例代码也放了一个自定义全双工管道实现的示例，感兴趣的话你可以克隆下来研究一下。

该示例运行截图：

本篇总结

本文通过示例演示了如何使用 StreamJsonRpc 基于 WebSocket 数据传输实现 JSON-RPC 协议的 RPC 通讯。其中客户端和服务端有共同的契约部分，实现了强类型编程。通过示例我们也清楚了 StreamJsonRpc 这个库为了实现 RPC 通讯做了哪些工作，其实它就是在现有传输管道（Stream、WebSocket 和 自定义 TCP 连接）上进行数据通讯。正如前一篇所说，由于 StreamJsonRpc 把大部分我们不必要知道的细节做了封装，所以在示例中感觉不到 JSON-RPC 协议带来的统一规范，也没看到具体的 JSON 格式的数据。其实只要遵循了 JSON-RPC 协议实现的客户端或服务端，不管是用什么语言实现，都是可以互相通讯的。

希望这三篇关于 StreamJsonRpc 的介绍能让你有所收获，如果你在工作中计划使用 StreamJsonRpc，这几篇文章包括示例代码应该有值得参考的地方。