Question

我像这样鼓掌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>>是唯一的选择。正确吗？

Answer 1

很少有“唯一的出路”，这种情况也没有什么不同。一种替代方法是使用静态调度而不是动态调度。

您的主要处理代码需要字符串的迭代器作为输入。因此，您可以定义如下处理函数：

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循环。选择哪一个取决于您对代码的其他要求，性能要求和个人偏好的权衡。

如果没有拍手给定位置参数，如何使用STDIN？

1 个答案: