string - 如何为 Vec<u8> 实现修剪？

Question

Rust 为字符串提供了一个 trim 方法:str.trim()删除前导和尾随空格。我想要一个对字节字符串执行相同操作的方法。它应该需要 Vec<u8>并删除前导和尾随空格(空格，0x20 和 htab，0x09)。

写一个 trim_left()很简单，你可以使用迭代器 skip_while() : Rust Playground

fn main() {
    let a: &[u8] = b"     fo o ";
    let b: Vec<u8> = a.iter().map(|x| x.clone()).skip_while(|x| x == &0x20 || x == &0x09).collect();
    println!("{:?}", b);
}

但是为了修剪正确的字符，如果在找到空格后列表中没有其他字母，我需要向前看。

Answer 1

这是一个返回切片而不是新的 Vec<u8> 的实现。 , 作为 str::trim()做。它也在 [u8] 上实现，因为这比 Vec<u8> 更通用(您可以很便宜地从向量中获取切片，但是从切片中创建向量的成本更高，因为它涉及堆分配和复制)。

trait SliceExt {
    fn trim(&self) -> &Self;
}

impl SliceExt for [u8] {
    fn trim(&self) -> &[u8] {
        fn is_whitespace(c: &u8) -> bool {
            *c == b'\t' || *c == b' '
        }

        fn is_not_whitespace(c: &u8) -> bool {
            !is_whitespace(c)
        }

        if let Some(first) = self.iter().position(is_not_whitespace) {
            if let Some(last) = self.iter().rposition(is_not_whitespace) {
                &self[first..last + 1]
            } else {
                unreachable!();
            }
        } else {
            &[]
        }
    }
}

fn main() {
    let a = b"     fo o ";
    let b = a.trim();
    println!("{:?}", b);
}

如果你真的需要 Vec<u8>在trim()之后, 你可以调用 into()在切片上将其变成 Vec<u8> .

fn main() {
    let a = b"     fo o ";
    let b: Vec<u8> = a.trim().into();
    println!("{:?}", b);
}