设计思路
状态模式的核心在于将对象的行为封装在特定的状态类中,使得对象在不同的状态下表现出不同的行为。每个状态实现同一个接口,允许对象在运行时通过改变其内部状态对象来改变其行为。状态模式使得状态转换更加明确,并且易于扩展新的状态和行为。
场景描述
假设我们有一个简单的文档编辑器,文档可以处于不同的编辑状态:草稿、审核、发布。每个状态对应不同的行为,比如在草稿状态下可以编辑内容,在审核状态下可以提交审核,在发布状态下可以查看内容但不能编辑。我们将使用状态模式来实现这一场景。
实现示例
下面是一个使用状态模式实现的 Go 语言示例,展示了一个简单的文档编辑系统:
package main
import "fmt"
// DocumentState 是状态接口,定义了文档状态的行为
type DocumentState interface {
Edit() error
Submit() error
Publish() error
}
// Document 是包含状态的上下文
type Document struct {
state DocumentState
}
// SetState 设置文档的当前状态
func (d *Document) SetState(state DocumentState) {
d.state = state
}
// Edit 触发编辑行为
func (d *Document) Edit() error {
return d.state.Edit()
}
// Submit 触发提交审核行为
func (d *Document) Submit() error {
return d.state.Submit()
}
// Publish 触发发布行为
func (d *Document) Publish() error {
return d.state.Publish()
}
// DraftState 是草稿状态
type DraftState struct {
document *Document
}
func (s *DraftState) Edit() error {
fmt.Println("Editing the document.")
return nil
}
func (s *DraftState) Submit() error {
fmt.Println("Submitting the document for review.")
s.document.SetState(&ReviewState{document: s.document})
return nil
}
func (s *DraftState) Publish() error {
return fmt.Errorf("document cannot be published directly from draft")
}
// ReviewState 是审核状态
type ReviewState struct {
document *Document
}
func (s *ReviewState) Edit() error {
return fmt.Errorf("document cannot be edited during review")
}
func (s *ReviewState) Submit() error {
return fmt.Errorf("document is already under review")
}
func (s *ReviewState) Publish() error {
fmt.Println("Publishing the document.")
s.document.SetState(&PublishedState{document: s.document})
return nil
}
// PublishedState 是已发布状态
type PublishedState struct {
document *Document
}
func (s *PublishedState) Edit() error {
return fmt.Errorf("document cannot be edited after publishing")
}
func (s *PublishedState) Submit() error {
return fmt.Errorf("document is already published")
}
func (s *PublishedState) Publish() error {
fmt.Println("Document is already published.")
return nil
}
func main() {
// 创建一个文档,并设置初始状态为草稿
document := &Document{}
initialState := &DraftState{document: document}
document.SetState(initialState)
// 编辑文档
err := document.Edit()
if err != nil {
fmt.Println("Error:", err)
}
// 提交文档审核
err = document.Submit()
if err != nil {
fmt.Println("Error:", err)
}
// 尝试编辑文档(应失败,因为在审核中)
err = document.Edit()
if err != nil {
fmt.Println("Error:", err)
}
// 发布文档
err = document.Publish()
if err != nil {
fmt.Println("Error:", err)
}
// 尝试发布文档(应成功,因为已发布)
err = document.Publish()
if err != nil {
fmt.Println("Error:", err)
}
}
代码解释
DocumentState 接口:定义了文档状态的行为接口,包括
Edit
、Submit
和Publish
方法。每个状态实现这些行为。Document 结构体:代表文档对象,包含一个
DocumentState
。通过SetState
方法可以改变文档的状态。DraftState、ReviewState、PublishedState:分别实现了
DocumentState
接口,表示文档的不同状态。每个状态根据其特性实现了相应的行为。main 函数:演示了如何使用状态模式管理文档的状态转换。文档初始状态为草稿,可以编辑和提交审核;审核状态下可以发布;发布状态下可以查看,但不能编辑或再次发布。
总结
「状态模式」抽象过程的核心是:
- 每一个状态映射对应行为
- 行为实现同一个接口interface
- 行为是内部的一个状态
- 状态是不断变化的