rust - 如果 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

很少有“唯一的出路”，这种情况也不异常(exception)。一种替代方法是使用静态分派(dispatch)而不是动态分派(dispatch)。

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

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 中只有一次分派(dispatch)到正确的函数。声明。

(我可能在这里弄错了一些细节，但你明白了。)

另一方面，您的版本使用类型 Box<dyn Iterator<Item = String>> ，所以迭代器会在堆上分配，每次都会有一个动态调度 next()在迭代器上调用。这可能没问题。

肯定有更多的方法来构建代码并在两种不同类型的输入之间进行调度，例如使用 Either从 either 输入 crate ，或者简单地写两个不同的 for这两种情况的循环。选择哪一个取决于您对代码的其他要求、您的性能要求和您的个人偏好的权衡。