如何创建具有可变数量参数的函数?

时间:2015-03-09 20:51:03

标签: rust variadic-functions

如何在Rust中创建具有可变数量参数的函数?

喜欢这个Java代码:

void foo(String... args) {
    for (String arg : args) {
        System.out.println(arg);
    }
}

3 个答案:

答案 0 :(得分:36)

通常,你不能 - Rust不支持可变参数函数,除非与使用varargs的C代码进行互操作。

this 的情况下,由于您的所有参数都是相同的类型,因此您可以接受切片:

fn foo(args: &[&str]) {
    for arg in args {
        println!("{}", arg);
    }
}

fn main() {
    foo(&["hello", "world", "I", "am", "arguments"]);
}

Playground

除此之外,您可以明确接受可选参数:

fn foo(name: &str, age: Option<u8>) {
    match age {
        Some(age) => println!("{} is {}.", name, age),
        None      => println!("Who knows how old {} is?", name),
    }
}

fn main() {
    foo("Sally", Some(27));
    foo("Bill", None);
}

Playground

如果您需要接受许多参数,可选或不可选,您可以实现构建器:

struct Arguments<'a> {
    name: &'a str,
    age: Option<u8>,
}

impl<'a> Arguments<'a> {
    fn new(name: &'a str) -> Arguments<'a> {
        Arguments {
            name: name,
            age: None
        }
    }

    fn age(self, age: u8) -> Self {
        Arguments {
            age: Some(age),
            ..self
        }
    }
}

fn foo(arg: Arguments) {
    match arg.age {
        Some(age) => println!("{} is {}.", arg.name, age),
        None      => println!("Who knows how old {} is?", arg.name),
    }
}

fn main() {
    foo(Arguments::new("Sally").age(27));
    foo(Arguments::new("Bill"));
}

Playground

答案 1 :(得分:3)

fn variable_func<T>(_vargs: &[T]) {}

fn main() {
    variable_func(&[1]);
    variable_func(&[1, 2]);
    variable_func(&["A", "B", "C"]);
}

答案 2 :(得分:0)

在通常情况下,您的类型可能会有所不同,您可以使用宏:

macro_rules! print_all {
    ($($args:expr),*) => {{
        $(
            println!("{}", $args);
        )*
    }}
}

fn main() {
    print_all!(1, 2, "Hello");
}

Playground

再举一个例子,如果您想折叠参数,可以执行以下操作:

macro_rules! sum {
    ($($args:expr),*) => {{
        let result = 0;
        $(
            let result = result + $args;
        )*
        result
    }}
}

fn main() {
    assert_eq!(sum!(1, 2, 3), 6);
}

Playground

有关另一个示例,请参见Rust book's simplified implementation of vec![...]