在迭代器中找到第一个特定的枚举变体并对其进行转换
Posted
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了在迭代器中找到第一个特定的枚举变体并对其进行转换相关的知识,希望对你有一定的参考价值。
我有一个具有不同变体的枚举,我想找到匹配的第一个变体然后通过返回变量值或将其映射到其他变换来变换它。
在Scala中,我会使用case类来执行以下操作:
data.collectFirst{ case d: DataD => d.data }
在Rust中,我必须两次模式匹配才能获得相同的结果。有没有办法让它减少冗长?
enum MyData {
DataA(String),
DataB(u64),
DataC(bool),
DataD { data: String, val: u32 },
}
fn main() {
// test data
let data = vec![
MyData::DataB(42),
MyData::DataD {
data: "meaning of life".to_owned(),
val: 42,
},
MyData::DataC(false),
];
// find first one that matches and map it
let found: Option<String> = data.iter()
.find(|d| match **d {
MyData::DataD { .. } => true,
_ => false,
})
.and_then(|d| match *d {
MyData::DataD { ref data, .. } => Some(data.to_owned()),
_ => None,
});
}
答案
你可以使用Iterator::filter_map
:
let found: Option<String> = data.iter()
.filter_map(|d| {
match *d {
MyData::DataD { ref data, .. } => Some(data.to_owned()),
_ => None,
}
})
.next();
我真的不喜欢在我的迭代器链中使用大的match
语句,所以我通常在MyData
上使用一个方法来代替:
enum MyData {
DataA(String),
DataB(u64),
DataC(bool),
DataD { data: String, val: u32 },
}
impl MyData {
fn as_data_d_data(&self) -> Option<&str> {
match *self {
MyData::DataD { ref data, .. } => Some(data),
_ => None,
}
}
}
fn main() {
let data = vec![
MyData::DataB(42),
MyData::DataD {
data: "meaning of life".to_owned(),
val: 42,
},
MyData::DataC(false),
];
let found: Option<String> = data.iter()
.filter_map(MyData::as_data_d_data)
.map(str::to_owned)
.next();
}
事实上,我的很多枚举都有这种模式:
enum MyData {
DataA(String),
DataB(u64),
DataC(bool),
DataD(D), // made into a struct
}
impl MyData {
fn is_a(&self) -> bool {
match *self {
MyData::DataA(..) => true,
_ => false,
}
}
fn as_a(&self) -> Option<&String> {
match *self {
MyData::DataA(ref x) => Some(x),
_ => None,
}
}
fn into_a(self) -> Option<String> {
match self {
MyData::DataA(x) => Some(x),
_ => None,
}
}
// Repeat for all variants
}
我甚至创建了一个箱子来自动导出这些功能。我从来没有发表它,因为我很确定类似的东西已经存在,我只是没有找到它......
以上是关于在迭代器中找到第一个特定的枚举变体并对其进行转换的主要内容,如果未能解决你的问题,请参考以下文章
(C++, boost::variant) boost 变体映射的数据类型并对其执行数学运算