Jak idiomatycznie/wydajnie potokować dane z Read + Seek to Write?

Question

Chcę pobierać dane z losowych lokalizacji w pliku wejściowym i wysyłać je sekwencyjnie do pliku wyjściowego. Najlepiej bez zbędnych przydziałów.

This is one kind of solution I have figured out:

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

#[test] 
fn read_write() { 
    // let's say this is input file 
    let mut input_file = Cursor::new(b"worldhello"); 
    // and this is output file 
    let mut output_file = Vec::<u8>::new(); 

    assemble(&mut input_file, &mut output_file).unwrap(); 

    assert_eq!(b"helloworld", &output_file[..]); 
} 

// I want to take data from random locations in input file 
// and output them sequentially to output file 
pub fn assemble<I, O>(input: &mut I, output: &mut O) -> Result<(), io::Error> 
    where I: Read + Seek, O: Write 
{ 
    // first seek and output "hello" 
    try!(input.seek(SeekFrom::Start(5))); 
    let mut hello_buf = [0u8; 5]; 
    try!(input.take(5).read(&mut hello_buf)); 
    try!(output.write(&hello_buf)); 

    // then output "world" 
    try!(input.seek(SeekFrom::Start(0))); 
    let mut world_buf = [0u8; 5]; 
    try!(input.take(5).read(&mut world_buf)); 
    try!(output.write(&world_buf)); 

    Ok(()) 
} 

Niech nie martwić się o I/O opóźnieniach tutaj.

Pytania:

1. Czy stabilny Rust mają pewne pomocnika do podjęcia x bajtów z jednego strumienia i przesunąć je do innego strumienia? A może muszę wykonać własne?
2. Jeśli muszę wykonać własne, może jest lepszy sposób?

Answer 1

Szukasz io::copy:

pub fn assemble<I, O>(input: &mut I, output: &mut O) -> Result<(), io::Error> 
    where I: Read + Seek, O: Write 
{ 
    // first seek and output "hello" 
    try!(input.seek(SeekFrom::Start(5))); 
    try!(io::copy(&mut input.take(5), output)); 

    // then output "world" 
    try!(input.seek(SeekFrom::Start(0))); 
    try!(io::copy(&mut input.take(5), output)); 

    Ok(()) 
} 

Jeśli spojrzeć na the implementation of io::copy, można zobaczyć, że jest podobny do kodu. Jednak trzeba uważać, aby obsłużyć więcej przypadków błędach:

1. write robi nie zawsze napisać wszystko poprosić go!
2. Napis "przerwany" zwykle nie kończy się śmiercią.

Używa również większego rozmiaru bufora, ale nadal przypisuje go do siebie.