Go语言中间件：优雅处理多个业务共享复杂逻辑
在多个业务场景（例如：A、B、C、D）需要复用同一复杂业务逻辑（例如：“创建A”）时，如何避免代码冗余并提升可维护性至关重要。 直接在“创建A”逻辑中使用switch语句处理不同业务，随着业务增长，代码将变得难以维护。本文介绍一种基于Go语言注册机制的中间件设计方案，有效解决此问题。
假设A、B、C、D等业务都需要调用“创建A”逻辑，并在调用前后执行各自的预处理和后处理操作。“创建A”逻辑接收一个业务标识符name和创建数据，并返回处理结果。现有方案使用switch语句，存在代码臃肿和维护困难的问题。
为了解决这个问题，我们定义一个接口HandlePublicA，包含BeforeCreate、AfterCreate和Name三个方法：
立即学习“go语言免费学习笔记（深入）”；
-
BeforeCreate: 调用“创建A”逻辑前的预处理。 -
AfterCreate: “创建A”逻辑调用后的结果处理。 -
Name: 返回业务标识符。
PublicA结构体表示“创建A”的公共业务逻辑，其Do方法包含核心业务处理逻辑。PublicAParam和PublicARes分别表示“创建A”逻辑的输入和输出数据结构。
每个业务（例如业务A）都需要实现HandlePublicA接口，并根据自身需求实现BeforeCreate和AfterCreate方法。主程序使用HandlePublicA类型的切片注册各个业务，然后循环调用BeforeCreate、Do和AfterCreate方法，完成整个业务流程。
这种设计避免了switch语句，解耦了各个业务逻辑，提升了代码的可维护性和可扩展性。新增业务只需实现HandlePublicA接口并注册即可，无需修改现有代码。 通过接口和注册机制，实现了业务逻辑的解耦和复用，优雅地解决了代码冗余问题。
示例代码：
type PublicAParam struct{} // 创建公共业务A所需数据
type PublicA struct{} // 公共业务A
type PublicARes struct{} // 公共业务A生成的数
func (p PublicA) Do(name string) PublicARes { // 公共业务A的处理逻辑
return PublicARes{}
}
type HandlePublicA interface {
BeforeCreate(PublicAParam) PublicA
AfterCreate(PublicARes)
Name() string
}
type A struct { // 业务A
MyName string
}
func (a A) BeforeCreate(PublicAParam) PublicA { return PublicA{} }
func (a A) AfterCreate(PublicARes) {}
func (a A) Name() string { return a.MyName }
func main() {
param := PublicAParam{}
hooks := []HandlePublicA{A{MyName: "A"}}
for _, hook := range hooks {
p := hook.BeforeCreate(param)
after := p.Do(hook.Name())
hook.AfterCreate(after)
}
}
这种设计将“创建A”的公共逻辑与各个业务的特定逻辑有效分离，实现了代码的模块化和可扩展性。 公共业务A返回的结果再次被各个业务调用，保证了业务的独立性和可维护性。
