Rust 如何强制匹配的返回类型为()

发布于08月17日

在下面的简化示例中，编译器报告match语句的一个分支返回bool，而另一个返回().

use std::collections::{HashMap, HashSet};

fn main() {
    let available = HashMap::from_iter([(2, "b"), (3, "c"), (4, "d")]);
    let mut set = HashSet::new();
    
    for i in [1, 2, 3, 4, 5] {
        match available.get(&i) {
            Some(s) => set.insert(*s),
            None => ()
        }
    }
}

但是，这会导致错误:

error[E0308]: `match` arms have incompatible types
  --> src/main.rs:10:21
   |
8  | /         match available.get(&i) {
9  | |             Some(s) => set.insert(*s),
   | |                        -------------- this is found to be of type `bool`
10 | |             None => ()
   | |                     ^^ expected `bool`, found `()`
11 | |         }
   | |_________- `match` arms have incompatible types

我如何通知编译器Match语句应该返回()，而从insert返回的bool应该被忽略？

推荐答案

通常的方法是将手臂放在一个块中，并为其加上;，因为这会 suppress 表达式的结果:

use std::collections::{HashMap, HashSet};

fn main() {
    let available = HashMap::from_iter([(2, "b"), (3, "c"), (4, "d")]);
    let mut set = HashSet::new();
    
    for i in [1, 2, 3, 4, 5] {
        match available.get(&i) {
            Some(s) => {
                set.insert(*s);
            }
            None => (),
        }
    }
}

一种可能不太常见的替代方法是将值反馈给一个函数，将其"转换"为()，例如drop

use std::collections::{HashMap, HashSet};

fn main() {
    let available = HashMap::from_iter([(2, "b"), (3, "c"), (4, "d")]);
    let mut set = HashSet::new();
    
    for i in [1, 2, 3, 4, 5] {
        match available.get(&i) {
            Some(s) => drop(set.insert(*s)),
            None => (),
        }
    }
}

或者，您也可以完全忽略第二个分支:

use std::collections::{HashMap, HashSet};

fn main() {
    let available = HashMap::from_iter([(2, "b"), (3, "c"), (4, "d")]);
    let mut set = HashSet::new();
    
    for i in [1, 2, 3, 4, 5] {
        if let Some(s) = available.get(&i) {
            set.insert(*s);
        }
    }
}

或者使用更高阶的函数来处理这种副作用，虽然这不是最常见的，但它是一件事:

use std::collections::{HashMap, HashSet};

fn main() {
    let available = HashMap::from_iter([(2, "b"), (3, "c"), (4, "d")]);
    let mut set = HashSet::new();
    
    for i in [1, 2, 3, 4, 5] {
        available.get(&i).map(|&s| set.insert(s));
    }
}