526互联

29. 干货系列从零用Rust编写正反向代理，异步回调(async trait)的使用

发布时间 2023-11-21 09:24:35作者: 问蒙服务框架

wmproxy

wmproxy已用Rust实现http/https代理, socks5代理, 反向代理, 静态文件服务器，四层TCP/UDP转发，七层负载均衡，内网穿透，后续将实现websocket代理等，会将实现过程分享出来，感兴趣的可以一起造个轮子

项目地址

国内: https://gitee.com/tickbh/wmproxy

github: https://github.com/tickbh/wmproxy

关于 ++trait++

  trait是Rust中的概念，类似于其他语言中的接口（interface）。
  在Rust中不存在继承的概念，所有关于结构体的拓展功能全部均由trait来代替。比如std::io::Read这是一个关于io的trait，在TcpStream中和在File中均实现了该功能，这样子如果上层只关心读操作的，我们就可以将其转化成std::io::Read的一个对象，比如io: std::io::Read，后面我们也可以把他包裹成BufferReader等实现功能的转化。

为什么不在Rust中使用继承

  • 继承破坏了封装性，父类的改变会影响子类。如果父类的出现发生变化，所有继承自该父类的子类都需要相应地进行修改，这会增加代码的维护成本。
  • Rust追求内存安全和无数据竞争，继承不利于编译器进行静态检查。
  • 继承关系的耦合度高。子类和父类之间是紧密耦合的关系，这会影响代码的灵活性和可移植性。
  • 继承往往被过度使用,导致子类与父类功能紧密耦合。

下面举下例子，设计关于车的通用基类，能跑能停等

public class BaseCar { 
  //... 省略其他属性和方法... 
  public void run() { //... }
  public void stop() { //... }
}

一开始自行车都很完美，接下来设计摩托车，摩托车需要加油，那么基类被改成

public class BaseCar { 
  //... 省略其他属性和方法... 
  public void run() { //... }
  public void stop() { //... }
  public void refuel() { //... }
}

但是自行车又没有加油的需求

// 自行车
public class Bicycle extends BaseCar { 
 //... 省略其他属性和方法... 
 public void refuel() { 
     throw new UnSupportedMethodException("我不需要加油！");
 }
}

如果接下来又有修理引擎的接口，那基类又得加repairEngine的接口。自行车继承这个基类将会产生严重的负担，不继承又得重新写一些关于基础能力的函数，又会增加重复代码。

那么接下来是以trait方案的实现

pub trait Base {
   fn run(&self); 
   fn stop(&self); 
}
pub trait Refuel {
   fn refuel(&mut self); 
}
pub trait RepairEngine {
   fn repair_engine(&mut self); 
}

那么自行车只需要实现Base能力，然后摩托车在自行车的基础上实现Refuel及RepairEngine即可实现解耦。

异步的trait

在程序中均使用的是异步(async)编程，那么我们可能需要将trait实现成：

pub trait Base {
   async fn run(&self); 
   async fn stop(&self); 
}

当我们如此写的时候编译器就会提示我们：

functions in traits cannot be declared `async`
`async` trait functions are not currently supported
consider using the `async-trait` crate: https://crates.io/crates/async-trait
see issue #91611 <https://github.com/rust-lang/rust/issues/91611> for more informationrustcClick for full compiler diagnostic

原来异步的trait实现还没有进入到stable阶段，暂时只能有预览版即nightly版本进行使用。

那么本文将探讨该功能在未stable前如何实现异步的trait。

假如返回一个异步的Future

trait Base {
    type FetchData<'a>: std::future::Future<Output = String> + 'a where Self: 'a;
    fn run<'a>(&'a self) -> Self::FetchData<'a>;
}

那么实现自行车的函数将为：

trait Base {
    type FetchData<'a>: /* 将要何种类型呢？？ */;
    fn run<'a>(&'a self) -> Self::FetchData<'a>;
}

我们尝试过各种类型，编译器都无法通过编译，所以我们需要进行返回值的修改，我们将通过运行时类型擦除来实现。

首先，我们可以通过用 擦除 future 类型来避免编写 future 类型。以上面的例子为例，你可以这样写你的特征：dyn

trait Base {
    fn run<'a>(&'a self) -> Pin<Box<dyn Future<Output = String> + Send + '_>>;
}

那么实现将为：

impl Base for Bicycle {
    fn run<'a>(&'a self) -> std::pin::Pin<Box<dyn std::future::Future<Output = String> + Send + '_>> {
        Box::pin(async {
            "ok".to_string()
        })
    }
}

可以看出整个函数非常的冗余，相当的让人难受。

那么此时我们可以借助async-trait的宏处理库，他将帮我们自动处理掉无用的数据，那么我们的代码将变成如下：

#[async_trait]
trait Base {
    async fn run(&self) -> String;
}
#[async_trait]
impl Base for Bicycle {
    async fn run(&self) -> String {
        "ok".to_string()
    }
}

当然现在此方法会造成额外的开销，像Box，Send等都会造成一定的性能损失，如果要零损失实现异步还可以尝试以下方案

手动实现Poll

需要零开销或在no_std上下文中工作的特征还有另一种选择：它们可以从 Future 特征中获取轮询的概念，并将其直接构建到它们的界面中。如果 future 已完成，并且 future 正在等待其他事件，则该方法将返回。Future::poll,Poll::Ready(Output),Poll::Pending

pub trait Base {
    type Item;
    fn poll_next(
        self: Pin<&mut Self>,
        cx: &mut Context<'_>
    ) -> Poll<Option<Self::Item>>;
}

当然控制Poll的方式相当的麻烦，只要在对性能要求极高的情况下在进行此操作。

预期的官方实现

在最新的Beta或者nightly版本中可以用#![feature(async_fn_in_trait)]来启用该能力，那么我们就可以如下编程：

#![feature(async_fn_in_trait)]

trait Base {
    async fn run(&self) -> String;
}

impl Base for Bicycle {
    async fn run(&self) -> String {
        "ok".to_string()
    }
}

这样子就和普通的实现没有什么差别了。

实现该功能的难点

理论上来说，一个异步只有你在调用await的时候他才会真正的被调用，如果在此前有引用对话的存在，那么他的生命周期管理才是比较麻烦的存在。

小结

当前的Rust版本为1.74.0，好消息的是当前async trait已经Beta Channel了，如果不出意外的话下一次发布版本的稳定版将会拥有该能力了。该功能的官方实现将会给异步编程的带来极大的方便。让async/await能力越来越强。预期2023年末就可以直接使用了。下一章节我们将讲async trait在项目中的应用。

点击 [关注]，[在看]，[点赞] 是对作者最大的支持