自己动手,丰衣足食------Rust实现自己的Stream

面对tokio::fs模块在处理大文件时的性能问题,作者决定自行实现Stream以提高效率。文章介绍了Stream的定义,即类似Future但增加了Ready(None)状态来指示结束。作者创建了一个线程读取Reader并将结果写入Stream,使用Waker唤醒线程并利用mpsc通道避免数据覆盖。尽管过程中遇到channel的多生产者单消费者特性以及编译器对Box和generic type的要求,作者成功实现了自定义Stream来优化异步磁盘操作。
摘要由CSDN通过智能技术生成

想写个简单的对象存储,折腾了两天大框写好了,结果一试,600M的稳健用了102秒,上网一查,这好像还不是个例, tokio::fs里面的东西好像普遍性能低下,而且低的不是一点半点。记得原来看的某篇文章里提过一句,说windows是有原生的异步磁盘操作的, linux下没有,不知道跟这个有没有关系。

想着用std::fs来替换,但是自己定的trait是围绕Stream来的,换的话等于推翻重写。思索在三,决定自己试着实现个Stream试一下。

先说Stream的定义:

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

跟Future差不多,只是有细微的差别,重点是对于返回值的定义,除了Ready(Some(Item))和Pending之外,还有个Ready(None)用来告诉调用者,所有内容都读取完毕了。别再poll我了。

以下是实现代码:

impl<T> MyStream<T>
where
    T: std::io::Read + Send + Sync,
{
    fn new(reader: T) -> Self {
        Self {
            inner: Arc::new(Mutex::new(reader)),
            cap: 1024 * 8,
            res: Arc::new(Mutex::new(None)),
      
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
你可以使用 Rust 编程语言来实现一个 STUN 客户端。STUN(Simple Traversal of UDP through NATs)是一种用于网络地址转换(NAT)穿透的协议。 下面是一个简单的示例代码,演示了如何使用 Rust 和 tokio 库来实现一个基本的 STUN 客户端: ```rust use std::net::{SocketAddr, UdpSocket}; use tokio::runtime::Runtime; fn main() { // 设置 STUN 服务器的地址 let stun_server: SocketAddr = "stun.example.com:3478".parse().unwrap(); // 创建 UDP 套接字 let socket = UdpSocket::bind("0.0.0.0:0").expect("Failed to bind socket"); // 发送 STUN 请求 let mut buf = [0u8; 1024]; let request: [u8; 20] = [ 0x00, 0x01, 0x00, 0x00, 0x21, 0x12, 0xA4, 0x42, // STUN 请求头部 0x00, 0x06, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // STUN 请求数据 0x00, 0x08, 0x00, 0x14, // STUN 请求数据长度 ]; socket .send_to(&request, stun_server) .expect("Failed to send STUN request"); // 接收 STUN 响应 let (recv_len, _) = socket.recv_from(&mut buf).expect("Failed to receive STUN response"); let response = &buf[..recv_len]; // 处理 STUN 响应 // 这里可以根据 STUN 协议解析 response 数据 println!("STUN response: {:?}", response); } ``` 这只是一个简单的示例,实际上你可能需要更多的代码来处理 STUN 协议的各种情况和错误处理。你可以使用其他 Rust 库来帮助解析和处理 STUN 响应数据。 请注意,这个示例使用了 tokio 库来实现异步网络编程。你需要在 `Cargo.toml` 文件中添加相应的依赖,例如: ```toml [dependencies] tokio = { version = "1", features = ["full"] } ``` 希望这个示例能对你有所帮助!如果你有任何其他问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值