Rust 创建文件流抽象

这里有一种方法可以帮助您做到这一点.这里的诀窍是完全忽略生存期，将类型留给泛型.我们真的不需要知道我们从中读取的数字是什么，因为我们只需要std::io::Read就可以实现该类型.通过这种方法，您可以使用File、TcpStream、BufReader和许多其他类型的 crate ，这些 crate 也使用Read.&mut T是一种与T类似的类型，如果我们查看文档，那么所有&mut T where T: Read都实现了Read.

use std::io::{self, ErrorKind, Read};

pub struct ValueIter<R> {
    reader: R,
    found_end: bool,
}

impl<R: Read> ValueIter<R> {
    pub fn for_reader(reader: R) -> Self {
        ValueIter { 
            reader,
            found_end: false,
        }
    }
}

// `Read` is the trait that gives us `read_exact` 
impl<R: Read> Iterator for ValueIter<R> {
    type Item = io::Result<i64>;
    
    fn next(&mut self) -> Option<Self::Item> {
        if self.found_end {
            return None
        }
        
        let mut buffer = [0u8; 8];
        match self.reader.read_exact(&mut buffer) {
            // Read the next number
            Ok(_) => Some(Ok(i64::from_be_bytes(buffer))),
            // Handle end of file reached
            Err(e) if e.kind() == ErrorKind::UnexpectedEof => {
                self.found_end = true;
                None
            }
            // Return any other errors as part of the iterator
            Err(e) => Some(Err(e)),
        }
    }
}

然后它可以像这样使用.

pub fn read_loop<R: Read>(reader: &mut R) -> io::Result<i64> {
    let mut sum = 0;
    let mut count = 0;
    
    for value in ValueIter::for_reader(reader) {
        sum += value?;
        count += 1;
    }
    
    Ok(sum / count)
}

值得注意的是，既然我们让它通用了Read多个，那么我们就可以使用File多个.例如，您可以通过将文件包装在BufReader中来提高性能，这样它就可以从操作系统中获取更大的文件块，而无需在每次读取新数字时询问操作系统.

use std::fs::File;
use std::io::BufReader;
let mut file = BufReader::new(File::open("data.dat").expect("Unable to open file"));

println!("{:?}", read_loop(&mut file).unwrap());