Question

作为rust的新手，我想知道是否有办法使用单个函数/宏/任何其他方法从传递的文件或stdin中读取一行，也许将一种缓冲区读取器作为参数传递？ / p>

我还没有找到任何有用的东西，下面的代码可以正常工作，一旦我能够在宏上包装一些验证，我知道代码可以改进。我愿意就如何真正改善该宏提出建议

...

macro_rules! validate {
    ($line:expr, $blank:expr, $squeeze:expr, $line_count:expr, $show_number:expr) => {
        if $line.len() <= 0 {
            $blank +=1;
        } else{
            $blank = 0;
        }
        if $squeeze & ($blank > 1) {
            continue;
        }
        if $show_number {
            $line_count += 1;
        }

    }
} 


...

for file in opt.files {
        blank_line_count = 0;
        line_count = 0;
        if file.to_str() != Some("-") {
            let f = File::open(file)?;
            for line in BufReader::new(f).lines() {
                let l = line.unwrap();
                validate!(l, blank_line_count, opt.squeeze_blank, line_count, opt.number); // will continue the loop if not valid
                println!("{}", format_line(l, opt.show_ends, opt.show_tabs, opt.show_nonprinting, line_count)); // will be skipped if not valid
            }
        } else {
            let stdin = io::stdin();
            let mut bytes_read: usize;
            loop {
                let mut line = String::new();
                bytes_read = stdin.lock().read_line(&mut line).expect("Could not read line");
                if bytes_read == 0 { break; }
                line.pop();
                validate!(line, blank_line_count, opt.squeeze_blank, line_count, opt.number);// will continue the loop if not valid
                println!("{}", format_line(line, opt.show_ends, opt.show_tabs, opt.show_nonprinting, line_count)); // will be skipped if not valid
            }
        }
    }
....

如图所示，File和stdin具有不同的处理方式，但是它们基本上都做同样的事情，通过循环查找有效条目来实现

Answer 1

感谢@PeterHall，这是Read trait东西点亮了，我没有意识到我可以将stdin传递给BufReader，所以就可以了：

 let stdin = io::stdin();
 for line in BufReader::new(stdin).lines() {
...

与某人相同的方式：

let f = File::open(file)?;
for line in BufReader::new(f).lines() {

这就是我想要的。

谢谢你

从stdin或rust的文件中读取特征？

1 个答案: