#rpc #server-api #microservices #api #networking #client-server #distributed-tracing

forked-tarpc

一个注重易用性的Rust RPC框架

2个版本

0.31.1 2023年6月19日
0.31.0 2023年6月16日

#1547 in 网络编程

MIT 许可证

205KB
4K SLoC

Crates.io MIT licensed Build status Discord chat

tarpc

免责声明:这不是一个官方的Google产品。

tarpc是一个注重易用性的Rust RPC框架。定义一个服务只需要几行代码,并且大多数服务器编写时的样板代码都由框架自动处理。

文档

什么是RPC框架?

"RPC"代表“远程过程调用”,是一种函数调用,其中产生返回值的工作在别处完成。当调用RPC函数时,在幕后,该函数会联系其他进程并请求它们评估该函数。原始函数随后返回其他进程产生的值。

RPC框架是大多数面向微服务的架构的基本构建块。其中两个著名的例子是gRPCCap'n Proto

tarpc与其他RPC框架的不同之处在于,它通过代码定义模式,而不是在如.proto等单独的语言中定义。这意味着没有单独的编译过程,也不需要在不同的语言之间切换上下文。

tarpc的一些其他功能

  • 可插拔传输:任何实现了Stream<Item = Request> + Sink<Response>的类型都可以用作传输以连接客户端和服务器。
  • Send + 'static可选:如果传输不需要它,tarpc也不需要!
  • 级联取消:丢弃请求将向服务器发送取消消息。服务器将停止对该请求的任何未完成的工作,然后取消其自己的请求,并重复此过程以取消整个传递依赖链中的请求。
  • 可配置的截止时间和截止时间传播:如果未指定,请求截止时间默认为10秒。服务器将在截止时间过后自动停止工作。服务器发送的任何使用请求上下文的请求都将传播请求截止时间。例如,如果一个服务器正在处理一个10秒截止时间的请求,做了2秒的工作,然后向另一个服务器发送请求,那个服务器将看到8秒的截止时间。
  • 分布式跟踪:tarpc通过跟踪原语与OpenTelemetry跟踪扩展。使用兼容的跟踪订阅者,如Jaeger,每个RPC都可以通过客户端、服务器以及服务器下游的其他依赖进行跟踪。即使对于未连接到分布式跟踪收集器的应用程序,该仪器也可以被常规记录器,如env_logger,所摄取。
  • Serde序列化:启用serde1 Cargo功能将使服务请求和响应Serialize + Deserialize。虽然这是完全可选的,但也可以使用内存传输,因此当不需要序列化时,不需要支付序列化的代价。

使用方法

将依赖项添加到您的Cargo.toml

tarpc = "0.31"

tarpc::service属性扩展为一个集合,该集合形成了一个RPC服务。这些生成的类型使得编写服务器变得简单且易于使用,减少了样板代码。只需实现生成的服务特性和,然后您就可以开始比赛了!

示例

此示例使用tokio,因此请将以下依赖项添加到您的Cargo.toml

anyhow = "1.0"
futures = "0.3"
tarpc = { version = "0.31", features = ["tokio1"] }
tokio = { version = "1.0", features = ["macros"] }

在以下示例中,我们使用进程内通道进行客户端和服务器之间的通信。在实际代码中,您可能会通过网络进行通信。有关更实际的示例,请参阅example-service

首先,让我们设置依赖项和服务定义。


use futures::{
    future::{self, Ready},
    prelude::*,
};
use tarpc::{
    client, context,
    server::{self, incoming::Incoming, Channel},
};

// This is the service definition. It looks a lot like a trait definition.
// It defines one RPC, hello, which takes one arg, name, and returns a String.
#[tarpc::service]
trait World {
    /// Returns a greeting for name.
    async fn hello(name: String) -> String;
}

此服务定义生成一个名为World的特性和。接下来,我们需要为我们的Server结构体实现它。

// This is the type that implements the generated World trait. It is the business logic
// and is used to start the server.
#[derive(Clone)]
struct HelloServer;

impl World for HelloServer {
    // Each defined rpc generates two items in the trait, a fn that serves the RPC, and
    // an associated type representing the future output by the fn.

    type HelloFut = Ready<String>;

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

最后,让我们编写我们的main,它将启动服务器。虽然此示例使用进程内通道,但tarpc还提供了一个在serde-transport功能背后的通用serde_transport,以及额外的TCP功能,在tcp功能背后可用。

#[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()));

    // WorldClient is generated by the #[tarpc::service] attribute. It has a constructor `new`
    // that takes a config and any Transport as input.
    let mut client = WorldClient::new(client::Config::default(), client_transport).spawn();

    // The client has an RPC method for each RPC defined in the annotated trait. It takes the same
    // args as defined, with the addition of a Context, which is always the first arg. The Context
    // specifies a deadline and trace information which can be helpful in debugging requests.
    let hello = client.hello(context::current(), "Stim".to_string()).await?;

    println!("{hello}");

    Ok(())
}

服务文档

使用您通常使用的cargo doc来查看由service!调用扩展的所有项创建的文档。

许可:MIT

依赖项

~4–17MB
~170K SLoC