如何在Rust中为String类型正确实现自定义.clean()方法

场景1：原地修改原字符串，支持链式调用 #

如果你需要修改原String本身，且需要支持链式调用，直接原地移除前缀即可：

trait Clean {
    fn clean(&mut self) -> &mut Self;
}

impl Clean for String {
    fn clean(&mut self) -> &mut Self {
        let prefix_removed = self.trim_start_matches('"');
        // 计算要移除的前缀长度
        let trim_length = self.len() - prefix_removed.len();
        // 原地删除前缀
        self.drain(0..trim_length);
        self
    }
}

使用示例：

let mut s = String::from(r#""test content"#);
// 直接修改原字符串
s.clean();
// 支持链式调用
s.clean().push_str(" append");

场景2：不修改原字符串，返回新的String #

如果你不需要改动原变量，直接返回处理后的新String即可：

trait Clean {
    fn clean(self) -> Self;
}

impl Clean for String {
    fn clean(self) -> Self {
        self.trim_start_matches('"').to_string()
    }
}

使用示例：

let s = String::from(r#""test content"#);
let cleaned_s = s.clean();
// 也支持链式调用处理新字符串
let chained = String::from(r#""test""#).clean().replace("test", "new");

原代码问题说明 #

• 原来的trait默认实现没有限定实现者类型，编译器无法确认实现类型存在trim_start_matches方法。直接将方法实现写在针对String的impl块中即可解决，无需在trait层提供默认实现。
• trim_start_matches返回的是原字符串的只读切片&str，不拥有所有权，也不会修改原字符串。根据需求选择原地修改原字符串或者将切片转为新的String返回即可。

扩展：同时支持String和&str调用 #

如果需要让所有字符串类型（包括&str）都支持clean方法，可以添加泛型约束：

trait Clean {
    fn clean(&self) -> String;
}

impl<T: AsRef<str>> Clean for T {
    fn clean(&self) -> String {
        self.as_ref().trim_start_matches('"').to_string()
    }
}

使用示例：

let s1 = "\"test str".clean(); // &str直接调用
let s2 = String::from("\"test string").clean(); // String调用

内容的提问来源于stack exchange，提问作者ilmoi