函数参数的 Rust 功能标志

发布于11月08日

有没有办法将Rust的功能标志支持应用到函数参数？我有一个函数，它接受一串输入，但其中一个输入应该只在功能处于活动状态时才被传递.当然，我可以将该参数设置为可选参数，如下所示:

pub fn my_function(input1: usize, input2: String, input3: Option<Vec<u32>>) -> () {...}

但有没有一种方法可以使用功能标志并定义如下内容:

pub fn my_function(
    input1: usize,
    input2: String,
    #[cfg(feature = "my-feature")]
    input3: Vec<u32>
) -> () {...}

同样，后来又称其为:my_function(0, "", #[cfg(feature = "my-feature")] vec![0])？

推荐答案

Rust的特性标志是对条件编译的简单包装.想象一下，如果该特性始终处于启用状态，那么您将如何编写代码，如果它始终处于禁用状态，您将如何编写代码，并用#[cfg(..])个表达式将两者之间的不同行包装起来.要定义不同的函数，可以定义两个不同的版本，如果它们几乎相同，则让它们调用共享内部块；如果它们应该做不同的事情，则只需分别定义它们:

struct Thing;

impl Thing {
    #[cfg(not(feature = "my-feature"))]
    pub fn my_function(input1: usize, input2: String) {
        Thing::my_function_inner(input1, input2, None);
    }

    #[cfg(feature = "my-feature")]
    pub fn my_function(input1: usize, input2: String, input3: Vec<u32>) {
        Thing::my_function_inner(input1, input2, Some(input3));
    }

    #[inline]
    fn my_function_inner(input1: usize, input2: String, input3: Option<Vec<u32>>) {
        todo!();
    }
}

要调用它，您可以依赖于始终启用/禁用功能标志并只使用相应的版本，或者您也可以包括这两个版本的代码，如下所示:

fn main() {
    #[cfg(not(feature = "my-feature"))]
    Thing::my_function(0, "".to_string());

    #[cfg(feature = "my-feature")]
    Thing::my_function(0, "".to_string(), vec![0]);
}