Rust 字符串与 &str 迭代器

Question

我正在尝试编写一个接受令牌列表的函数。但是我在使它足够通用以处理两个非常相似的调用时遇到了问题：

let s = String::from("-abc -d --echo");

parse( s.split_ascii_whitespace() );
parse( std::env::args() );

• String::split_ascii_whitespace() 返回实现 Iterator<Item=&'a str> 的 std::str:SplitAsciiWhitespace。
• std::env::args() 返回实现 Iterator<Item=String> 的 std::env::Args。

我有没有办法为parse 编写一个函数签名来接受这两种方法？

我现在的解决方案需要复制函数体：

fn main() {
    let s = String::from("-abc -d --echo");

    parse_args( s.split_ascii_whitespace() );
    parse_env( std::env::args() );
}

fn parse_env<I: Iterator<Item=String>>(mut it: I) {
    loop {
        match it.next() {
            None => return,
            Some(s) => println!("{}",s),
        }
    }
}

fn parse_args<'a, I: Iterator<Item=&'a str>>(mut it: I) {
    loop {
        match it.next() {
            None => return,
            Some(s) => println!("{}",s),
        }
    }
}

如果不可能，那么一些关于如何使用特征以便函数可以使用相同名称的建议会很好。

Answer 1

您可以要求项目类型为AsRef<str>，这将包括&str 和String：

fn parse<I>(mut it: I)
where
    I: Iterator,
    I::Item: AsRef<str>,
{
    loop {
        match it.next() {
            None => return,
            Some(s) => println!("{}", s.as_ref()),
        }
    }
}

Answer 2

根据您的用例，您可以尝试：

fn main() {
    let s = String::from("-abc -d --echo");

    parse( s.split_ascii_whitespace() );
    parse( std::env::args() );
}

fn parse<T: std::borrow::Borrow<str>, I: Iterator<Item=T>>(mut it: I) {
    loop {
        match it.next() {
            None => return,
            Some(s) => println!("{}",s.borrow()),
        }
    }
}

我使用Borrow 作为获得&str 的一种方式，但您的具体用例可能由其他可能自定义的特征提供服务。