当前位置：首页 > article >正文

揭秘rust中默认参数类型不为人知的秘密，你确定不来了解下吗？

article 2024/12/25 13:51:34

在 Rust 中，你可以为泛型类型参数指定默认类型参数，这样在调用函数或结构体时，如果没有指定具体的类型参数，就会使用默认的类型参数。这提供了一种灵活的方式来使用泛型代码，同时保持了代码的简洁性和可读性。在使用过程中不用考虑泛型带来传入具体参数的问题。

基本语法

默认类型参数的语法是在泛型类型参数后面加上 = 和一个默认值：

在trait中指定默认类型参数

trait Add<Rhs = Self>{
  type Output;
    
  #[must_use]
  fn add(self, rhs: RHS) -> Self::Output;
}

这里Add<RHS = Self>是默认泛型类型参数，表示如果不显示指定泛型类型，就默认泛型类型为Self。当使用泛型类型参数时，可以为泛型指定一个默认的具体类型。如果默认类型就足够的话，这消除了为具体类型实现 trait 的需要。为泛型类型指定默认类型的语法是在声明泛型类型时使用 <PlaceholderType=ConcreteType>。

在结构体中指定默认类型参数

struct Stack<T = i32> {
    items: Vec<T>,
}

用法示例

trait的默认类型参数

pub trait Watch<Inner=String> {
    type Item;
    fn inner(&self) -> Option<Self::Item>;
    fn info(&self) -> Inner;
}

struct A {
    data: i32,
}

impl Watch<i32> for A {
    type Item = i32;
    fn inner(&self) -> Option<Self::Item> {
        Some(self.data)
    }
    fn info(&self) -> i32 {
        println!("A inner is {}", self.data);
        self.data
    }
}

struct B {
    data: String,
}

impl Watch for B {
    type Item = String;
    fn inner(&self) -> Option<Self::Item> {
        Some(self.data.clone())
    }
    fn info(&self) -> String {
        println!("B inner is {}", self.data);
        self.data.clone()
    }
}

fn call_trait_default_param_type(){
  let a = A{data: 10};
  let b = B{data: String::from("B")};
  assert_eq!(10, a.info());
  assert_eq!(Some(String::from("B")), b.inner());
}

我们在上述的例子中我们创建了一个Watch的trait里面的默认类型为String,trait中定义了一个关联变量Item和两个函数分别是inner和info。再创建一个结构体B，里面包含一个字段类型是String，为结构体B实现trait Watch并把关联类型设置为String，分别实现trait中定义的两个函数。在main函数中创建两个结构体实例，分别调用实现了trait的两个方法。

结构体中默认参数类型

struct Stack<T = i32> {
    items: Vec<T>,
}

impl<T> Stack<T> {
    fn new() -> Self {
        Stack { items: Vec::new() }
    }
}

impl Stack<i32> {
    fn new_with_int() -> Self {
        Stack { items: vec![1, 2, 3] }
    }
}

fn main() {
    let stack_default: Stack<i32> = Stack::new(); // 使用默认类型参数 i32
    let int_stack = Stack::new_with_int(); // 明确指定 i32 类型参数
}

上述代码中我们创建一个泛型结构体Stack，其中该泛型结构体的默认参数类型是i32类型。接着我们实现不指定泛型参数类型的函数new和指定默认类型的new_with_int函数。然后在main函数中调用上述方法。