装饰模式（Decorator Pattern）在 Go 语言中的应用

引言

在软件开发中，设计模式是解决常见问题的模板。装饰模式（Decorator Pattern）是一种结构型设计模式，它允许你在运行时动态地给对象添加职责。本文将详细介绍装饰模式在 Go 语言中的应用，并通过一个具体的例子来展示其使用方法。此外，我们还将探讨装饰模式与中间件的区别，帮助你更好地理解这两种模式的应用场景。

什么是装饰模式？

装饰模式的核心思想是通过创建一个包装类（装饰器）来动态地给对象添加新的功能。这种方式比继承更灵活，因为它不会改变原始类的结构，同时还能在运行时动态地添加或移除功能。

在Go语言中的应用

假设我们有一个权限校验接口 RightsChecker，它负责检查用户的权限。我们希望在权限校验的过程中添加一些额外的行为，比如日志记录、权限检查等。下面是一个具体的实现示例：

定义接口

首先，定义 RightsChecker 接口：

type CheckRightsRequest struct {
	Authorization string `header:"authorization" validate:"required"`
	User          string `form:"user"`
	OpenKfID      string `json:"open_kf_id" validate:"required"`
	ChildrenId    int64  `json:"children_id,optional"` // 档案id
}

type CheckRightsResponse struct {
	RemainTimes int64 `json:"remain_times"`
	Enabled     int64 `json:"enabled"`
}

type RightsChecker interface {
	CheckRights(req *CheckRightsRequest) (resp *CheckRightsResponse, err error)
}

实现具体逻辑

接下来，实现具体的权限校验逻辑 CheckRightsLogic：

type CheckRightsLogic struct {
	ctx    context.Context
	svcCtx context.Context
}

func (l *CheckRightsLogic) CheckRights(req *CheckRightsRequest) (resp *CheckRightsResponse, err error) {
	// 原始的权限校验逻辑
	riskService := risk.NewRisk()
	enable, times, err := riskService.Check()
	if err != nil {
		return nil, err
	}
	resp = &CheckRightsResponse{
		RemainTimes: times,
		Enabled:     enable,
	}
	return resp, nil
}

创建装饰器

然后，创建一个装饰器 RightsCheckerDecorator，在其中添加额外的行为：

type RightsCheckerDecorator struct {
	checker RightsChecker
}

func (d *RightsCheckerDecorator) CheckRights(req *CheckRightsRequest) (resp *CheckRightsResponse, err error) {
	// 在这里可以添加额外的行为，例如日志记录、权限检查等
	fmt.Println("Logging request:", req)

	// 调用嵌入的 checker 的 CheckRights 方法
	resp, err = d.checker.CheckRights(req)
	if err != nil {
		fmt.Println("Error occurred during rights check:", err)
		return
	}

	// 在这里可以添加额外的行为，例如日志记录、结果处理等
	fmt.Println("Logging response:", resp)

	return
}

使用装饰器

最后，创建装饰器并使用它：

func NewRightsCheckerDecorator(checker RightsChecker) RightsChecker {
	return &RightsCheckerDecorator{
		checker: checker,
	}
}

func main() {
	logic := &CheckRightsLogic{ctx: context.Background(), svcCtx: context.Background()}
	decoratedChecker := NewRightsCheckerDecorator(logic)

	// 使用装饰后的 checker 进行权限校验
	req := &CheckRightsRequest{
		Authorization: "Bearer token123",
		User:          "user456",
		OpenKfID:      "kf123",
		ChildrenId:    12345,
	}
	resp, err := decoratedChecker.CheckRights(req)
	if err != nil {
		// 处理错误
		fmt.Println(err)
	}
	// 处理响应
	fmt.Println(resp)
}

装饰模式 vs 中间件

装饰模式

• 定义：装饰模式通过创建一个包装类（装饰器）来动态地给对象添加新的功能。
• 应用场景：适用于需要在运行时动态地给对象添加职责的情况，特别是在不改变原有对象结构的前提下。
• 特点：
  • 灵活性高，可以在运行时动态地添加或移除功能。
  • 不改变原有对象的结构。
  • 适合复杂的业务逻辑，可以逐步添加功能。

中间件

• 定义：中间件是一种位于请求处理链中的组件，通常用于处理请求和响应的通用任务，如日志记录、认证、错误处理等。
• 应用场景：适用于 Web 框架、API 服务器等需要处理大量请求的场景。
• 特点：
  • 通常用于处理请求和响应的通用任务。
  • 可以在请求处理链中按顺序执行多个中间件。
  • 适合处理跨切面的通用逻辑，如日志记录、认证等。

区别

1. 目的：

   • 装饰模式主要用于给对象动态地添加职责，而不改变原有对象的结构。
   • 中间件主要用于处理请求和响应的通用任务，通常用于 Web 框架和 API 服务器。

2. 使用场景：

   • 装饰模式适用于需要在运行时动态地给对象添加功能的场景。
   • 中间件适用于需要处理大量请求的场景，特别是在 Web 开发中。

3. 实现方式：

   • 装饰模式通过创建包装类来实现。
   • 中间件通过在请求处理链中按顺序执行多个中间件来实现。

总结

装饰模式是一种强大的设计模式，适用于需要在运行时动态地给对象添加职责的场景。通过本文的介绍和示例代码，相信你已经掌握了如何在 Go 语言中使用装饰模式。同时，我们也探讨了装饰模式与中间件的区别，帮助你更好地选择合适的设计模式来解决实际问题。

如果你有任何问题或建议，欢迎留言交流！