TiKV 源码解析系列文章（七）gRPC Server 的初始化和启动流程

每天读点书学堂

于 2023-04-14 10:13:48 发布

阅读量211

点赞数 1

CC 4.0 BY-SA版权

文章标签： rust 开发语言后端 tidb 数据库

本文链接：https://blog.youkuaiyun.com/weixin_42241611/article/details/130147235

本文介绍了TiKV项目中的grpc-rs组件，它是gRPC的Rust实现，使用Future和事件驱动的EventLoop。文章通过一个简单的gRPC服务示例，展示了grpc-rs生成的服务端代码框架，并详细解释了如何启动服务监听端口，以及服务如何在后台线程中运行。重点讨论了Environment的创建、完成队列、事件循环和Call对象在处理RPC请求中的作用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

本篇 TiKV 源码解析将为大家介绍 TiKV 的另一周边组件—— grpc-rs 。grpc-rs 是 PingCAP 实现的一个 gRPC 的 Rust 绑定，其 Server/Client 端的代码框架都基于 Future ，事件驱动的 EventLoop 被隐藏在了库的内部，所以非常易于使用。本文将以一个简单的 gRPC 服务作为例子，展示 grpc-rs 会生成的服务端代码框架和需要服务的实现者填写的内容，然后会深入介绍服务器在启动时如何将后台的事件循环与这个框架挂钩，并在后台线程中运行实现者的代码。

基本的代码生成及服务端 API

gRPC 使用 protobuf 定义一个服务，之后调用相关的代码生成工具就可以生成服务端、客户端的代码框架了，这个过程可以参考我们的官方文档。客户端可以直接调用这些生成的代码，向服务端发送请求并接收响应，而服务端则需要服务的实现者自己来定制对请求的处理逻辑，生成响应并发回给客户端。举一个例子：

#[derive(Clone)]
struct MyHelloService {}
impl Hello for MyHelloService {
    // trait 中的函数签名由 grpc-rs 生成，内部实现需要用户自己填写
    fn hello(&mut self, ctx: RpcContext, req: HelloRequest, sink: UnarySink<Hel