< 返回版块

lijobs 发表于 2024-11-05 21:39

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

我们在上述的例子中我们创建了一个Watchtrait里面的默认类型为String,trait中定义了一个关联变量Item和两个函数分别是innerinfo。 再创建一个结构体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函数中调用上述方法。

使用场景

  1. 提供默认行为:当你想要提供一个通用的函数或结构体时,可以为不经常变化的参数提供默认值。
  2. 简化函数调用:在大多数情况下,如果有一个常用的类型,你可以让函数调用者不必每次都指定类型参数。
  3. 保持向后兼容:当你更新库或 API 时,添加新的泛型参数并为它们提供默认值可以避免破坏现有代码。

注意事项

默认参数类型智能在结构体,枚举,trait以及type中使用,不能在函数中使用默认类型参数。

评论区

写评论

还没有评论

1 共 0 条评论, 1 页