Rust 中是否有丢弃输入的标准方法？

Question

在 Rust 中从 stdin 读取输入非常简单：

use std::io;

let mut some_string = String::new();
let _ = io::stdin().read_line(&mut some_string);

但有时您可能只想丢弃它而不在任何地方确认它或获取缓冲区，例如：

println!("Press Enter to close");
io::stdin().discard_input(); // discard_input() doesn't exist

它可以读取任意数量的字符，直到遇到\n，忽略它并且什么也不返回或io::Result。有没有标准的方法，还是我必须自己实现？我追踪了read_line 的实现到BufRead 实现的read_until，但我不确定从那里开始是否是个好主意。

Answer 1

您可以在标准输入上实现.discard_until_newline()，例如：

这应该是一个正确且高效的实现（使用缓冲区读取，但不将读取的部分复制到任何地方）。当然，如果您不期望任何长行并且一次只读取一个字节，那可能会简单得多。

use std::io::Stdin;
use std::io;
use std::io::prelude::*;

pub trait DiscardUntil {
    fn discard_until_newline(&mut self) -> Result<(), io::Error>;
}

impl DiscardUntil for Stdin {
    fn discard_until_newline(&mut self) -> Result<(), io::Error> {
        let mut buffered = self.lock();
        loop {
            let (consume, done) = {
                let data = try!(buffered.fill_buf());
                if data.is_empty() {
                    (0, true)
                } else if let Some(newline_pos) = data.iter().position(|b| *b == b'\n') {
                    (newline_pos + 1, true)
                } else {
                    (data.len(), false)
                }
            };
            buffered.consume(consume);
            if done {
                break;
            }
        }
        Ok(())
    }
}

fn main() {
    println!("hi");
    io::stdin().discard_until_newline().unwrap();
    println!("again");
    io::stdin().discard_until_newline().unwrap();
    println!("goodbye");
}

Answer 2

我对 bluss 的代码进行了实验，并且能够提出一个性能稍差的实现，并且不太适合某些极端情况（参见下面的 cmets），但更简单、更短：

use std::io;
use std::io::{Stdin, BufRead};

pub trait DiscardUntil {
    fn discard_until_newline(&mut self) -> Result<(), io::Error>;
}

impl DiscardUntil for Stdin {
    fn discard_until_newline(&mut self) -> Result<(), io::Error> {
        let mut buffered = self.lock();
        let amount = {
            let data = try!(buffered.fill_buf());
            data.len()
        };
        buffered.consume(amount);
        Ok(())
    }
}

我欢迎尝试设计一个更简单的解决方案。