【最佳实践】Go 状态模式

news2025/3/19 4:45:18

设计思路

状态模式的核心在于将对象的行为封装在特定的状态类中，使得对象在不同的状态下表现出不同的行为。每个状态实现同一个接口，允许对象在运行时通过改变其内部状态对象来改变其行为。状态模式使得状态转换更加明确，并且易于扩展新的状态和行为。

场景描述

假设我们有一个简单的文档编辑器，文档可以处于不同的编辑状态：草稿、审核、发布。每个状态对应不同的行为，比如在草稿状态下可以编辑内容，在审核状态下可以提交审核，在发布状态下可以查看内容但不能编辑。我们将使用状态模式来实现这一场景。

实现示例

下面是一个使用状态模式实现的 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 函数：演示了如何使用状态模式管理文档的状态转换。文档初始状态为草稿，可以编辑和提交审核；审核状态下可以发布；发布状态下可以查看，但不能编辑或再次发布。