如何使用Golang实现策略模式_使用Strategy Pattern动态切换算法

P粉602998670 2025-12-16 00:00:00 次阅读

Go中策略模式通过接口定义算法契约、结构体实现具体策略、上下文组合策略实现动态切换，遵循“组合优于继承”，支持运行时更换策略及工厂注册扩展。

在 Go 语言中实现策略模式，核心是用接口定义算法族，用结构体实现具体策略，再通过组合让上下文（Context）动态切换行为。它不依赖继承，而是靠“组合优于继承”的思想，天然契合 Go 的接口机制。

定义策略接口

所有算法需满足同一契约，所以先声明一个通用接口。比如处理支付的场景：

type PaymentStrategy interface {
    Pay(amount float64) error
}

这个接口简单清晰，只暴露行为，不关心内部实现。

实现多个具体策略

每个策略是一个独立结构体，实现接口方法。例如微信支付和支付宝支付：

type WechatPay struct{}

func (w WechatPay) Pay(amount float64) error {
    fmt.Printf("WeChat Pay: ¥%.2f\n", amount)
    return nil
}

type Alipay struct{}

func (a Alipay) Pay(amount float64) error {
    fmt.Printf("Alipay: ¥%.2f\n", amount)
    return nil
}

策略之间完全解耦，新增策略只需实现接口，无需修改已有代码
结构体可以是空结构体（如上），也可以带字段（如含 API key、签名配置等）
方法接收者用值类型还是指针，取决于是否需要修改内部状态

创建上下文并支持运行时切换

上下文持有策略接口，并提供设置和执行方法：

type PaymentContext struct {
    strategy PaymentStrategy
}

func (p *PaymentContext) SetStrategy(s PaymentStrategy) {
    p.strategy = s
}

func (p *PaymentContext) ExecutePayment(amount float64) error {
    if p.strategy == nil {
        return errors.New("no payment strategy set")
    }
    return p.strategy.Pay(amount)
}

使用时可随时更换策略：

ctx := &PaymentContext{}
ctx.SetStrategy(WechatPay{})
ctx.ExecutePayment(99.9)

ctx.SetStrategy(Alipay{})
ctx.ExecutePayment(199.9)

进阶：策略注册与工厂封装（可选）

当策略较多时，可用 map + 工厂函数统一管理：

var strategies = map[string]PaymentStrategy{
    "wechat": WechatPay{},
    "alipay": Alipay{},
}

func GetStrategy(name string) (PaymentStrategy, bool) {
    s, ok := strategies[name]
    return s, ok
}