在命令行上接受可选文件，默认为stdin

Question

Closed. This question is opinion-based. It is not currently accepting answers.

---

想要改进这个问题吗？更新问题，以便可以通过。

闭合 2年前。

该帖子已编辑并提交以进行评论 2年前，未能重新打开帖子：

原始关闭原因未解决

Answer 1

您可以使用 option＆lt; pathbuf＆gt; 并使用 stdin 当 none 时，所有内容都包含在方法中（是，您必须实现它自己）：

use std::io::BufReader;
use clap;
use std::io::BufRead;
use std::path::PathBuf; // 3.1.6

#[derive(clap::Parser)]
struct Reader {
    input: Option<PathBuf>,
}

impl Reader {
    fn reader(&self) -> Box<dyn BufRead> {
        self.input.as_ref()
            .map(|path| {
                Box::new(BufReader::new(std::fs::File::open(path).unwrap())) as Box<dyn BufRead>
            })
            .unwrap_or_else(|| Box::new(BufReader::new(std::io::stdin())) as Box<dyn BufRead>)
    }
}

Answer 2

该解决方案：不使用动态调度，是详细的，使用 default_value_t ，需要夜间参见 ＃93965 （应该不需要每晚都需要，但对我来说更简单）

use clap; // 3.1.6

use std::fmt::{self, Display, Formatter};
use std::fs::File;
use std::io::{self, stdin, BufRead, BufReader, Read, StdinLock};
use std::str::FromStr;

enum Input {
    Stdin(StdinLock<'static>),
    File(BufReader<File>),
}

impl Display for Input {
    fn fmt(&self, fmt: &mut Formatter<'_>) -> Result<(), fmt::Error> {
        write!(fmt, "Input")
    }
}

impl Default for Input {
    fn default() -> Self {
        Self::Stdin(stdin().lock())
    }
}

impl FromStr for Input {
    type Err = io::Error;

    fn from_str(path: &str) -> Result<Self, <Self as FromStr>::Err> {
        File::open(path).map(BufReader::new).map(Input::File)
    }
}

impl BufRead for Input {
    fn fill_buf(&mut self) -> Result<&[u8], io::Error> {
        match self {
            Self::Stdin(stdin) => stdin.fill_buf(),
            Self::File(file) => file.fill_buf(),
        }
    }

    fn consume(&mut self, amt: usize) {
        match self {
            Self::Stdin(stdin) => stdin.consume(amt),
            Self::File(file) => file.consume(amt),
        }
    }
}

impl Read for Input {
    fn read(&mut self, buf: &mut [u8]) -> Result<usize, io::Error> {
        match self {
            Self::Stdin(stdin) => stdin.read(buf),
            Self::File(file) => file.read(buf),
        }
    }
}

#[derive(clap::Parser)]
struct Reader {
    #[clap(default_value_t)]
    input: Input,
}