实现跨语言通信：Rust 和 Thrift 的最佳实践

server/2024/11/30 10:43:43/

前言

在分布式系统中，服务之间高效且安全的通信至关重要。Apache Thrift 是一个被广泛应用的跨语言 RPC（远程过程调用）框架，它支持多种编程语言，包括 Rust。Rust 以其卓越的性能和内存安全保障，成为越来越多开发者的首选语言。
本文将深入探讨如何在 Rust 项目中集成 Thrift，帮助开发者实现跨服务的高效通信，并且探讨异步编程和 TLS 安全通信的高级实现方式。

什么是 Thrift？

Thrift 是由 Facebook 开发并开源的一个高效的服务框架。它允许你定义数据类型和服务接口，然后生成跨多种编程语言的代码。简单来说，Thrift 提供了一个统一的接口来实现不同语言之间的通信。

集成步骤

你可以通过以下命令安装 Thrift 编译器：

# For macOS using Homebrew
brew install thrift# For Ubuntu
sudo apt-get install thrift-compiler

1. 定义 Thrift 文件

首先，你需要定义一个 .thrift 文件，描述你的数据结构和服务接口。创建一个文件 example.thrift：

namespace rs examplestruct User {1: i32 id,2: string name,3: i32 age
}service UserService {User getUser(1: i32 id),void saveUser(1: User user)
}

这个文件描述了一个 User 结构体和一个 UserService 服务。

2. 生成 Rust 代码

使用 Thrift 编译器生成 Rust 代码：

thrift --gen rs example.thrift

这将会在当前目录下生成一个 gen-rs 目录，里面包含了 Thrift 为 Rust 生成的代码。

3. 在 Rust 项目中使用 Thrift

创建一个新的 Rust 项目：

cargo new rust_thrift_example
cd rust_thrift_example

编辑 Cargo.toml 文件，添加 Thrift 依赖：

[dependencies]
thrift = "0.14.1"

将生成的 gen-rs 文件夹复制到 src 目录下，以便在项目中使用。

接下来，我们编写一个简单的客户端和服务器。

服务器端代码

在 src 目录下创建一个新文件 server.rs，编写服务器端代码：

use thrift::protocol::{TBinaryInputProtocol, TBinaryOutputProtocol};
use thrift::server::{TServer, TSimpleServer};
use thrift::transport::{TBufferedReadTransport, TBufferedWriteTransport, TIoChannel, TTcpChannel, TTcpListener, TTransport};mod gen_rs {pub mod example;
}use gen_rs::example::{User, UserServiceSyncProcessor};struct UserServiceHandler;impl UserServiceSyncProcessor for UserServiceHandler {fn getUser(&self, id: i32) -> thrift::Result<User> {Ok(User { id, name: format!("User{}", id), age: 30 })}fn saveUser(&self, user: User) -> thrift::Result<()> {println!("User saved: {:?}", user);Ok(())}
}fn main() -> thrift::Result<()> {let listener = TTcpListener::new("127.0.0.1:9090")?;let server = TSimpleServer::new(UserServiceHandler,TBinaryInputProtocol::new,TBinaryOutputProtocol::new,TBufferedReadTransport::new,TBufferedWriteTransport::new,listener,);println!("Starting the server...");server.serve()?;Ok(())
}
客户端代码

在 src 目录下创建一个新文件 client.rs，编写客户端代码：

use thrift::protocol::{TBinaryInputProtocol, TBinaryOutputProtocol};
use thrift::transport::{TBufferedReadTransport, TBufferedWriteTransport, TTcpChannel, TTransport};mod gen_rs {pub mod example;
}use gen_rs::example::{UserServiceSyncClient};fn main() -> thrift::Result<()> {let mut transport = TTcpChannel::new();transport.open("127.0.0.1:9090")?;let (i_prot, o_prot) = (TBinaryInputProtocol::new(TBufferedReadTransport::new(transport.try_clone()?)),TBinaryOutputProtocol::new(TBufferedWriteTransport::new(transport)),);let client = UserServiceSyncClient::new(i_prot, o_prot);let user = client.getUser(1)?;println!("Got user: {:?}", user);client.saveUser(user)?;Ok(())
}

4. 运行服务器和客户端

首先，编译并运行服务器：

cargo run --bin server

然后，在另一个终端窗口中运行客户端：

cargo run --bin client

你应该会看到客户端从服务器获取到用户信息并将其保存。

实际应用

在上面的教程中，我们已经成功地实现了一个基础的 Thrift 服务和客户端。但是，在实际应用中，我们可能会遇到更多的需求和挑战。接下来，我们将深入探讨一些常见的需求和解决方案。

异步编程

随着现代应用对性能和并发的要求越来越高，异步编程变得越来越重要。Rust 提供了强大的异步编程支持，我们可以利用这些特性来提升 Thrift 服务的性能。

使用 tokio 和 async 实现异步 Thrift 服务

tokio 是一个用于异步编程的强大框架。我们可以结合 tokio 和 Rust 的 async 特性来实现异步 Thrift 服务。

首先，确保在 Cargo.toml 文件中添加 tokio 依赖：

[dependencies]
thrift = "0.14.1"
tokio = { version = "1", features = ["full"] }然后，修改服务器端代码以支持异步：
use thrift::protocol::{TBinaryInputProtocol, TBinaryOutputProtocol};
use thrift::server::TServer;
use thrift::transport::{TBufferedReadTransport, TBufferedWriteTransport, TIoChannel, TTcpChannel, TTcpListener};
use tokio::net::TcpListener;
use tokio::sync::Mutex;
use std::sync::Arc;mod gen_rs {pub mod example;
}use gen_rs::example::{User, UserServiceAsyncProcessor};struct UserServiceHandler;#[async_trait::async_trait]
impl UserServiceAsyncProcessor for UserServiceHandler {async fn getUser(&self, id: i32) -> thrift::Result<User> {Ok(User { id, name: format!("User{}", id), age: 30 })}async fn saveUser(&self, user: User) -> thrift::Result<()> {println!("User saved: {:?}", user);Ok(())}
}#[tokio::main]
async fn main() -> thrift::Result<()> {let listener = TcpListener::bind("127.0.0.1:9090").await?;let handler = Arc::new(Mutex::new(UserServiceHandler));loop {let (socket, _) = listener.accept().await?;let handler = handler.clone();tokio::spawn(async move {let (i_prot, o_prot) = (TBinaryInputProtocol::new(TBufferedReadTransport::new(TTcpChannel::new(socket))),TBinaryOutputProtocol::new(TBufferedWriteTransport::new(TTcpChannel::new(socket))),);let processor = UserServiceAsyncProcessor::new(handler);let mut server = TServer::new(i_prot, o_prot, processor);server.serve().await.unwrap();});}
}

在这个示例中，我们使用 tokio::net::TcpListener 来异步监听连接，并使用 tokio::spawn 来处理每个连接，从而实现并发处理。

使用 TLS 加密通信

在生产环境中，安全性是极为重要的考量。我们可以使用 TLS （传输层安全） 来加密 Thrift 服务的通信。

配置 TLS
首先，确保在 Cargo.toml 文件中添加 tokio-rustls 依赖：

[dependencies]
thrift = "0.14.1"
tokio = { version = "1", features = ["full"] }
tokio-rustls = "0.22"
rustls = "0.20"

然后，生成自签名证书或使用受信任的证书。为了简化演示，我们使用自签名证书：
openssl req -x509 -newkey rsa:4096 -keyout key.pem -out cert.pem -days 365 -nodes

接着，修改服务器端代码以支持 TLS：

use thrift::protocol::{TBinaryInputProtocol, TBinaryOutputProtocol};
use thrift::server::TServer;
use thrift::transport::{TBufferedReadTransport, TBufferedWriteTransport, TIoChannel, TTcpChannel};
use tokio::net::TcpListener;
use tokio::sync::Mutex;
use tokio_rustls::rustls::{ServerConfig, NoClientAuth, Certificate, PrivateKey};
use tokio_rustls::TlsAcceptor;
use tokio_rustls::rustls::internal::pemfile::{certs, rsa_private_keys};
use std::sync::Arc;
use std::fs::File;
use std::io::{BufReader, self};mod gen_rs {pub mod example;
}use gen_rs::example::{User, UserServiceAsyncProcessor};struct UserServiceHandler;#[async_trait::async_trait]
impl UserServiceAsyncProcessor for UserServiceHandler {async fn getUser(&self, id: i32) -> thrift::Result<User> {Ok(User { id, name: format!("User{}", id), age: 30 })}async fn saveUser(&self, user: User) -> thrift::Result<()> {println!("User saved: {:?}", user);Ok(())}
}#[tokio::main]
async fn main() -> thrift::Result<()> {let mut config = ServerConfig::new(NoClientAuth::new());let cert_file = &mut BufReader::new(File::("cert.pem")?);let key_file = &mut BufReader::new(File::open("key.pem")?);let cert_chain = certs(cert_file).map_err(|_| io::Error::new(io::ErrorKind::InvalidInput, "invalid cert"))?;let mut keys = rsa_private_keys(key_file).map_err(|_| io::Error::new(io::ErrorKind::InvalidInput, "invalid key"))?;config.set_single_cert(cert_chain, keys.remove(0)).map_err(|_| io::Error::new(io::ErrorKind::InvalidInput, "invalid key"))?;let acceptor = TlsAcceptor::from(Arc::new(config));let listener = TcpListener::bind("127.0.0.1:9090").await?;let handler = Arc::new(Mutex::new(UserServiceHandler));loop {let (socket, _) = listener.accept().await?;let handler = handler.clone();let acceptor = acceptor.clone();tokio::spawn(async move {let tls_socket = acceptor.accept(socket).await.unwrap();let (i_prot, o_prot) = (TBinaryInputProtocol::new(TBufferedReadTransport::new(TTcpChannel::new(tls_socket))),TBinaryOutputProtocol::new(TBufferedWriteTransport::new(TTcpChannel::new(tls_socket))),);let processor = UserServiceAsyncProcessor::new(handler);let mut server = TServer::new(i_prot, o_prot, processor);server.serve().await.unwrap();});}
}

使用 TLS 的客户端代码类似，只需在创建连接时使用 tokio-rustls 的 TlsConnector 来进行加密连接。

总结

本文详细介绍了如何在 Rust 项目中集成 Apache Thrift，以及如何通过异步编程和 TLS 实现更高效和安全的服务通信。Rust 和 Thrift 的结合为开发者提供了一种可靠的跨语言通信解决方案，能够满足现代分布式系统的高性能和高安全性需求。希望通过本文的教程，开发者能够更好地理解和应用 Rust 和 Thrift，实现高效的分布式系统开发。


http://www.ppmy.cn/server/146140.html

相关文章

读书分享（二）| 心智的成熟之旅

读书分享（二）| 心智的成熟之旅

概述 人生苦难重重&#xff08;承认这一现实&#xff0c;就不会这么痛苦了。在苦这个事儿上&#xff0c;人人生而平等。苦中作乐才是人生的使命&#xff09;规避问题和逃避痛苦的趋向&#xff0c;是人类心理疾病的根源。只有心智成熟的人&#xff0c;才有能力改变人生。我们唯…
阅读更多...
未来已来！联想推出汽车智能空间解决方案

未来已来！联想推出汽车智能空间解决方案

2024年11月28日&#xff0c;联想车计算在2024联想创新开放日系列活动中&#xff0c;隆重推出联想汽车智能空间解决方案&#xff0c;并于联想北京及上海未来中心开放展示。联想Xphere未来座舱&#xff0c;以前瞻性的理念打造未来移动生活空间&#xff0c;为用户带来前所未有的驾…
阅读更多...
三维地形图计算软件(三)-原基于PYQT5+pyqtgraph旧代码

三维地形图计算软件(三)-原基于PYQT5+pyqtgraph旧代码

最先入手设计三维地形图及平基挖填方计算软件时&#xff0c;地形图的显示方案是&#xff1a;三维视图基于pyqtgraph.opengl显示和二维视图基于pyqtgraph的PlotWidget来显示地形地貌&#xff0c;作到一半时就发现&#xff0c;地形点过多时&#xff0c;将会造成系统卡顿(加载时主…
阅读更多...
Oracle-—系统包使用

Oracle-—系统包使用

文章目录 系统包dbms_redefinition 系统包 dbms_redefinition 功能介绍&#xff1a;该包体可以实现将Oracle库下的表在线改为分区结构或者重新定义&#xff1b; 说明&#xff1a;在检查表是否可以重定义和开始重定义的过程中&#xff0c;按照表是否存在主键&#xff0c;参数 o…
阅读更多...
量化交易系统开发-实时行情自动化交易-8.1.TradingView平台

量化交易系统开发-实时行情自动化交易-8.1.TradingView平台

19年创业做过一年的量化交易但没有成功&#xff0c;作为交易系统的开发人员积累了一些经验&#xff0c;最近想重新研究交易系统&#xff0c;一边整理一边写出来一些思考供大家参考&#xff0c;也希望跟做量化的朋友有更多的交流和合作。 接下来会对于TradingView平台介绍。 T…
阅读更多...
Android习题第五章数据存储

Android习题第五章数据存储

数据存储 本章习题 一、判断题 1&#xff0e;SQLite是Android自带的一个轻量级的数据库&#xff0c;支持基本SQL语法。&#xff08; &#xff09; 2&#xff0e;Android中的文件存储方式&#xff0c;分为内部存储方式和外部存储方式。&#xff08; &#xff09; 3&#xf…
阅读更多...
视频监控实现画面缩放功能

视频监控实现画面缩放功能

文章目录 概要一、功能说明二、核心实现代码三、技术细节 概要 在视频监控系统中&#xff0c;经常需要查看视频画面中的细节。通过实现区域放大、滚轮缩放和拖拽平移等功能&#xff0c;可以让用户更方便地观察视频细节。本文介绍如何在 Windows 系统下实现这些交互功能。 一、…
阅读更多...
k8s存储实验-使用nfs进行跨命名空间的多pod共享存储

k8s存储实验-使用nfs进行跨命名空间的多pod共享存储

提示&#xff1a;文章写完后&#xff0c;目录可以自动生成&#xff0c;如何生成可参考右边的帮助文档 文章目录 实验目标一、环境准备二、nfs环境的部署及连通性验证三、yaml文件的整理1. pv文件2.pvc文件3.pod文件 四、创建资源1.创建PV2.创建PVC3.创建pod 五、进行文件创建实…
阅读更多...
最新文章

Copyright @ 2022~2023