Skip to content

Trait 特征

从多种类型中抽取共性的行为并定义为规范,是一种强大的代码复用方式,也是多态的体现。在面向对象语言中,这种机制通常通过接口(interface)实现;在 Rust 中,通过 Trait 实现。

Trait 有两个核心作用: 约束泛型(规定泛型必须具备哪些能力)和实现多态(通过 Trait 对象在运行时统一处理不同类型)。Rust 是组合优于继承的语言: 一个类型通过实现 Trait 来获得对应的能力(has a),而不是通过继承父类(is a)。

初学者可以简单的将 Trait 理解为能力的集合,强调的是行为规范而非类型关系。

定义 Trait

语法 : trait TraitName { ... }

使用 trait 关键字定义 Trait。Trait 中可以包含:

  • 抽象方法: 只有签名,没有方法体,实现者必须提供实现
  • 默认方法: 有方法体,实现者可以选择覆盖或直接使用默认实现
rust
trait Playable {
    fn play(&self);                // 抽象方法,必须实现
    fn get_duration(&self) -> f32; // 抽象方法,必须实现
    fn pause(&self) {              // 默认方法,可选覆盖
        println!("pause");
    }
    fn stop() -> String;           // 抽象方法,关联函数(不使用 self 形式参数)
}

方法的第一个参数可以是 self&self&mut self,或不含 self (关联函数)。

实现 Trait

语法: impl TraitName for Type { ... }

  • 所有没有默认方法体的方法都必须实现
  • 有默认方法体的方法可以选择覆盖
rust
struct Audio { name: String, duration: f32 }
struct Video { name: String, duration: f32 }

impl Playable for Audio {
    fn play(&self) { println!("listening: {}", self.name); }
    fn get_duration(&self) -> f32 { self.duration }
    // pause() 不实现,使用默认实现
    fn stop() -> String { "audio stopped".to_string() } // 实现没有默认方法体的关联函数
}

impl Playable for Video {
    fn play(&self) { println!("watching: {}", self.name); }
    fn get_duration(&self) -> f32 { self.duration }
    fn pause(&self) { println!("video paused"); }       // 覆盖默认实现
    fn stop() -> String { "video stopped".to_string() } // 实现没有默认方法体的关联函数
}

实现后,类型实例就可以调用该 Trait 中的方法:

rust
fn main() {
    let audio = Audio { name: "song.mp3".to_string(), duration: 3.5 };
    audio.play();                         // 调用 Audio 的 play 实现
    println!("{}", audio.get_duration()); // 调用 Audio 的 get_duration 实现
    audio.pause();                        // 使用 Playable 中的默认实现
    println!("{}", Audio::stop());        // 调用 Audio 的关联函数 stop 实现
}

孤儿规则(Orphan Rule)

实现 Trait 时,类型和 Trait 至少有一个必须是你自己定义的:

  • ✅ 给自己的类型实现标准库的 Trait(如为自定义 struct 实现 Display)
  • ✅ 给标准库的类型实现自己的 Trait(如为 Vec 实现自定义 Trait)
  • ❌ 给标准库的类型实现标准库的 Trait(会破坏生态一致性)

#[derive] 派生 Trait

对于常见 Trait,可以在 structenum 前使用 #[derive()] 让编译器自动实现,无需手动编写:

rust
#[derive(Debug, Clone, PartialEq)]
struct Point {
    x: i32,
    y: i32,
}

Point 现在自动具备了调试输出、克隆、相等比较的能力

常见可派生的 Trait:

Trait功能备注
Debug{:?} 调试输出最常用,几乎总是派生
Clone.clone() 深拷贝堆分配类型需要
Copy赋值时自动位拷贝需同时派生 Clone
PartialEq== 相等比较大多数类型使用
Eq完全相等关系需先派生 PartialEq
PartialOrd< > 大小比较需先派生 PartialEq
Ord完全排序需先派生 EqPartialOrd
Hash哈希计算用于 HashMap 的键
Default提供类型默认值.default() 构造

Trait 作用域

调用 Trait 中的方法时,必须将该 Trait 导入当前作用域,即使某个类型已经实现了该 Trait:

rust
// Vec 已实现 std::io::Write,但不导入就无法调用 Write 中的方法
let mut buf: Vec<u8> = vec![];

buf.write_all(b"hello")?; // ❌ 编译错误: 找不到 write_all 方法

use std::io::Write;       // 必须导入 Trait
buf.write_all(b"hello")?; // ✅ 正确

原因: 多个库可能为同一类型实现了同名方法,不显式导入编译器无法确定调用哪一个。大多数成熟的库(如 tokioserde)会提供 prelude 模块一次性导入常用 Trait:

rust
use tokio::prelude::*;

Trait 作为函数参数

Trait 作为参数时,一般是为了通过 Trait 约束泛型,限制参数必须具备指定能力。有三种等价写法:

1. impl Trait 语法(最简洁,适合单个参数)

rust
// 参数 item 必须实现上文中的 Playable Trait
fn notify(item: &impl Playable) {
    println!("duration: {}", item.get_duration());
}

由于约束了 item 必须实现 Playable Trait,因此 item 能在函数体内直接调用 Playable 的方法 get_duration

2. Trait Bound 语法(可强制多个参数为同一类型)

rust
// 泛型 T 必须实现上文中的 Playable Trait
fn notify<T: Playable>(item1: T, item2: T) {
    println!("{} vs {}", item1.get_duration(), item2.get_duration());
}

由于约束了 T 必须实现 Playable Trait,因此 item1item2 都能在函数体内直接调用 Playable 的方法 get_duration

3. where 子句(多个泛型时可读性更好)

rust
fn notify<T, U>(item: T, msg: U)
where
    // 泛型 T 必须实现上文中的 Playable 和 Debug 两个 Trait
    T: Playable + std::fmt::Debug,
    // 泛型 U 必须实现 Display Trait
    U: std::fmt::Display,
{
    println!("{:?}: {}", item, msg);
}

多个约束用 + 连接,例如 T: Add<Output=T> + Copy + Clone 表示 T 必须同时实现三个 Trait。

Trait 作为函数返回值

语法: fn function_name() -> impl Trait

返回一个实现了 Trait 的类型,而无需指定具体类型; 适合只关心能力而不关心具体类型的场景:

rust
fn create_audio() -> impl Playable {
    Audio { name: "song.mp3".to_string(), duration: 3.5 }
    // 或者
    // Video { name: "movie.mp4".to_string(), duration: 120.0 }
}

Audio 和 Video 是不同的类型,但它们都实现了 Playable Trait,因此函数可以返回任意一个。

重要限制: impl Trait 只能返回一种具体类型,以下代码无法编译:

rust
// ❌ 编译错误: 同一函数不能在不同分支返回不同类型
fn create_media(is_video: bool) -> impl Playable {
    if is_video {
        Video { name: "movie.mp4".to_string(), duration: 120.0 } 
    } else {
        Audio { name: "song.mp3".to_string(), duration: 3.5 } 
    }
}

原因: impl Trait 是编译期泛型的语法糖,返回类型在编译期必须确定为某一具体类型。需要返回不同类型时,使用 Trait 对象

Trait 对象(dyn Trait)

Trait 本身不能作为数据类型使用,但 Trait 对象(Trait Object)可以。Trait 对象使用 dyn 关键字(dynamic)标记,几乎总是以引用方式使用: &dyn TraitBox<dyn Trait>Rc<dyn Trait> 等。

rust
// x 是一个 Trait 对象,要求是实现了 Playable 的某个类型实例
let x: &dyn Playable = &Audio { name: "song.mp3".to_string(), duration: 3.5 };
x.play(); // 运行时动态分发

// y 是一个具体类型实例,编译时静态分发
let y: Audio = Audio { name: "song.mp3".to_string(), duration: 3.5 };
y.play(); // 编译时静态分发

如果说类型是编译器用来区分不同数据的标签,那么 Trait 对象就是编译器用来区分不同能力的标签。类型关注的是是什么,Trait 对象关注的是能做什么

内存布局: 胖指针

Trait 对象的引用是一个胖指针,由两个指针组成,固定 16 字节(64 位系统):

&dyn Playable(栈上,16 字节)
┌──────────────────┬──────────────────┐
│  ptr(数据指针)  │  vptr(虚表指针) │
└────────┬─────────┴────────┬─────────┘
         │                  │
         ▼ 指向具体实例      ▼ 指向虚表(只读数据区)
    Audio { ... }     ┌─────────────────────┐
                      │  drop 函数指针       │
                      │  play 函数指针       │
                      │  get_duration 指针   │
                      │  pause 函数指针      │
                      └─────────────────────┘
  • 数据指针: 指向具体类型实例的内存(如 Audio 的数据)
  • 虚表指针(vtable): 指向一张函数指针表,记录该类型对 Trait 中每个方法的具体实现

调用方法时,程序通过 vtable 查找对应的函数指针再调用,这一过程在运行时完成,称为动态分发(dynamic dispatch)。

使用 Trait 对象的场景

Trait 对象的核心价值: 允许在同一个集合中存放不同类型的值,只要它们都实现了同一个 Trait:

rust
fn main() {
    // 这个 Vec 要求存放实现了 Playable Trait 的类型实例,但具体类型可以不同
    // Vec 中同时存放 Audio 和 Video,因为两者都实现了 Playable, 但类型不同
    let playlist: Vec<Box<dyn Playable>> = vec![
        Box::new(Audio { name: "song.mp3".to_string(), duration: 3.5 }),
        Box::new(Video { name: "movie.mp4".to_string(), duration: 120.0 }),
    ];

    for item in &playlist {
        item.play(); // 运行时动态分发,分别调用各自的 play 实现
    }
}

也可以用于需要返回不同类型的函数:

rust
// 返回一个实现了 Playable Trait 的 Box<dyn Playable>,具体类型在运行时确定
fn create_media(is_video: bool) -> Box<dyn Playable> {
    if is_video {
        Box::new(Video { name: "movie.mp4".to_string(), duration: 120.0 })
    } else {
        Box::new(Audio { name: "song.mp3".to_string(), duration: 3.5 })
    }
}

Trait 对象安全

只有对象安全(object-safe)的 Trait 才能创建 Trait 对象。Trait 方法必须满足:

  • 返回值类型不是 Self,否则无法在运行时确定具体类型也就无法在编译期确定大小。
  • 方法没有泛型类型参数,否则无法生成统一的虚函数表。

❌ 错误示例: 返回值类型不是 Self

rust
trait Clone {
    // 错误:返回了 Self
    fn clone(&self) -> Self;
}

// 尝试编译会报错:the trait `Clone` cannot be made into an object
fn make_object(obj: &dyn Clone) {}

原因:Self 代表实现该 Trait 的具体类型。当使用 dyn Trait 时,具体类型已经被抹去了。如果一个方法返回 Self,调用者在编译时根本不知道该分配多少内存来接收这个返回值(因为不同实现类的大小不同)。

❌ 错误示例: 方法带有泛型参数

rust
trait Container {
    // 错误:方法带有泛型参数 T
    fn insert<T>(&self, item: T);
}

// 编译报错:method `insert` has generic type parameters
fn process(c: &dyn Container) {}

原因:Rust 的泛型采用的是单态化(Monomorphization)机制。这意味着编译器会为每一个调用该泛型方法的具体类型生成一份独立的函数代码。如果 Trait 方法带了泛型,理论上它可以被无数种不同的类型调用,这就需要生成无数个函数指针。虚表(vtable)的大小是固定的,根本装不下无限个函数指针。

不安全 Trait 对象的共存

如果你设计了一个 Trait,里面既有面向对象的方法,又有由于上述原因不满足对象安全的方法,你可以通过给不安全的方法加上 where Self: Sized 约束,将其从虚表中“剔除”。这样,这个 Trait 依然可以转为对象。

rust
trait MyTrait {
    // 满足对象安全,会进入虚表
    fn safe_method(&self);

    // 加上 where Self: Sized,表示只有大小固定的具体类型能调用
    // dyn Trait 是 unsized 的,所以会直接忽略这个方法,从而保证 Trait 整体对象安全
    fn generic_method<T>(&self, arg: T) where Self: Sized;
}

// 编译成功!
fn handle_object(obj: &dyn MyTrait) {
    obj.safe_method(); // 可以调用
    // obj.generic_method(42); // 编译报错:dyn MyTrait 无法调用该方法
}

&BoxRc

这是初学者最容易困惑的地方—— &dyn TraitBox<dyn Trait>Rc<dyn Trait> 为什么不能直接写 dyn Trait

  • 大小不固定: Rust 在编译时需要知道每个变量的大小。Dog 类型可能占 8 字节,Cat 类型可能占 800 字节。如果只说"给我个会叫的动物",Rust 不知道该分配多少内存。
  • 指针大小固定: 无论指向什么,指针在 64 位系统上始终是 8 字节。
  • 胖指针(Fat Pointer): &dyn Trait 实际存储了两个指针:
    • 一个指向数据本身
    • 一个指向虚函数表(vtable),记录了该类型所有 trait 方法的地址。
&dyn Trait 内存布局:
┌─────────────────┬─────────────────┐
│  data pointer   │  vtable pointer │
│  (指向实际数据)  │  (指向方法表)   │
└─────────────────┴─────────────────┘

栈引用(只读借用):

rust
let x: &dyn Playable = &Audio { ... };

堆分配(拥有所有权):

rust
// Box 本身是一个指针,大小固定(8字节),指向堆内存
let x: Box<dyn Playable> = Box::new(Audio { ... });

引用计数(多端共享):

rust
// Rc/Arc 本身也是指针,大小固定
use std::rc::Rc;
let x: Rc<dyn Playable> = Rc::new(Audio { ... });

泛型 与 Trait 对象对比

对比项泛型(T/T: Trait)Trait 对象(dyn Trait)
分发时机编译期静态分发运行时动态分发
性能更高(单态化,无额外开销)稍低(vtable 查找)
同一集合只能放同一类型的元素可以放不同类型的元素
函数返回不同类型❌ 不可以✅ 可以
二进制体积较大(单态化膨胀)较小
使用限制无额外限制Trait 必须满足对象安全规则

T: Trait 表示泛型 T 必须实现 Trait,既要类型参数又要 Trait 约束;

dyn Trait 只关心能力,不关心具体类型

条件实现

条件实现(Blanket Implementation)通过带 Trait 约束的 impl 块,可以有条件地只为实现了特定 Trait 的类型实现方法或另一个 Trait:

rust
pub trait GetName {
    fn get_name(&self) -> &String;
}

pub trait PrintName {
    fn print_name(&self);
}

// 为所有实现了 GetName Trait 的类型自动实现 PrintName Trait
impl<T: GetName> PrintName for T {
    fn print_name(&self) {
        println!("name = {}", self.get_name());
    }
}

struct Student { name: String }

impl GetName for Student {
    fn get_name(&self) -> &String { &self.name }
}

fn main() {
    let s = Student { name: "Alice".to_string() };
    s.print_name(); // Student 实现了 GetName,因此自动获得 PrintName 的方法
}

标准库大量使用这种模式,例如为所有实现了 Display 的类型自动实现 ToString

使用原则

  • impl Struct: 私有逻辑、内部实现,不对外暴露
  • impl Trait: 对外接口、公开能力、通用算法
  • 当发现多个结构体有相同操作,或第三方函数需要某种能力时,将共性逻辑提取为 Trait

基于 MIT 协议发布