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 。
0x02 使用 Cow作为返回类型
有的时候,需要编写接收字符串切片 (&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/