.NET7中gRPC如何复用TCP连接切换客户端与服务端角色？

可行性说明 #

HTTP/2原生支持多路复用，而gRPC基于HTTP/2实现，因此.NET7完全支持在已建立的TCP连接上双向发起gRPC调用——也就是让原本的服务端A转为客户端，原本的客户端B转为服务端，且复用同一条初始连接。

核心实现思路 #

1. 客户端B内嵌gRPC服务：B在发起对A的连接前，先启动一个内嵌的ASP.NET Core gRPC服务，用于后续接收A的调用请求。
2. 认证阶段传递B的服务信息：B连接A完成认证后，将自身gRPC服务的地址（或标识）告知A。
3. A复用连接调用B：A借助与B已建立的HTTP/2连接池，创建gRPC客户端调用B的服务，确保复用同一条TCP连接。

具体代码实现步骤 #

1. 客户端B的改造（兼具服务端能力） #

• 安装依赖：在控制台项目中安装Grpc.AspNetCore包。
• 定义B的gRPC服务契约（ReverseService.proto）：

syntax = "proto3";

option csharp_namespace = "B.GrpcServices";

package reverse;

service ReverseGreeter {
  rpc SayHelloBack (HelloRequest) returns (HelloReply);
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}

• 实现B的gRPC服务逻辑：

public class ReverseGreeterService : ReverseGreeter.ReverseGreeterBase
{
    public override Task<HelloReply> SayHelloBack(HelloRequest request, ServerCallContext context)
    {
        return Task.FromResult(new HelloReply
        {
            Message = $"Hello from B (now server)! Received: {request.Name}"
        });
    }
}

• 在Program.cs中启动内嵌服务并连接A：

// 启动内嵌gRPC服务
var builder = WebApplication.CreateBuilder(args);
builder.Services.AddGrpc();
var app = builder.Build();
app.MapGrpcService<ReverseGreeterService>();
var hostTask = app.RunAsync("http://localhost:5001");

// 作为客户端连接A并完成认证
var channel = GrpcChannel.ForAddress("http://localhost:5000");
var greeterClient = new Greeter.GreeterClient(channel);
var authReply = await greeterClient.SayHelloAsync(new HelloRequest { Name = "B (client)" });
Console.WriteLine(authReply.Message);

// 将B的服务地址告知A（通过自定义RPC接口传递）
var registryClient = new ClientRegistry.ClientRegistryClient(channel);
await registryClient.RegisterAsync(new ClientInfo { Address = "http://localhost:5001" });

// 保持程序运行
await hostTask;

2. 服务端A的改造（兼具客户端能力） #

• 安装依赖：安装Grpc.Net.Client包。
• 定义接收B地址的RPC契约（ClientRegistry.proto）：

syntax = "proto3";

option csharp_namespace = "A.GrpcServices";

package registry;

service ClientRegistry {
  rpc Register (ClientInfo) returns (RegisterReply);
}

message ClientInfo {
  string address = 1;
}

message RegisterReply {
  bool success = 1;
}

• 实现注册服务并复用连接调用B：

public class ClientRegistryService : ClientRegistry.ClientRegistryBase
{
    private readonly IHttpClientFactory _httpClientFactory;

    public ClientRegistryService(IHttpClientFactory httpClientFactory)
    {
        _httpClientFactory = httpClientFactory;
    }

    public override async Task<RegisterReply> Register(ClientInfo request, ServerCallContext context)
    {
        // 复用连接池创建指向B的gRPC通道
        var channel = GrpcChannel.ForAddress(request.Address, new GrpcChannelOptions
        {
            HttpClient = _httpClientFactory.CreateClient()
        });

        var reverseClient = new ReverseGreeter.ReverseGreeterClient(channel);
        var reply = await reverseClient.SayHelloBackAsync(new HelloRequest { Name = "A (now client)" });
        Console.WriteLine(reply.Message);

        return new RegisterReply { Success = true };
    }
}

• 在Program.cs中注册服务与HttpClient：

var builder = WebApplication.CreateBuilder(args);
builder.Services.AddGrpc();
builder.Services.AddHttpClient(); // 管理连接池实现复用

var app = builder.Build();
app.MapGrpcService<GreeterService>();
app.MapGrpcService<ClientRegistryService>();

app.Run();

关键注意事项 #

• 连接复用保障：通过IHttpClientFactory管理HttpClient实例，利用HTTP/2多路复用特性，确保A与B的双向调用复用同一条TCP连接。
• 认证安全性：初始的SayHello调用需完成双向身份认证，确保后续反向调用的可信性。
• 内网场景适配：若B处于防火墙后无法直接被访问，可基于HTTP/2的流特性实现反向调用（无需暴露B的端口），但实现复杂度更高，上述内嵌服务方案更适合常规场景。

内容的提问来源于stack exchange，提问作者KarloX