我有一个拍手App
像这样:
let m = App::new("test")
.arg(
Arg::with_name("INPUT")
.help("a string to be frobbed")
.multiple(true),
)
.get_matches();
如果有任何myapp str1 str2 str3
,我想将参数读作字符串的迭代,但如果没有,则作为过滤器并从stdin cat afile | myapp
读取可迭代的行。这是我的尝试:
let stdin = io::stdin();
let strings: Box<Iterator<Item = String>> = if m.is_present("INPUT") {
Box::new(m.values_of("INPUT").unwrap().map(|ln| ln.to_string()))
} else {
Box::new(stdin.lock().lines().map(|ln| ln.unwrap()))
};
for string in strings {
frob(string)
}
我相信,因为我只是要求Iterator
特性,Box<Iterator<Item = String>>
是唯一的出路。那是对的吗?
很少有“唯一的出路”,这种情况也不例外。一种替代方法是使用静态分派而不是动态分派。
您的主要处理代码需要一个字符串迭代器作为输入。所以你可以定义一个像这样的处理函数:
fn process<I: IntoIterator<Item = String>>(strings: I) {
for string in strings {
frob(string);
}
}
调用此代码可能如下所示:
match m.values_of("INPUT") {
Some(values) => process(values.map(|ln| ln.to_string())),
None => process(io::stdin().lock().lines().map(|ln| ln.unwrap())),
}
编译器将发出两个不同版本的process()
,每个迭代器类型一个。每个版本静态调用它为其编译的迭代器函数,并且在match
语句中只有一个调度到右函数。
(我可能在这里弄错了一些细节,但你明白了。)
另一方面,您的版本使用类型Box<dyn Iterator<Item = String>>
,因此迭代器将在堆上分配,并且每次在迭代器上调用next()
时都会有动态调度。这可能很好。
当然,有更多的方法来构造代码并在两种不同类型的输入之间进行调度,例如:使用来自Either
箱子的either
类型,或者简单地为两个案例写两个不同的for
循环。选择哪一个取决于与代码的其他要求,您的性能要求和个人偏好的权衡。