10.1 提取函数以消除重复代码
10.1.1. 重复代码
看个例子:
fn main(){
let number_list = vec![1,2,3,4,5];
let mut largest = number_list[0];
for &item in number_list.iter(){
if item > largest{
largest = item;
}
}
println!("The largest number is {}", largest);
}
这个程序的目的是找出 Vector 中的最大值。其逻辑很好理解:把第一个元素作为临时最大值,再使用循环比较 Vector 中的每个元素。如果当前元素大于当前存储的最大值,就把当前元素赋给 largest。
输出:
The largest number is 5
如果此时又新增了一个需求,需要在另一个 Vector 中找出最大值,仍然可以按同样的逻辑来写:
fn main(){
let number_list = vec![1,2,3,4,5];
let mut largest = number_list[0];
for &item in number_list.iter(){
if item > largest{
largest = item;
}
}
println!("The largest number is {}", largest);
let number_list = vec![6,7,8,9,10];
let mut largest = number_list[0];
for &item in number_list.iter(){
if item > largest{
largest = item;
}
}
println!("The largest number is {}", largest);
}
但可以看出,这样写会产生大量重复代码。
重复代码容易出错。一旦需要修改逻辑,就必须在多处做同样的修改。
因此非常推荐通过定义函数来创建抽象。代码如下:
fn largest(list: &[i32]) -> i32{
let mut largest = list[0];
for &item in list.iter(){
if item > largest{
largest = item;
}
}
largest
}
fn main(){
let number_list = vec![1,2,3,4,5];
let largest_num = largest(&number_list);
println!("The largest number is {}", largest_num);
let number_list = vec![6,7,8,9,10];
let largest_num = largest(&number_list);
println!("The largest number is {}", largest_num);
}
这里声明了一个名为 largest 的函数。它接收元素类型为 i32 的切片,并返回一个 i32。函数体内的逻辑与上文相同。需要注意的是,参数 &[i32] 是切片,本质上就是引用。切片的具体介绍见 4.5. 切片(Slice),这里不再赘述。
在保持逻辑不变的情况下,这个函数还可以写成:
#![allow(unused)]
fn main() {
fn largest(list: &[i32]) -> i32{
let mut largest = list[0];
for &item in list{
if item > largest{
largest = item;
}
}
largest
}
}
与前一版本相比,这里去掉了显式的迭代器调用 .iter(),但不影响代码行为,因为切片引用本身实现了 IntoIterator,所以 for 可以直接遍历 list。这两种写法在语义上完全等价。Rust 的 for 循环会为切片自动调用 iter(),因此可以省略显式的迭代器调用。选择哪种写法主要取决于代码风格和个人偏好。
还有另一种写法:
#![allow(unused)]
fn main() {
fn largest(list: &[i32]) -> i32{
let mut largest = list[0];
for item in list{
if *item > largest{
largest = *item;
}
}
largest
}
}
这个版本与前两种最大的不同是:它显式地对 item 解引用(*item),以便比较其值。
在前两种写法中,使用了通过解引用模式匹配进行解构。可以这样理解:&item = &i32,两边同时去掉 &,于是 item = i32。largest 也是 i32 类型,两者类型一致,可以直接比较,后面自然就不需要再解引用。如果 item 前面不加 &,那么 item 的类型就是 &i32,而 largest 是 i32,两者不能直接比较,所以必须先解引用,也就是在 item 前面加 *。
输出:
The largest number is 5
The largest number is 10
10.1.2. 消除重复的步骤
- 识别重复代码
- 创建函数,把重复代码提取到函数体中,并在函数签名中指定函数的输入和返回值
- 用函数调用替换重复代码