tarpc：Rust中的易用RPC框架

2024-05-22 02:48:59作者：冯爽妲Honey

简介

tarpc是一个专为Rust设计的远程过程调用（RPC）框架，注重简洁和便捷性。定义服务只需几行代码，并且大部分服务器的繁琐工作都已为你处理好。不是Google的官方产品，但由其开发团队维护。

项目技术分析

tarpc与其他RPC框架的最大区别在于它在代码中定义服务规范，而非依赖如.proto这样的独立语言。这意味着没有额外的编译步骤，也不需要在不同语言之间切换。核心特性包括：

可插拔传输：任何实现了Stream<Item = Request> + Sink<Response>的类型都可以作为客户端和服务端之间的传输层。
非必需的'Send + 'static：如果传输层不需要，tarpc也不会强制要求。
级联取消：当请求被取消时，会向服务器发送取消消息。服务器将停止对该请求的未完成工作，并可能进一步取消自身产生的请求，形成级联效应。
自定义超时与超时传播：默认请求超时时间为10秒。服务器会在超时时自动停止工作，并将超时传递给下游请求。
分布式追踪：集成tracing和OpenTelemetry，支持Jaeger等兼容的跟踪收集器，或者直接与常规日志记录器配合使用。
Serde序列化：启用serde1 Cargo功能后，服务请求和响应可以实现Serialize + Deserialize，但不是强制性的，不需序列化的场景可以选择内存传输。

应用场景

tarpc适用于构建微服务架构，尤其适合那些希望在代码中直接定义服务接口的项目。例如，可以在多个内部服务之间进行高效的数据交互，或者搭建跨网络通信的服务集群。

项目特点

简单快速的定义服务：使用tarpc::service宏，可以轻松快捷地定义服务并实现其业务逻辑。
灵活的传输层：能够适应各种传输方式，包括但不限于网络和内存通道。
无侵入的API：tarpc提供的API既不增加额外的编程模型复杂度，又保持了Rust本身的良好体验。
高性能与低延迟：由于无需跨语言边界，tarpc通常能提供更高效的性能和更低的延迟。

示例代码

以下是一个使用tokio的简单示例，展示了如何创建一个简单的RPC服务：

use futures::future::{self, Ready};
use tarpc::{
    client, context,
    server::{self, Channel},
};

// 定义服务
#[tarpc::service]
trait World {
    async fn hello(name: String) -> String;
}

// 实现服务
#[derive(Clone)]
struct HelloServer;

impl World for HelloServer {
    type HelloFut = Ready<String>;

    fn hello(self, _: context::Context, name: String) -> Self::HelloFut {
        future::ready(format!("Hello, {name}!"))
    }
}

// 主程序启动服务并创建客户端
#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let (client_transport, server_transport) = tarpc::transport::channel::unbounded();
    
    let server = server::BaseChannel::with_defaults(server_transport);
    tokio::spawn(server.execute(HelloServer.serve()));
    
    let client = WorldClient::new(client::Config::default(), client_transport).spawn();
    let hello = client.hello(context::current(), "Stim".to_string()).await?;
    
    println!("{}", hello);
    
    Ok(())
}