rust trait 熟识系列:一日一trait之Seek trait

Seek 顾名思义就是用来方便查找的trait,其内部提供了一个在字节流(stream of bytes)移动的游标(cursor),移动游标
可以改变操作字节流的位置。

Seek trait 定义

pub trait Seek {
    fn seek(&mut self, pos: SeekFrom) -> Result<u64>;
    fn rewind(&mut self) -> Result<()> { ... }
	fn stream_len(&mut self) -> Result<u64> { ... }
	fn stream_position(&mut self) -> Result<u64> { ... }
}

seek所需方法参数SeekFrom定义了IO对象查找的可能方法

pub enum SeekFrom {
    Start(u64),
    End(i64),
    Current(i64),
}

例子

标准库的File实现了Seek trait

use std::io;
use std::io::prelude::*;
use std::fs::File;
use std::io::SeekFrom;

fn main() -> io::Result<()> {
    let mut f = File::open("foo.txt")?;

    // move the cursor 42 bytes from the start of the file
    f.seek(SeekFrom::Start(42))?;
    Ok(())
}

Cursor

cursor 实现了Seek trait,用来包装 内存缓存(in-memory buffer)。
说白了,任意实现了AsRef<[u8]>的类型,用Cursor包装后,就实现了Read trait 和Write trait。
比如标准库里就有Cursor<Vec<u8>>Cursor<&[u8]>这样的使用例子。

use std::io::Cursor;
use std::io::{SeekFrom};
use std::io::prelude::*;

fn main() {

    let mut buff = Cursor::new(vec![0;15]);
    buff.seek(SeekFrom::End(-10)).unwrap();
    for i in 0..10 {
    	buff.write(&[i]).unwrap();
    }
    println!("{:?}", buff);
}

上面用Cursor包装了Vec, 并且将cursor移动到了倒数第10个的位置写入,结果为:

Cursor { inner: [0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9], pos: 15 }

在rust中怎么创建一个可读写的内存对象 里提到了用Cursor来实现类似python StringIO库的方法,值得一看

posted @ 2022-01-02 21:47  yihailin  阅读(415)  评论(0编辑  收藏  举报