参考:https://kaisery.github.io/trpl-zh-cn/ch17-00-oop.html
当 Rust 程序变得更大时,有哪些符合语言习惯的问题建模方法和结构化解决方案,以及 Rust 的风格是如何与面向对象编程(Object Oriented Programming)中那些你所熟悉的概念相联系的。
面向对象编程(Object-Oriented Programming,OOP)是一种模式化编程方式。
对象(Object)来源于 20 世纪 60 年代的 Simula 编程语言。这些对象影响了 Alan Kay 的编程架构中对象之间的消息传递。他在 1967 年创造了 面向对象编程 这个术语来描述这种架构。
关于 OOP 是什么有很多相互矛盾的定义;在一些定义下,Rust 是面向对象的;在其他定义下,Rust 不是。在本章节中,我们会探索一些被普遍认为是面向对象的特性和这些特性是如何体现在 Rust 语言习惯中的。接着会展示如何在 Rust 中实现面向对象设计模式,并讨论这么做与利用 Rust 自身的一些优势实现的方案相比有什么取舍。

Rust 中的面向对象

面向对象编程语言所共享的一些特性往往是对象、封装和继承。让我们看一下这每一个概念的含义以及 Rust 是否支持他们。

对象包含数据和行为

Object-oriented programs are made up of objects. An object packages both data and the procedures that operate on that data. The procedures are typically called methods or operations . 面向对象的程序是由对象组成的。一个 对象 包含数据和操作这些数据的过程。这些过程通常被称为 方法操作 。 —— Design Patterns: Elements of Reusable Object-Oriented Software (被俗称为 The Gang of Four

在这个定义下,Rust 是面向对象的:结构体和枚举包含数据而 impl 块提供了在结构体和枚举之上的方法。虽然带有方法的结构体和枚举并不被 称为 对象,但是他们提供了与对象相同的功能,参考 The Gang of Four 中对象的定义。

封装隐藏了实现细节

封装encapsulation ):

  • 对象的实现细节不能被使用对象的代码获取到
  • 唯一与对象交互的方式是通过对象提供的公有 API
  • 使用对象的代码无法深入到对象内部并直接改变数据或者行为
  • 封装使得改变和重构对象的内部时无需改变使用对象的代码

Rust 满足这个要求:在代码中不同的部分使用 pub 与否可以封装其实现细节。
Example:封装 AveragedCollection 的实现细节,将来可以轻松改变类似数据结构这些方面的内容。
例如,可以使用 HashSet<i32> 代替 Vec<i32> 作为 list 字段的类型。只要 addremoveaverage 公有函数的签名保持不变,使用 AveragedCollection 的代码就无需改变。
相反如果使得 list 为公有,就未必都会如此了: HashSet<i32>Vec<i32> 使用不同的方法增加或移除项,所以如果要想直接修改 list 的话,外部的代码可能不得不做出修改。

  1. // src/lib.rs
  2. #[derive(Debug)]
  3. pub struct AveragedCollection {
  4. // 传入 Vec,自动计算其均值
  5. // 删除、添加元素之后自动更新均值
  6. // 如果把 list 字段公开,则用户直接可以修改值
  7. // 然而 average 的值不会被自动更新
  8. // 所以这两个值都是私有,需要访问时使用同名关联函数
  9. list: Vec<i32>,
  10. average: f64,
  11. }
  12. impl AveragedCollection {
  13. pub fn new(v: Vec<i32>) -> Self {
  14. let mut s = Self { list: v, average: 0. };
  15. s.update_average();
  16. s
  17. }
  18. pub fn add(&mut self, value: i32) {
  19. self.list.push(value);
  20. self.update_average();
  21. }
  22. pub fn remove(&mut self) -> Option<i32> {
  23. let result = self.list.pop();
  24. match result {
  25. Some(value) => {
  26. self.update_average();
  27. Some(value)
  28. }
  29. None => None,
  30. }
  31. }
  32. // 利用不可变引用来提供结构体内部的值,无法通过外部代码修改
  33. pub fn list(&self) -> &Vec<i32> { &self.list }
  34. pub fn average(&self) -> f64 { self.average }
  35. fn update_average(&mut self) {
  36. let total: i32 = self.list.iter().sum();
  37. self.average = total as f64 / self.list.len() as f64;
  38. }
  39. }

使用:

  1. // src/main.rs
  2. use crate_full_name::oop::AveragedCollection;
  3. fn main() {
  4. let mut ac = AveragedCollection::new(vec![1, 3]);
  5. println!("{:?}", ac);
  6. println!("average: {}", ac.average());
  7. ac.add(5);
  8. println!("{:?}", ac);
  9. println!("average: {}", AveragedCollection::average(&ac));
  10. ac.remove();
  11. // 不可通过外部外部代码修改结构体的字段值
  12. // ac.list.pop();
  13. println!("{:?}", ac);
  14. println!("list: {:?}", ac.list());
  15. // list 和 average 都是私有字段,无法直接访问
  16. // println!("list: {:?}", ac.list);
  17. // println!("average: {}", ac.average);
  18. }

打印结果:

  1. AveragedCollection { list: [1, 3], average: 2.0 }
  2. average: 2
  3. AveragedCollection { list: [1, 3, 5], average: 3.0 }
  4. average: 3
  5. AveragedCollection { list: [1, 3], average: 2.0 }
  6. list: [1, 3]

继承:作为类型系统与代码共享

继承Inheritance )是一个很多编程语言都提供的机制,一个对象可以定义为继承另一个对象的定义,这使其可以获得父对象的数据和行为,而无需重新定义。
如果一个语言必须有继承才能被称为面向对象语言的话,那么 Rust 就不是面向对象的。Rust 无法定义一个结构体继承父结构体的成员和方法。
选择继承有两个主要的原因:

  1. 为了重用代码:一旦为一个类型实现了特定行为,继承可以对一个不同的类型重用这个实现。相反 Rust 代码可以使用默认 trait 方法实现来进行共享,也可以选择覆盖默认实现,这类似于子类覆盖从父类继承的方法实现。
  2. 与类型系统有关:表现为子类型可以用于父类型被使用的地方。这也被称为多态polymorphism),这意味着如果多种对象共享特定的属性,则可以相互替代使用。Rust 通过泛型来对不同的可能类型进行抽象,并通过 trait bounds 对这些类型所必须提供的内容施加约束。

多态(Polymorphism)是一个有关可以用于多种类型的代码的更广泛的概念,而不限于继承。
Rust 选择使用 trait 对象而不是继承的原因:继承时常带有共享多于所需的代码的风险。子类不应总是共享其父类的所有特征,但是继承却始终如此。如此会使程序设计更为不灵活,并引入无意义的子类方法调用,或由于方法实际并不适用于子类而造成错误的可能性。某些语言还只允许子类继承一个父类,进一步限制了程序设计的灵活性。

trait 对象:为使用不同类型的值而设计

Rust 刻意不将结构体与枚举称为 “对象”,以便与其他语言中的对象相区别。
在结构体或枚举中,结构体字段中的数据和 impl 块中的行为是分开的,不同于其他语言中将数据和行为组合进一个称为对象的概念中。
trait 对象 (object) 将数据和行为两者相结合,从这种意义上说 其更类似其他语言中的对象。不过 trait 对象不同于传统的对象,因为不能向 trait 对象增加数据。trait 对象并不像其他语言中的对象那么通用:其(trait 对象)具体的作用是允许对通用行为进行抽象。
语法为:在具体的 trait 前加上 指针类型+dyn关键字。指针包括 &、和智能指针。例子 &dyn OneTraitBox<dyn OneTrait>

几种使用多类型的方法

  1. 利用枚举类型:在固定的几种不同类型中选择

优点:编译时就能确定类型需要的大小,对于枚举体,我们知道其空间上限,运行效率高
缺点:需要列举出所有可能需要的类型组合

  1. enum SpreadsheetCell {
  2. Int(i32),
  3. Float(f64),
  4. Text(String),
  5. }
  6. let row = vec![ // Vec 的所有元素都是 SpreadsheetCell 类型
  7. SpreadsheetCell::Int(3),
  8. SpreadsheetCell::Text(String::from("blue")),
  9. SpreadsheetCell::Float(10.12),
  10. ];

vector 只能存储同种类型元素的局限,可以定义 SpreadsheetCell 枚举来储存整型,浮点型和文本成员。这意味着可以在每个单元中储存不同类型的数据,并仍能拥有一个代表一排单元的 vector。

  1. 泛型+trait bound:在 trait bound 内使用一种具体类型

优点:其定义会在编译时采用具体类型进行单态化,适合只需要同质(相同类型)集合;比枚举类型更加灵活。
缺点:一次只能替代一个具体类型

  1. pub trait Draw {
  2. fn draw(&self);
  3. }
  4. pub struct Screen<T: Draw> {
  5. // Vec 里面是泛型 T,其元素是带有 Draw trait 的任何一种类型
  6. // 由于泛型编译时的单态化,一旦这个类型固定下来,所有元素只能是这种具体类型
  7. pub components: Vec<T>,
  8. }
  9. impl<T> Screen<T>
  10. where T: Draw {
  11. pub fn run(&self) {
  12. for component in self.components.iter() {
  13. component.draw();
  14. }
  15. }
  16. }
  1. trait 对象 (object):允许在运行时替代多种具体类型

trait 对象指向一个实现了我们指定 trait 的类型的实例,以及一个用于在运行时查找该类型的trait方法的表。我们通过指定某种指针来创建 trait 对象,例如&引用或Box<T>智能指针,还有dynkeyword, 以及指定相关的 trait。
优点:我们可以使用 trait 对象代替泛型或具体类型,无需在编译时就知晓所有可能的类型。
缺点:运行时性能会受到影响

  1. // gui crate: src/lib.rs
  2. pub trait Draw {
  3. fn draw(&self);
  4. }
  5. pub struct Screen {
  6. // 可以存放实现 Draw trait 的任何类型
  7. // Vec 的所有元素是 Box 类型,Box 是一个智能指针,把内部的数据放在堆上,Box 里面可以是任何类型
  8. // 因此 Box<Button>、Box<TextField> 等多种类型可以同时放入 Vec
  9. // 需要注意的是,trait 对象是编译时不能完全确定下来的类型,所以需要 dyn 关键字来表明
  10. pub components: Vec<Box<dyn Draw>>,
  11. }
  12. impl Screen {
  13. pub fn run(&self) {
  14. for component in self.components.iter() {
  15. component.draw();
  16. }
  17. }
  18. }
  19. pub struct Button {
  20. pub width: u32,
  21. pub height: u32,
  22. pub label: String,
  23. }
  24. impl Draw for Button {
  25. fn draw(&self) {
  26. // 实际绘制按钮的代码
  27. }
  28. }
  29. pub struct TextField {
  30. pub width: u32,
  31. pub height: u32,
  32. pub content: String,
  33. }
  34. impl Draw for TextField {
  35. fn draw(&self) {
  36. // 实际绘制按钮的代码
  37. }
  38. }

如果 crate 的使用者决定实现一个包含 widthheightoptions 字段的结构体 SelectBox,并且也为其实现了 Draw trait:

  1. // crate user: src/lib.rs
  2. use gui::Draw;
  3. struct SelectBox {
  4. width: u32,
  5. height: u32,
  6. options: Vec<String>,
  7. }
  8. impl Draw for SelectBox {
  9. fn draw(&self) {
  10. // code to actually draw a select box
  11. }
  12. }

使用者就可以他们的 main 函数中创建一个 Screen 实例。至此可以通过将 SelectBoxButton 放入 Box<T> 转变为 trait 对象来增加组件。接着可以调用 Screenrun 方法,它会调用每个组件的 draw 方法。

  1. // crate user: src/main.rs
  2. use gui::{Screen, Button};
  3. fn main() {
  4. let screen = Screen {
  5. components: vec![
  6. Box::new(SelectBox {
  7. width: 75,
  8. height: 10,
  9. options: vec![
  10. String::from("Yes"),
  11. String::from("Maybe"),
  12. String::from("No")
  13. ],
  14. }),
  15. Box::new(Button {
  16. width: 50,
  17. height: 10,
  18. label: String::from("OK"),
  19. }),
  20. ],
  21. };
  22. screen.run();
  23. }
  • 当编写 gui crate 的时候,我们不知道何人会在何时增加 SelectBox 类型,不过 Screen 的实现能够操作并绘制这个新类型,因为 SelectBox 实现了 Draw trait,这意味着它实现了 draw 方法。这个概念 —— 只关心值所反映的信息而不是其具体类型 —— 类似于动态类型语言中称为 鸭子类型duck typing )的概念:如果它走起来像一只鸭子,叫起来像一只鸭子,那么它就是一只鸭子!
  • 实现中(这里是使用 gui crate 的时候),run 并不需要知道各个组件的具体类型是什么。它并不检查组件是 Button 或者 SelectBox 的实例。通过指定 Box<dyn Draw> 作为 components vector 中值的类型,我们就定义了 Screen 为需要可以在其上调用 draw 方法的值。

    static & dynamic dispatch

    静态/动态分派拓展阅读:https://www.jianshu.com/p/e0659093eaac
    分派 (dispatch):处理方法调用的过程。一个方法会在运行时被调用,是因为编译器有一个计算机制,用来选择正确的方法,然后通过传递参数来唤起它。这个机制通常被称为 dispatch。
    静态分派 (static dispatch):在编译期确定需要调用的方法,在运行期进行调用。特点如下:
  1. 标准做法:在编译器确定使用 static dispatch 后,会在生成的可执行文件内 直接指定包含了方法实现内存地址的指针;运行时 直接通过指针调用特定的方法。
  2. 优化做法:内联 (inline):编译期从指定被调用的方法指针,改为将方法的实现平铺在调用方的可执行文件内。
    • inline 可以人为声明,也可以通过编译器优化来实现
    • 和宏展开类似,区别在于:内联发生在编译期,并且不会改变源文件;宏展开是在编译前就完成的,会改变源码本身。之后再对此进行编译
    • 内联可以用于消减方法被调用的时间,非常适用于会被频繁调用的方法。如果方法本身很小的话,可以降低内存上的消耗。内联还为进一步的编译优化提供了基础。
    • 有些内联可以通过很小的内存消耗来提升运行速度。但是无节制的内联,也可能会降低速度,因为内联的代码需要大量的CPU 缓存,并且也会消耗内存空间

动态分派 (dynamic dispatch):在运行期选择调用方法的实现的流程,目的是为了支持在编译期无法确定最终最合适的实现的操作。

  • 比静态分派带来更多的性能损耗。优点在于编程时灵活。
  • single dispatch:通过对象类型去选择调用方法的模式。如 dividend.divide(divisor) 调用 divide 方法。
  • multiple dispatch:根据方法名结合方法的参数,一起来判断需要执行的方法。
  • 实现方式:虚函数表

当对泛型使用 trait bound 时编译器所进行单态化处理:编译器为每一个被泛型类型参数代替的具体类型生成了非泛型的函数和方法实现。单态化所产生的代码进行 静态分发static dispatch )。静态分发发生于编译器在编译时就知晓调用了什么方法的时候。
动态分发dynamic dispatch ):编译器在编译时无法知晓调用了什么方法。在动态分发的情况下,编译器会生成在运行时确定调用了什么方法的代码。

  • 当使用 trait 对象时,Rust 必须使用动态分发。编译器无法知晓所有可能用于 trait 对象代码的类型,所以它也不知道应该调用哪个类型的哪个方法实现。因此 Rust 在运行时使用 trait 对象中的指针来知晓需要调用哪个方法。
  • 动态分发阻止编译器有选择的内联方法代码,这会相应的禁用一些优化。

    trait 对象要求对象安全

    只有 对象安全object safe )的 trait 才可以组成 trait 对象。如果一个 trait 中所有的方法有如下属性时,则该 trait 是对象安全的:

  • 返回值类型不为 Self

  • 方法没有任何泛型类型参数

Self 关键字是我们要实现 trait 或方法的类型的别名。对象安全对于 trait 对象是必须的,因为一旦有了 trait 对象,就不再知晓实现该 trait 的具体类型是什么了。如果 trait 方法返回具体的 Self 类型,但是 trait 对象忘记了其真正的类型,那么方法不可能使用已经忘却的原始具体类型。
同理对于泛型类型参数来说,当使用 trait 时其会放入具体的类型参数:此具体类型变成了实现该 trait 的类型的一部分。当使用 trait 对象时其具体类型被抹去了,故无从得知放入泛型参数类型的类型是什么。
trait object 不安全的例子:标准库中的 Clone trait。

  1. pub trait Clone {
  2. fn clone(&self) -> Self;
  3. }

如果违反 object safe 规则:

  1. pub struct Screen {
  2. pub components: Vec<Box<dyn Clone>>,
  3. }

编译器会不通过:

  1. the trait `std::clone::Clone` cannot be made into an object

oo design pattern

面向对象设计模式:状态模式state pattern

  • 关键在于一个值有某些内部状态,体现为一系列的 状态对象 ,同时值的行为随着其内部状态而改变
  • 在 Rust 中使用结构体和 trait 让状态对象共享功能
  • 每一个状态对象负责其自身的行为,以及该状态何时应当转移至另一个状态
  • 状态模式的优点:灵活性、拓展性。
    • 当程序的业务需求改变时,无需改变值持有状态或者使用值的代码。我们只需更新某个状态对象中的代码来改变其规则,或者是增加更多的状态对象。
    • 根据所处的状态来封装不同类型的行为,从而实现各个状态独立的功能
    • 状态共有的功能则通过 trait 对象实现

例子:发布博文的工作流

  1. 博文从空白的草案开始。
  2. 一旦草案完成,请求审核博文。
  3. 一旦博文过审,它将被发表。
  4. 只有被发表的博文的内容会被打印,这样就不会意外打印出没有被审核的博文的文本。

任何其他对博文的修改尝试都是没有作用的。例如,如果尝试在请求审核之前通过一个草案博文,博文应该保持未发布的状态。
这里包括 三种状态:草稿、待审核、发布成功。每种状态可能会有其单独的信息,因而把它们定义成结构体;定义这些状态对象联系的纽带 State trait 和共有的方法。

状态对象 Draft PendingReview Published
状态转移 Post::new() Post::request_review() Post::approve()
  1. // crate name: blog
  2. // src/main.rs
  3. use blog::Post;
  4. fn main() {
  5. let mut post = Post::new();
  6. post.add_text("I ate a salad for lunch today");
  7. assert_eq!("", post.content());
  8. post.request_review();
  9. assert_eq!("", post.content());
  10. post.approve();
  11. assert_eq!("I ate a salad for lunch today", post.content());
  12. }
  1. // crate name: blog
  2. // src/lib.rs
  3. #![allow(unused)]
  4. pub struct Post {
  5. // state 和 content 都是私有的
  6. state: Option<Box<dyn State>>,
  7. // 调用同名方法 .content() 来获取不同状态下的 content
  8. // 具体来说,Draft、PendingReview 状态时外部调用 .content() 为 空串
  9. // Approve 状态下 .content() 才会返回 Post.content
  10. content: String,
  11. }
  12. impl Post {
  13. // 无论何时新建一个 Post 实例,它都会从 Draft 状态开始
  14. // 因为 Post 的 state 字段是私有的,也就无法创建任何其他状态的 Post 了
  15. pub fn new() -> Post {
  16. Post { state: Some(Box::new(Draft {})), content: String::new() }
  17. }
  18. // add_text 方法完全不与 state 状态交互。因为它的行为并不依赖博文所处的状态
  19. pub fn add_text(&mut self, text: &str) { self.content.push_str(text); }
  20. pub fn content(&self) -> &str {
  21. // as_ref 获取 Option 中值的引用而不是其所有权,它返回 Option<&Box<State>>
  22. // 如果不调用 as_ref,会得到错误:不能将 state 移动出借用的 &self 函数参数
  23. // 调用 unwrap 方法,这里我们知道它永远也不会 panic,因为 Post
  24. // 的所有方法都确保在他们返回时 state 会有一个 Some 值
  25. self.state.as_ref().unwrap().content(self)
  26. }
  27. // Post 的 request_review 方法无论 state 是何值都是一样的
  28. pub fn request_review(&mut self) {
  29. // Option.take 方法会把 Option 内的值取出,把默认值 None 替换成原变量值
  30. if let Some(s) = self.state.take() {
  31. // 这使得我们将 state 值移动出 Post 而不是借用它
  32. // 接着将博文的 state 值设置为这个操作的结果:Some(PendingReview)
  33. self.state = Some(s.request_review())
  34. }
  35. }
  36. pub fn approve(&mut self) {
  37. if let Some(s) = self.state.take() {
  38. self.state = Some(s.approve())
  39. }
  40. }
  41. }
  42. // State trait 定义了所有不同状态的博文所共享的行为
  43. // 同时 Draft、PendingReview 和 Published 状态都会实现 State 状态
  44. trait State {
  45. // self: Box<Self> 这个语法意味着这个方法调用只对这个类型的 Box 有效
  46. // 这个语法获取了 Box<Self> 的所有权,使老状态无效化
  47. // 以便 Post 的状态值可以将自身转换为新状态
  48. // 待审核状态
  49. fn request_review(self: Box<Self>) -> Box<dyn State>;
  50. // 发布状态
  51. fn approve(self: Box<Self>) -> Box<dyn State>;
  52. // content 方法的默认实现来返回一个空字符串 slice
  53. // 这意味着无需为 Draft 和 PendingReview 结构体实现 content 了
  54. // 返回的引用的生命周期与 post 参数相关
  55. fn content<'a>(&self, post: &'a Post) -> &'a str { "" }
  56. }
  57. struct Draft {}
  58. impl State for Draft {
  59. fn request_review(self: Box<Self>) -> Box<dyn State> { Box::new(PendingReview {}) }
  60. fn approve(self: Box<Self>) -> Box<dyn State> { Box::new(Published {}) }
  61. }
  62. struct PendingReview {}
  63. impl State for PendingReview {
  64. fn request_review(self: Box<Self>) -> Box<dyn State> { self }
  65. fn approve(self: Box<Self>) -> Box<dyn State> { Box::new(Published {}) }
  66. }
  67. struct Published {}
  68. impl State for Published {
  69. fn request_review(self: Box<Self>) -> Box<dyn State> { self }
  70. fn approve(self: Box<Self>) -> Box<dyn State> { self }
  71. fn content<'a>(&self, post: &'a Post) -> &'a str { &post.content }
  72. }

17 OOP 面向对象编程的特性 - 图1
通过面向对象的状态模式,我们能根据博文所处的状态来封装不同类型的行为:Post 的方法并不知道这些不同类型的行为。通过这种组织代码的方式,要找到所有已发布博文的不同行为只需查看一处代码:PublishedStatetrait 的实现。
如果要创建一个不使用状态模式的替代实现,则可能会在 Post 的方法中,或者甚至于在 main 代码中用到 match 语句,来检查博文状态并在这里改变其行为。这意味着需要查看很多位置来理解处于发布状态的博文的所有逻辑!这在增加更多状态时会变得更糟:每一个 match 语句都会需要另一个分支。
对于状态模式来说,Post 的方法和使用 Post 的位置无需 match 语句,同时增加新状态只涉及到增加一个新 struct 和为其实现 trait 的方法。
这个实现易于扩展增加更多功能。为了体会使用此模式维护代码的简洁性,请尝试如下一些建议:

  • 增加 reject 方法将博文的状态从 PendingReview 变回 Draft
  • 在将状态变为 Published 之前需要两次 approve 调用
  • 只允许博文处于 Draft 状态时增加文本内容。提示:让状态对象负责内容可能发生什么改变,但不负责修改 Post

当前例子使用状态模式的缺点:

  1. 是因为状态实现了状态之间的转换,一些状态会相互联系。如果在 PendingReviewPublished 之间增加另一个状态,比如 Scheduled,则不得不修改 PendingReview 中的代码来转移到 Scheduled。如果 PendingReview 无需因为新增的状态而改变就更好了,不过这意味着切换到另一种设计模式。
  2. 存在一些重复的逻辑。 Postrequest_reviewapprove 这两个类似的实现都调用了 state 字段中 Option 值的同一方法,并在结果中为 state 字段设置了新值。如果Post中的很多方法都遵循这个模式,我们可能会考虑定义一个宏来消除重复。#todo:用宏改写这个例子#

当然,如果不使用 Option 改变 Post.state 的值也是可以的,关键在于 state 值的改变发生在 lib crate 中,而无需使用者更改。
所以可以使用 RefCell 达到内部可变性,从而替代了 Option.take 做的事情。当然,从效率上说,肯定 Option 的方式更高。

  1. // crate name: blog
  2. // src/lib.rs
  3. use std::cell::RefCell;
  4. pub struct Post {
  5. state: RefCell<Box<dyn State>>,
  6. content: String,
  7. }
  8. impl Post {
  9. pub fn new() -> Post {
  10. Post { state: RefCell::new(Box::new(DraftPost {})), content: String::new() }
  11. }
  12. pub fn add_text(&mut self, content: &str) { self.content.push_str(content); }
  13. pub fn content(&self) -> &str { self.state.borrow().content(self) }
  14. pub fn request_review(&mut self) {
  15. let new_state = self.state.borrow().request_review();
  16. let new_state = RefCell::new(new_state);
  17. self.state.swap(&new_state);
  18. }
  19. pub fn approve(&mut self) {
  20. let new_state = self.state.borrow().approve();
  21. let new_state = RefCell::new(new_state);
  22. self.state.swap(&new_state);
  23. }
  24. }
  25. trait State {
  26. fn content<'a>(&self, _post: &'a Post) -> &'a str { "" }
  27. fn request_review(&self) -> Box<dyn State>;
  28. fn approve(&self) -> Box<dyn State>;
  29. }
  30. struct DraftPost {}
  31. struct PendingReviewPost {}
  32. struct PublishedPost {}
  33. impl State for DraftPost {
  34. fn request_review(&self) -> Box<dyn State> { Box::new(PendingReviewPost {}) }
  35. fn approve(&self) -> Box<dyn State> { Box::new(Self {}) }
  36. }
  37. impl State for PendingReviewPost {
  38. fn request_review(&self) -> Box<dyn State> { Box::new(Self {}) }
  39. fn approve(&self) -> Box<dyn State> { Box::new(PublishedPost {}) }
  40. }
  41. impl State for PublishedPost {
  42. fn content<'a>(&self, post: &'a Post) -> &'a str { post.content.as_str() }
  43. fn request_review(&self) -> Box<dyn State> { Box::new(Self {}) }
  44. fn approve(&self) -> Box<dyn State> { Box::new(Self {}) }
  45. }

完全按照面向对象语言的定义实现这个模式并没有尽可能地利用 Rust 的优势。这个例子可以利用不同的类型来实现状态转移:

  1. // src/main.rs
  2. use blog::Post;
  3. fn main() {
  4. let mut post = Post::new(); // mut post: DraftPost
  5. post.add_text("I ate a salad for lunch today");
  6. let post = post.request_review(); // post: PendingReviewPost
  7. let post = post.approve(); // post: Post
  8. assert_eq!("I ate a salad for lunch today", post.content());
  9. }
  1. // `blog` crate: src/lib.rs
  2. pub struct Post {
  3. content: String,
  4. }
  5. pub struct DraftPost {
  6. content: String,
  7. }
  8. impl Post {
  9. pub fn new() -> DraftPost { DraftPost { content: String::new() } }
  10. pub fn content(&self) -> &str { &self.content }
  11. }
  12. impl DraftPost {
  13. pub fn add_text(&mut self, text: &str) { self.content.push_str(text); }
  14. pub fn request_review(self) -> PendingReviewPost {
  15. PendingReviewPost { content: self.content }
  16. }
  17. }
  18. pub struct PendingReviewPost {
  19. content: String,
  20. }
  21. impl PendingReviewPost {
  22. pub fn approve(self) -> Post { Post { content: self.content } }
  23. }

“改进” 版:

  1. 使用 trait object 和面向对象编程范式:适合逻辑重合度高的场景,但它并不总是最好的解决方案,因为 Rust 拥有像所有权这样的面向对象语言所没有的功能,而且动态分配会有少量的运行时性能损耗。

    1. // `blog` crate: src/lib.rs
    2. pub struct Post {
    3. state: Option<Box<dyn State>>,
    4. state_count: u32,
    5. content: String,
    6. editable: bool,
    7. }
    8. impl Post {
    9. pub fn new() -> Post {
    10. Post { state: Some(Box::new(Draft {})),
    11. content: String::new(),
    12. state_count: 1,
    13. editable: true, }
    14. }
    15. pub fn add_text(&mut self, text: &str) {
    16. if self.editable {
    17. self.content.push_str(text);
    18. } else {
    19. println!("Contents are not editable and not changed.");
    20. }
    21. }
    22. pub fn content(&self) -> &str { self.state.as_ref().unwrap().content(self) }
    23. pub fn request_review(&mut self) {
    24. if let Some(s) = self.state.take() {
    25. self.state_count += 1;
    26. self.editable = false;
    27. self.state = Some(s.request_review())
    28. }
    29. }
    30. pub fn approve(&mut self) {
    31. if self.state_count == 3 {
    32. if let Some(s) = self.state.take() {
    33. self.state = Some(s.approve())
    34. }
    35. }
    36. }
    37. pub fn reject(&mut self) {
    38. if let Some(s) = self.state.take() {
    39. self.state_count = 0;
    40. self.editable = true;
    41. self.state = Some(s.reject())
    42. }
    43. }
    44. }
    45. trait State {
    46. fn request_review(self: Box<Self>) -> Box<dyn State>;
    47. fn approve(self: Box<Self>) -> Box<dyn State>;
    48. fn content<'a>(&self, post: &'a Post) -> &'a str { "" }
    49. fn reject(self: Box<Self>) -> Box<dyn State> { Box::new(Draft {}) }
    50. }
    51. struct Draft {}
    52. impl State for Draft {
    53. fn request_review(self: Box<Self>) -> Box<dyn State> { Box::new(PendingReview {}) }
    54. fn approve(self: Box<Self>) -> Box<dyn State> { Box::new(Published {}) }
    55. }
    56. struct PendingReview {}
    57. impl State for PendingReview {
    58. fn request_review(self: Box<Self>) -> Box<dyn State> { self }
    59. fn approve(self: Box<Self>) -> Box<dyn State> { Box::new(Published {}) }
    60. }
    61. struct Published {}
    62. impl State for Published {
    63. fn request_review(self: Box<Self>) -> Box<dyn State> { self }
    64. fn approve(self: Box<Self>) -> Box<dyn State> { self }
    65. fn content<'a>(&self, post: &'a Post) -> &'a str { &post.content }
    66. }
    1. // src/main.rs
    2. use blog::Post;
    3. fn main() {
    4. let mut post = Post::new();
    5. post.add_text("I ate a salad for lunch today");
    6. assert_eq!("", post.content());
    7. post.request_review();
    8. assert_eq!("", post.content());
    9. post.reject();
    10. assert_eq!("", post.content());
    11. // reject 让 state 回到 Draft,因而可以编辑
    12. post.add_text(" lol");
    13. post.request_review();
    14. assert_eq!("", post.content());
    15. // 必须两次调用 request_review 才能进入到 published state
    16. post.approve();
    17. assert_eq!("", post.content());
    18. post.request_review();
    19. assert_eq!("", post.content());
    20. post.request_review();
    21. assert_eq!("", post.content());
    22. // 这里实际没有编辑,会打印 `Contents are not editable and not changed.`
    23. post.add_text(" foo");
    24. post.approve();
    25. assert_eq!("I ate a salad for lunch today lol", post.content());
    26. }
  2. 使用类型系统:显然思路也清晰,而且代码量少、拓展性同样高;充分发挥 Rust 的语言特性,比如这里的所有权、数据类型、编译时检查。

    1. // `blog` crate: src/lib.rs
    2. pub struct Post {
    3. content: String,
    4. }
    5. pub struct DraftPost {
    6. content: String,
    7. }
    8. impl Post {
    9. pub fn new() -> DraftPost { DraftPost { content: String::new() } }
    10. pub fn content(&self) -> &str { &self.content }
    11. }
    12. impl DraftPost {
    13. pub fn add_text(&mut self, text: &str) { self.content.push_str(text); }
    14. pub fn request_review(self) -> PendingReviewPost {
    15. PendingReviewPost { content: self.content }
    16. }
    17. }
    18. pub struct PendingReviewPost {
    19. content: String,
    20. }
    21. impl PendingReviewPost {
    22. pub fn reject(self) -> DraftPost { DraftPost { content: self.content } }
    23. pub fn request_review(self) -> ReviewApprovedPost {
    24. ReviewApprovedPost { content: self.content }
    25. }
    26. }
    27. pub struct ReviewApprovedPost {
    28. content: String,
    29. }
    30. impl ReviewApprovedPost {
    31. pub fn approve(self) -> Post { Post { content: self.content } }
    32. }
    1. // src/main.rs
    2. use rust_book_coding_in_practice::blog_use_type_extended::Post;
    3. fn main() {
    4. let mut post = Post::new();
    5. post.add_text("I ate a salad for lunch today");
    6. let post = post.request_review();
    7. // reject 将退回到 DraftPost,并且只能在 DraftPost 类型中编辑
    8. let mut post = post.reject();
    9. post.add_text(" lol");
    10. // 两次调用 request_review 的实例类型不一样
    11. // DraftPost -> PendingReviewPost
    12. let post = post.request_review();
    13. // PendingReviewPost -> ReviewApprovedPost
    14. let post = post.request_review();
    15. // ReviewApprovedPost -> Post
    16. let post = post.approve();
    17. assert_eq!("I ate a salad for lunch today lol", post.content());
    18. }