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

0x00 开篇

Rust 是一种很棒的编程语言：可靠、快速，但是也相当复杂，学习曲线相对来说比较陡峭。今天给大家介绍下Rust的10个小技巧剩余的5个小技巧。

0x01 自定义并链接 Panic handlers

Panic Handler（通常也被称为 hooks）可以被重写和链接，这在为应用程序设置自定义错误报告和日志记录时会很有用。示例代码如下：

use std::panic::{set_hook, take_hook};

fn main() {
    let prev_hook = take_hook();
    
    set_hook(Box::new(move |panic| {
        println!("custom logging logic {}", panic);
        
        prev_hook(panic);    
    }));
    
    let prev_hook = take_hook();
    
    set_hook(Box::new(move |panic| {
        println!("custom error reporting logic {}", panic);
        
        prev_hook(panic);    
    }));
    
    panic!("test")
}

代码运行结果:

custom error reporting logic panicked at 'test', src/main.rs:20:5
custom logging logic panicked at 'test', src/main.rs:20:5

0x02 使用闭包时使用 impl Trait

尽可能的使用impl Fn/FnOnce/FnMut 来将闭包传递给函数（impl Trait）来保证签名的干净，而不是使用泛型。对于非正常的情况，你可能需要使用 Box<Fn()>装箱，但是可能将会耗费额外的内存开销。示例代码如下：

fn setup_teardown_generic<A: FnOnce()>(action: A) {
    println!("setting up...");
    action();
    println!("tearing down...")
}

fn setup_teardown(action: impl FnOnce()) {
    println!("setting up...");
    action();
    println!("tearing down...")
}

fn main() {
    setup_teardown(|| {
        println!("Action!");
    })
}

0x03 在VSCode中使用cargo clippy来lint

如果你使用的是VSCode + Rust Analyzer来编写Rust，我建议你使用cargo clippy来代替cargo check你进入 Rust Analyzer > Check On Save:Command，将check改为clippy。设备方法如下：

image-20220126220051982

0x04 与 C/C++ 代码集成

如果你的Rust代码要与 C/C++ 代码集成，推荐你使用这个库cc crate。它是一个将 C/C++/Assembly编译为 Rust 的crate或者app。示例代码如下：

// build.rs
// 将`foo.c`和`bar.c`编译为 `libfoo.a`
fn main() {
    cc::Build::new()
        .file("foo.c")
        .file("bar.c")
        .compile("foo");
}

下面是foo.c和bar.c的内容

void foo_function(void) { ... }

int32_t bar_function(int32_t x) { ... }

可以在Rust中使用:

extern {
    fn foo_function();
    fn bar_function(x: i32) -> i32;
}

pub fn call() {
    unsafe {
        foo_function();
        bar_function(42);
    }
}

fn main() {
   call();
}

0x05 使用Cargo-Make来打包

Build Script适用于很多事情，但通常不适用于打包。本人比较推荐的解决方案是Sagiegurari大神的的Cargo Make，一个基于Rust的任务运行器和构建工具。

0x06 部分工具地址

• Panic Handler：https://doc.rust-lang.org/std/panic/fn.set_hook.html
• Crossbeam Github：https://github.com/crossbeam-rs/crossbeam
• Cargo-Make Github： https://github.com/sagiegurari/cargo-make
• scopeguard crate ：https://docs.rs/scopeguard/latest/scopeguard/
• cc crate：https://crates.io/crates/cc
• Cargo-Make Github： https://github.com/sagiegurari/cargo-make