Rust private function 在尝试返回值元组时抛出错误

Question

我有一个 function 我正在尝试从以下位置返回一个值元组：

fn get_two_bytes(data: &[u8]) -> (Vec<(u8, u8)>, BTreeSet<(u8, u8)>) {
    let two_byte_vec = data
        .chunks(2)
        .map(|x| (x[0], x[1]))
        .collect::<Vec<_>>();

    let mut two_byte_set : BTreeSet<&(u8, u8)> = BTreeSet::new();
    for n in &two_byte_vec {
        two_byte_set.insert(n);
    }

    return (two_byte_vec, two_byte_set);
}

生成此错误：

error[E0308]: mismatched types
  --> src/lib.rs:13:27
   |
13 |     return (two_byte_vec, two_byte_set);
   |                           ^^^^^^^^^^^^ expected tuple, found `&(u8, u8)`
   |
   = note: expected struct `BTreeSet<(u8, u8)>`
              found struct `BTreeSet<&(u8, u8)>`

我显然不想返回&two_byte_set - 我想将所有权从 function 转移出去。如何让这两个变量正确返回？

Answer 1

此处的错误消息具有误导性。

它谈论的是BTreeMap持有的类型，它是&(u8, u8)而不是(u8, u8)你 promise Int function 签名。 由于u8是Copy ，元组(u8, u8)也是Copy ，这意味着您可以在插入 map 之前取消引用n并从two_byte_set中删除类型注释。

use std::collections::BTreeSet;

fn get_two_bytes(data: &[u8]) -> (Vec<(u8, u8)>, BTreeSet<(u8, u8)>) {
    let two_byte_vec = data.chunks(2).map(|x| (x[0], x[1])).collect::<Vec<_>>();

    let mut two_byte_set = BTreeSet::new();
    for &n in &two_byte_vec {
        two_byte_set.insert(n);
    }

    (two_byte_vec, two_byte_set);
}

Answer 2

由于您借用向量在for循环中迭代，因此该循环中n的类型是&(u8, u8) 。 幸运的是，可以克隆u8的元组，因此只需将它们克隆到您的集合中即可。

let mut two_byte_set : BTreeSet<(u8, u8)> = BTreeSet::new();
for n in &two_byte_vec {
    two_byte_set.insert(n.to_owned());
}