组合算子:and_then
map() 以链式调用的方式来简化 match 语句。然而,如果以返回类型是 Option<T>
的函数作为 map() 的参数,会导致出现嵌套形式 Option<Option<T>>。这样多层串联
调用就会变得混乱。所以有必要引入 and_then(),在某些语言中它叫做 flatmap。
and_then() 使用被 Option 包裹的值来调用其输入函数并返回结果。 如果 Option
是 None,那么它返回 None。
在下面例子中,cookable_v2() 会产生一个 Option<Food>。如果在这里使用 map()
而不是 and_then() 将会得到 Option<Option<Food>>,这对 eat() 来说是一个
无效类型。
1. #![allow(dead_code)]
3. #[derive(Debug)] enum Food { CordonBleu, Steak, Sushi }
4. #[derive(Debug)] enum Day { Monday, Tuesday, Wednesday }
6. // 我们没有制作寿司所需的原材料(ingredient)(有其他的原材料)。
7. fn have_ingredients(food: Food) -> Option<Food> {
8. match food {
9. Food::Sushi => None,
10. _ => Some(food),
11. }
12. }
14. // 我们拥有全部食物的食谱,除了法国蓝带猪排(Cordon Bleu)的。
15. fn have_recipe(food: Food) -> Option<Food> {
16. match food {
17. Food::CordonBleu => None,
18. _ => Some(food),
19. }
20. }
23. // 要做一份好菜,我们需要原材料和食谱。
24. // 我们可以借助一系列 `match` 来表达这个逻辑:
25. fn cookable_v1(food: Food) -> Option<Food> {
26. match have_ingredients(food) {
27. None => None,
28. Some(food) => match have_recipe(food) {
29. None => None,
30. Some(food) => Some(food),
31. },
32. }
33. }
35. // 也可以使用 `and_then()` 把上面的逻辑改写得更紧凑:
36. fn cookable_v2(food: Food) -> Option<Food> {
37. have_ingredients(food).and_then(have_recipe)
38. }
40. fn eat(food: Food, day: Day) {
41. match cookable_v2(food) {
42. Some(food) => println!("Yay! On {:?} we get to eat {:?}.", day, food),
43. None => println!("Oh no. We don't get to eat on {:?}?", day),
44. }
45. }
47. fn main() {
48. let (cordon_bleu, steak, sushi) = (Food::CordonBleu, Food::Steak, Food::Sushi);
50. eat(cordon_bleu, Day::Monday);
51. eat(steak, Day::Tuesday);
52. eat(sushi, Day::Wednesday);
53. }
