Golang多业务共享逻辑如何优雅处理？

Golang中间件模式：高效处理共享业务逻辑

本文介绍如何在Golang中设计高效的中间件层，优雅地处理多个业务共享同一复杂逻辑的情况，避免代码冗余和维护难题。

假设有多个业务（例如A、B、C、D），都需要调用一个名为“创建A”的复杂业务逻辑。每个业务在调用“创建A”前需要准备各自的数据（数据结构相同），“创建A”完成后会生成新的数据（数据结构也相同），然后由各个业务分别进行后续处理。此外，各个业务可能还需要在“创建A”前后执行一些特定操作，这些操作依赖于输入和输出参数。

直接使用switch语句处理多个业务会导致代码膨胀，难以维护。因此，我们需要一种更灵活的注册机制。

立即学习“go语言免费学习笔记（深入）”；

改进方案的核心是定义一个接口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{}      // 公共业务Atype 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)    }}

登录后复制