面试题：Rust中如何实现字符串扩展以添加自定义方法

Question

请描述在Rust中，为字符串类型（`String`或`&str`）实现自定义扩展方法的一般步骤，并给出一个简单示例，比如为字符串添加一个计算单词数量的方法。

Answer 1

1. 实现自定义扩展方法的一般步骤：
   • 定义trait：使用trait关键字定义一个新的trait，在这个trait中声明要添加的方法。
   • 为类型实现trait：使用impl关键字为目标类型（String或&str）实现刚才定义的trait，在实现中编写方法的具体逻辑。
2. 示例：为字符串添加计算单词数量的方法：

// 定义trait
trait StringExtensions {
    fn count_words(&self) -> usize;
}

// 为&str实现trait
impl StringExtensions for &str {
    fn count_words(&self) -> usize {
        self.split_whitespace().count()
    }
}

fn main() {
    let s = "Hello world Rust";
    let word_count = s.count_words();
    println!("单词数量: {}", word_count);
}

在上述代码中：

• 首先定义了StringExtensions trait，其中声明了count_words方法。
• 然后为&str类型实现了这个trait，在count_words方法中通过split_whitespace方法将字符串按空白字符分割，再用count方法统计单词数量。
• 在main函数中创建了一个字符串并调用count_words方法输出单词数量。

如果要为String类型实现，只需将impl StringExtensions for &str改为impl StringExtensions for String，并在实现中通过&self来访问字符串内容，实现逻辑与&str类似。例如：

// 定义trait
trait StringExtensions {
    fn count_words(&self) -> usize;
}

// 为String实现trait
impl StringExtensions for String {
    fn count_words(&self) -> usize {
        self.split_whitespace().count()
    }
}

fn main() {
    let s = String::from("Hello world Rust");
    let word_count = s.count_words();
    println!("单词数量: {}", word_count);
}