Go设计模式(21)-职责链模式

Go语言实现职责链模式：轻松构建中间件

原创已于 2022-06-02 10:59:07 修改 · 292 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#golang #设计模式 #中间件

于 2021-07-21 10:39:44 首次发布

Go设计模式专栏收录该内容

32 篇文章

订阅专栏

本文介绍了职责链模式，通过UML类图和代码示例展示了如何在Go中实现类似Gin框架的全局中间件功能。职责链模式允许请求被多个对象处理，提供扩展性，常用于框架的过滤器和拦截器。文中提供的代码示例展示了如何创建和执行中间件链，以及如何在链中检查和响应请求状态。

职责链将处理模块串联成链，请求沿着链条被处理，提供了很好的扩展性，而且能够去掉if-else。

UML类图位置：https://www.processon.com/view/link/60d29bf3e401fd49502afd25

本文代码链接为：https://github.com/shidawuhen/asap/blob/master/controller/design/21chain.go

1.定义

1.1职责链模式

职责链模式（Chain of Responsibility)：使多个对象都有机会处理请求，从而避免请求的发送者和接收者之间的耦合关系。将这个对象连成一条链，并沿着这条链传递该请求，直到有一个对象处理它为止。

UML：

1.2分析

职责链模式的定义已经将模式说的比较清楚了，一个请求需要被多个对象进行处理，可以将处理对象连成一条链，挨个处理请求。

连成链的方式比较多样，可以用UML中展示的那样，一个处理对象使用SetSuccessor引用下一个处理对象。也可以使用array或者list存储所有处理对象，使用循环方式遍历。

对于第二种方式，是否感觉有些像Go设计模式(18)-观察者模式。两者具体实现、目的都差不多，主要区别在观察者模式中的处理对象功能可能完全不相似，而且观察者模式主要负责将信息传递给处理对象即可。职责链模式的处理对象功能一般相似，另外职责链模式也关注请求是否正确被处理。

另外，定义里说”直到有一个对象处理它“也不太准确，有多少对象可以处理请求看具体需求，极端情况下每一个对象都可以处理请求。

职责链模式的核心在于将处理对象整理成链路。

2.应用场景

如果请求被多个对象进行处理，就可以用职责链模式。具体业务的像敏感词脱敏，框架中的过滤器、拦截器等。

总体感觉框架中使用的比较多一些，研发人员能够快速扩展出自己的过滤器和拦截器。

以前写过Gin源码剖析，里面有全局中间件的概念，而全局中间件使用的便是职责链模式。

3.代码实现

我们仿照Gin，实现Gin的全局中间件功能。

package main

import "fmt"

var status int8 = 0

type HandlerFunc func()

type HandlersChain []HandlerFunc

/**
 * @Author: Jason Pang
 * @Description:
 */
type RouterGroup struct {
   Handlers HandlersChain
   index    int8
}

/**
 * @Author: Jason Pang
 * @Description: 添加中间件，将其组成链式
 * @receiver group
 * @param middleware
 */
func (group *RouterGroup) Use(middleware ...HandlerFunc) {
   group.Handlers = append(group.Handlers, middleware...)
}

/**
 * @Author: Jason Pang
 * @Description: 链顺序执行
 * @receiver group
 */
func (group *RouterGroup) Next() {
   for group.index < int8(len(group.Handlers)) {
      group.Handlers[group.index]()
      group.index++
   }
}

/**
 * @Author: Jason Pang
 * @Description: 中间件
 */
func middleware1() {
   fmt.Println("全局中间件1执行完毕")
}

/**
 * @Author: Jason Pang
 * @Description: 中间件
 */
func middleware2() {
   fmt.Println("全局中间件2执行失败")
   status = 1
}

func main() {
   r := &RouterGroup{}
   //添加中间件
   r.Use(middleware1, middleware2)
   //运行中间件
   r.Next()
   //状态检查
   if status == 1 {
      fmt.Println("中间件检查失败，请重试")
      return
   }
   //执行后续流程
}