如果没有给出clap的位置参数，我该如何使用STDIN？

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