Jak utworzyć obiekt in-memory, który może być używany jako czytnik lub program zapisujący w Rust?

Question

Potrzebuję obiektu całkowicie in-memory, który mogę dać BufReader i BufWriter. Coś jak Python StringIO. Chcę pisać i czytać z takiego obiektu za pomocą metod zwykle używanych z File s.

Czy można to zrobić za pomocą standardowej biblioteki?

Answer 1

W rzeczywistości istnieje sposób. Poznaj Cursor<T>!

W dokumentacji można zobaczyć, że istnieją następujące impls:

impl<T> Seek for Cursor<T> where T: AsRef<[u8]> 
impl<T> Read for Cursor<T> where T: AsRef<[u8]> 
impl Write for Cursor<Vec<u8>> 
impl<T> AsRef<[T]> for Vec<T> 

Z tego widać, że można użyć typu Cursor<Vec<u8>> tylko jako zwykły plik, ponieważ Read, Write i Seek są zaimplementowane dla tego typu!

mały przykład (Playground):

use std::io::{Cursor, Read, Seek, SeekFrom, Write}; 

// Create fake "file" 
let mut c = Cursor::new(Vec::new()); 

// Write into the "file" and seek to the beginning 
c.write_all(&[1, 2, 3, 4, 5]).unwrap(); 
c.seek(SeekFrom::Start(0)).unwrap(); 

// Read the "file's" contents into a vector 
let mut out = Vec::new(); 
c.read_to_end(&mut out).unwrap(); 

println!("{:?}", out); 

Dla bardziej użyteczny przykład sprawdzić dokumentację związaną powyżej.

Answer 2

Jeśli chcesz użyć BufReader z in-memory String, można użyć metody as_bytes():

use std::io::BufRead; 
use std::io::BufReader; 
use std::io::Read; 

fn read_buff<R: Read>(mut buffer: BufReader<R>) { 
    let mut data = String::new(); 
    let _ = buffer.read_line(&mut data); 

    println!("read_buff got {}", data); 
} 

fn main() { 
    read_buff(BufReader::new("Potato!".as_bytes())); 
} 

ten drukuje read_buff got Potato!. W tym przypadku nie ma potrzeby używania kursora.

Aby użyć wbudowanej pamięci String z BufWriter, można użyć metody as_mut_vec. Niestety jest to unsafe i nie znalazłem żadnego innego sposobu. Nie podoba mi się podejście Cursor, ponieważ zużywa ono wektor i nie znalazłem jeszcze sposobu na użycie Cursor razem z BufWriter.

use std::io::BufWriter; 
use std::io::Write; 

pub fn write_something<W: Write>(mut buf: BufWriter<W>) { 
    buf.write("potato".as_bytes()); 
} 

#[cfg(test)] 
mod tests { 
    use super::*; 
    use std::io::{BufWriter}; 

    #[test] 
    fn testing_bufwriter_and_string() { 
     let mut s = String::new(); 

     write_something(unsafe { BufWriter::new(s.as_mut_vec()) }); 

     assert_eq!("potato", &s); 
    } 
}