Golang中间件模式:高效处理共享业务逻辑
本文介绍如何在Golang中设计高效的中间件层,优雅地处理多个业务共享同一复杂逻辑的情况,避免代码冗余和维护难题。
假设有多个业务(例如A、B、C、D),都需要调用一个名为“创建A”的复杂业务逻辑。每个业务在调用“创建A”前需要准备各自的数据(数据结构相同),“创建A”完成后会生成新的数据(数据结构也相同),然后由各个业务分别进行后续处理。此外,各个业务可能还需要在“创建A”前后执行一些特定操作,这些操作依赖于输入和输出参数。
直接使用switch语句处理多个业务会导致代码膨胀,难以维护。因此,我们需要一种更灵活的注册机制。
改进方案的核心是定义一个接口HandlePublicA
,所有业务都需要实现该接口中的BeforeCreate
、AfterCreate
和Name
方法。BeforeCreate
方法在调用“创建A”前执行,AfterCreate
方法在调用“创建A”后执行,Name
方法返回业务名称。
PublicA
结构体封装了“创建A”的核心逻辑,Do
方法执行“创建A”逻辑并返回结果。PublicAParam
和PublicARes
分别表示“创建A”的输入和输出数据结构。
通过这种设计,每个业务实现HandlePublicA
接口,将各自的预处理和后处理逻辑封装在BeforeCreate
和AfterCreate
方法中。主程序循环遍历注册的业务,依次调用BeforeCreate
、PublicA.Do
和AfterCreate
方法,完成整个业务流程。这种方式避免了庞大的switch语句,提高了代码的可维护性和可扩展性。
示例代码:
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(param PublicAParam) PublicA { return PublicA{} } func (a A) AfterCreate(res 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) res := p.Do(hook.Name()) hook.AfterCreate(res) } }
公共业务A返回的结果会被各个业务再次调用,从而有效解耦业务模块,提高代码的可维护性和可扩展性。