Rust技巧——你可能还不知道的10个小技巧（上）

2022-01-26 本文已影响0人 L我是小学生

0x00 开篇

Rust 是一种很棒的编程语言：可靠、快速，但是也相当复杂，学习曲线相对来说比较陡峭。在过去的三年里，我一直在致力于学习Rust，并使用Rust来写了一些小工具，感觉效果还是蛮不错的。今天给大家介绍下Rust的10个小技巧。

0x01 在 VS Code 上使用 Rust Analyzer

VS Code中有两款插件，其中一款是Rust官方插件，另一款是 matklad 大神的 Rust Analyzer 。其实在使用中很明显Rust Analyzer是优于Rust官方插件的（即使官方插件下载量第一）。如果喜欢使用VS Code的小伙伴学习Rust，那我还是强推Rust Analyzer 。

image-20220126215751505

0x02 使用 Cow<str> 作为返回类型

有的时候，需要编写接受字符串切片（&str）并有条件地返回其修改的值或原始值的方法。对于上面说的两种情况，可以使用 Cow ，以便于仅在必要时分配新内存。示例代码如下：

use std::borrow::Cow;

/// 
/// 将单词的首字母转化为大写
fn capitalize(name: &str) -> Cow<str> {
    match name.chars().nth(0) {
        Some(first_char) if first_char.is_uppercase() => {
            // 如果首字母已经是大写，则无需分配内存
            Cow::Borrowed(name)
        }
        Some(first_char) => {
            // 如果首字母非大写，则需分配新内存
            let new_string: String = first_char.to_uppercase()
              .chain(name.chars().skip(1))
              .collect();
            
            Cow::Owned(new_string)
        },
        None => Cow::Borrowed(name),
    }
}

fn main() {
    println!("{}", capitalize("english"));   // 分配内存
    println!("{}", capitalize("China"));  // 不会分配内存
}

0x03 使用crossbeam_channel代替标准库中的channel

crossbeam_channel为标准库中的channel提供了强大的替代方案，支持 Select 操作、超时等。类似于你在Golang和传统的Unix Sockets中开箱即用的东西。示例代码如下：

use crossbeam_channel::{select, unbounded};
use std::time::Duration;

fn main() {
    let (s1, r1) = unbounded::<i32>();
    let (s2, r2) = unbounded::<i32>();
    s1.send(10).unwrap();

    select! {
        recv(r1) -> msg => println!("r1 > {}", msg.unwrap()),
        recv(r2) -> msg => println!("r2 > {}", msg.unwrap()),
        default(Duration::from_millis(100)) => println!("timed out"),
    }
}

0x04 使用dbg!()代替println!()

如果你是在调试代码，那么我推荐你使用dbg!()这个宏，它可以用更少的代码来输出更多的信息。示例代码如下：

fn main() {
    let test = 5;

    println!("{}", test); // 输出: 5
    dbg!(test);// 输出:[src/main.rs:4] test = 2
}

0x05 类似Golang中的 defer 操作

如果你熟悉Golang，那你肯定知道 defer用于注册延迟调用。像在使用原始指针和关闭Sockets时释放内存都会用到。当然在Rust中出了RAII模式之外，也可以使用 scopeguard crate 来实现“clean up”操作。示例代码如下：

#[macro_use(defer)] extern crate scopeguard;

fn main() {
    println!("start");
    {
        // 将在当前作用域末尾执行
        defer! {
           println!("defer");
        }

        println!("scope end");
    }
    println!("end");
}

程序执行结果：

start
scope end
defer
end

0x06 部分工具地址

Cow<str>文档：https://doc.rust-lang.org/std/borrow/enum.Cow.html
Crossbeam Github：https://github.com/crossbeam-rs/crossbeam
scopeguard crate ：https://docs.rs/scopeguard/latest/scopeguard/