Rust 有没有办法在不分配另一个字符串的情况下修剪一个字符串

发布于07月08日

我有一个CSV格式的文件，第一列数据表示项目代码，可选以"UNIUNI"个字符或混合字符结尾，通过条形码阅读器加载.我需要删掉最后的"UNI".

我写了这个函数:

fn main() {
    // Ok: from "9846UNIUNI" to "9846"
    println!("{}", read_csv_rilev("9846UNIUNI".to_string()));
    
    // Wrong: from "9846uniuni" to "9846"
    println!("{}", read_csv_rilev("9846uniuni".to_string()));
}

fn read_csv_rilev(code: String) -> String {
    code
        //.to_uppercase() /*Unstable feature in Rust 1.1*/
        .trim_right_matches("UNI")
        .to_string()
}

理想的函数签名如下所示:

fn read_csv_rilev(mut s: &String)

但对String人采取就地行动可能不是一个好主意.事实上，在Rust标准库中，除了String::pop()之外，没有任何东西可以这样做.

推荐答案

有没有办法在不分配另一个的情况下削减一个String？

是的，可以使用truncate删除字符串的尾部:

const TRAILER: &'static str = "UNI";

fn read_csv_rilev(s: &mut String) {
    while s.ends_with(TRAILER) {
        let len = s.len();
        let new_len = len.saturating_sub(TRAILER.len());
        s.truncate(new_len);
    }
}

fn main() {
    let mut code = "Hello WorldUNIUNIUNI".into();
    
    read_csv_rilev(&mut code);
    
    assert_eq!("Hello World", code);
}

您根本不需要弄乱分配的字符串.您可以使用相同的逻辑，对字符串进行连续的子切片.这基本上就是trim_right_matches的工作原理，但有点不那么通用:

const TRAILER: &'static str = "UNI";

fn read_csv_rilev(mut s: &str) -> &str {
    while s.ends_with(TRAILER) {
        let len = s.len();
        let new_len = len.saturating_sub(TRAILER.len());
        s = &s[..new_len];
    }
    s
}

fn main() {
    let code = "Hello WorldUNIUNIUNI";

    let truncated = read_csv_rilev(code);

    assert_eq!("Hello World", truncated);
}

一般来说，我可能会 Select 第二种解决方案.