如果 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，我想将参数作为可迭代的字符串读取，但如果没有，则充当过滤器并从标准输入 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

很少有 "only way to go"，这种情况也不例外。一种替代方法是使用静态分派而不是动态分派。

您的主要处理代码需要一个字符串迭代器作为输入。所以你可以像这样定义一个处理函数：

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

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

How do I use STDIN if no positional arguments are given with clap?

rust

clap