如何在Golang中实现责任链模式分发请求

责任链模式通过接口定义处理器行为,用组合串联多个处理器,请求沿链传递直至被处理。

在Golang中实现责任链模式,核心是让多个处理器依次处理请求,直到某个处理器能够处理为止。每个处理器持有下一个处理器的引用,形成一条“链”。当一个请求到来时,它被传递给链上的第一个处理器,如果当前处理器无法处理,就转发给下一个,直到被处理或链结束。

定义请求和处理器接口

先明确请求数据结构和处理器行为。通常用接口来抽象处理器,便于扩展。

Handler 接口:包含处理方法和设置后继者的方法。

type Request struct {
    Type  string
    Data  string
}

type Handler interface {
    Handle(req *Request)
    SetNext(next Handler)
}

实现具体处理器

每个处理器实现自己的逻辑,判断是否能处理请求。不能则交给下一个。

type AuthHandler struct {
    next Handler
}

func (h *AuthHandler) Handle(req *Request) {
    if req.Type == "auth" {
        println("AuthHandler 处理认证请求:", req.Data)
        return
    }
    if h.next != nil {
        h.next.Handle(req)
    }
}

func (h *AuthHandler) SetNext(next Handler) {
    h.next = next
}

类似地,可定义日志、缓存等处理器:

type LogHandler struct {
    next Handler
}

func (h *LogHandler) Handle(req *Request) {
    println("LogHandler 记录请求:", req.Type, req.Data)
    if h.next != nil {
        h.next.Handle(req)
    }
}

func (h *LogHandler) SetNext(next Handler) {
    h.next = next
}

构建和使用责任链

将处理器串联起来,从链头开始分发请求。

func main() {
    // 创建处理器实例
    logHandler := &LogHandler{}
    authHandler := &AuthHandler{}

    // 组装链:日志 -> 认证
    logHandler.SetNext(authHandler)

    // 发起请求
    req1 := &Request{Type: "auth", Data: "login"}
    logHandler.Handle(req1)

    // 输出:
    // LogHandler 记录请求: auth login
    // AuthHandler 处理认证请求: login
}

若请求类型不是 "auth",则只有日志输出,认证处理器会跳过并尝试下一级(如果有)。

灵活扩展与实际应用建议

责任链适合处理多阶段过滤或拦截场景,如中间件、审批流程、API网关请求处理等。

  • 可通过函数式编程简化链式构造,例如返回 Handler 的构造函数
  • 加入上下文(context.Context)支持超时和取消
  • 避免链过长导致性能下降,必要时添加终止条件
  • 可结合选项模式(Option Pattern)配置处理器行为
基本上就这些。Golang没有继承,但通过接口和组合轻松实现责任链,代码清晰且易于维护。