You've already forked comprehensive-rust
mirror of
https://github.com/google/comprehensive-rust.git
synced 2025-07-05 14:10:29 +02:00
41 lines
975 B
Markdown
41 lines
975 B
Markdown
![]() |
# `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(())
|
||
|
}
|
||
|
```
|