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