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),这里不再赘述。
在保持逻辑不变的情况下,这个函数还可以写成:
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(),因此可以省略显式的迭代器调用。选择哪种写法主要取决于代码风格和个人偏好。
还有另一种写法:
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 1010.1.2. 消除重复的步骤
- 识别重复代码
- 创建函数,把重复代码提取到函数体中,并在函数签名中指定函数的输入和返回值
- 用函数调用替换重复代码