1
0
mirror of https://github.com/google/comprehensive-rust.git synced 2025-07-05 14:10:29 +02:00
Files
comprehensive-rust/src/traits/read-write.md

41 lines
975 B
Markdown
Raw Normal View History

2022-12-21 16:36:30 +01:00
# `Read` and `Write`
Using `Read` and `BufRead`, you can abstract over `u8` sources:
```rust,editable
use std::io::{BufRead, BufReader, Read, Result};
fn count_lines<R: Read>(reader: R) -> usize {
let buf_reader = BufReader::new(reader);
buf_reader.lines().count()
}
fn main() -> Result<()> {
let slice: &[u8] = b"foo\nbar\nbaz\n";
println!("lines in slice: {}", count_lines(slice));
let file = std::fs::File::open(std::env::current_exe()?)?;
println!("lines in file: {}", count_lines(file));
Ok(())
}
```
Similarly, `Write` lets you abstract over `u8` sinks:
```rust,editable
use std::io::{Result, Write};
fn log<W: Write>(writer: &mut W, msg: &str) -> Result<()> {
writer.write_all(msg.as_bytes())?;
writer.write_all("\n".as_bytes())
}
fn main() -> Result<()> {
let mut buffer = Vec::with_capacity(1024);
log(&mut buffer, "Hello")?;
log(&mut buffer, "World")?;
println!("Logged: {:?}", buffer);
Ok(())
}
```