Как я могу создать функцию с переменным количеством аргументов?

Как я могу создать функцию с переменным количеством аргументов в Rust?

Как этот код Java:

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

2 ответа

Решение

В общем, вы не можете - Rust не поддерживает функции с переменным числом, кроме случаев взаимодействия с кодом C, который использует varargs.

В этом случае, поскольку все ваши аргументы имеют одинаковый тип, вы можете принять фрагмент:

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

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

( Детская площадка)

Кроме того, вы можете явно принять необязательные аргументы:

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);
}

( Детская площадка)

Если вам нужно принять много аргументов, необязательных или нет, вы можете реализовать конструктор:

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"));
}

( Детская площадка)

В общем случае, когда ваши типы могут различаться, вы можете использовать макрос:

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

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

Детская площадка

В качестве другого примера, если вы хотите свернуть аргументы, вы можете сделать что-то вроде этого:

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

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

Детская площадка

См . Упрощенную реализацию Rust.vec![...] для другого примера.

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

fn main() {
    variable_func(&[1]);
    variable_func(&[1, 2]);
    variable_func(&["A", "B", "C"]);
}
Другие вопросы по тегам