组合算子:map

match 是处理 Option 的一个可用的方法,但你会发现大量使用它会很繁琐,特别是当 操作只对一种输入是有效的时。这时,可以使用组合算子(combinator),以 模块化的风格来管理控制流。

Option 有一个内置方法 map(),这个组合算子可用于 Some -> SomeNone -> None 这样的简单映射。多个不同的 map() 调用可以串起来,这样更加灵活。

在下面例子中,process() 轻松取代了前面的所有函数,且更加紧凑。

  1. #![allow(dead_code)]
  2. #[derive(Debug)] enum Food { Apple, Carrot, Potato }
  3. #[derive(Debug)] struct Peeled(Food);
  4. #[derive(Debug)] struct Chopped(Food);
  5. #[derive(Debug)] struct Cooked(Food);
  6. // 削皮。如果没有食物,就返回 `None`。否则返回削好皮的食物。
  7. fn peel(food: Option<Food>) -> Option<Peeled> {
  8. match food {
  9. Some(food) => Some(Peeled(food)),
  10. None => None,
  11. }
  12. }
  13. // 切食物。如果没有食物,就返回 `None`。否则返回切好的食物。
  14. fn chop(peeled: Option<Peeled>) -> Option<Chopped> {
  15. match peeled {
  16. Some(Peeled(food)) => Some(Chopped(food)),
  17. None => None,
  18. }
  19. }
  20. // 烹饪食物。这里,我们使用 `map()` 来替代 `match` 以处理各种情况。
  21. fn cook(chopped: Option<Chopped>) -> Option<Cooked> {
  22. chopped.map(|Chopped(food)| Cooked(food))
  23. }
  24. // 这个函数会完成削皮切块烹饪一条龙。我们把 `map()` 串起来,以简化代码。
  25. fn process(food: Option<Food>) -> Option<Cooked> {
  26. food.map(|f| Peeled(f))
  27. .map(|Peeled(f)| Chopped(f))
  28. .map(|Chopped(f)| Cooked(f))
  29. }
  30. // 在尝试吃食物之前确认食物是否存在是非常重要的!
  31. fn eat(food: Option<Cooked>) {
  32. match food {
  33. Some(food) => println!("Mmm. I love {:?}", food),
  34. None => println!("Oh no! It wasn't edible."),
  35. }
  36. }
  37. fn main() {
  38. let apple = Some(Food::Apple);
  39. let carrot = Some(Food::Carrot);
  40. let potato = None;
  41. let cooked_apple = cook(chop(peel(apple)));
  42. let cooked_carrot = cook(chop(peel(carrot)));
  43. // 现在让我们试试看起来更简单的 `process()`。
  44. let cooked_potato = process(potato);
  45. eat(cooked_apple);
  46. eat(cooked_carrot);
  47. eat(cooked_potato);
  48. }

参见:

闭包, Option, 和 Option::map()