Go 中策略模式通过定义统一接口(如 PaymentStrategy)封装不同算法,各策略类型独立实现,上下文(PaymentContext)持有接口并支持运行时切换,结合依赖注入与工厂函数实现动态、可插拔、易测试的算法选择。

在 Go 中用策略模式实现动态算法选择,核心是把不同算法封装成独立类型,通过统一接口调用,运行时按需切换——不改逻辑代码,只换策略实例。
定义策略接口,约束行为一致性
所有算法必须满足同一契约。比如处理支付的场景,不同渠道(微信 、 支付宝、银行卡)有各自签名和提交逻辑,但对外都叫 Pay():
type PaymentStrategy interface {Pay(amount float64, orderID string) error Validate() bool}
接口越小越专注,只暴露必要方法。避免塞入配置、日志等无关行为,保持策略“可插拔”。
实现具体策略,各自封装差异逻辑
每个结构体实现接口,内部只管自己那块事。例如 微信支付 策略负责组装 XML、调用统一下单 API;支付宝策略则生成 JSON 请求并验签:
立即学习“go 语言免费学习笔记(深入)”;
type WechatPay struct{appID, mchID, apiKey string} func (w WechatPay) Pay(amount float64, orderID string) error {// 构造微信下单请求,签名,HTTP 调用…… return nil} func (w WechatPay) Validate() bool { return w.appID != "" && w.mchID != ""} type AlipayPay struct{appID, privateKey, publicKey string} func (a AlipayPay) Pay(amount float64, orderID string) error {// 组装 Alipay SDK 参数,RSA 签名,发起请求…… return nil} func (a AlipayPay) Validate() bool { return a.appID != "" && a.privateKey != ""}
策略之间完全解耦,新增一种支付方式只需加新类型,不影响已有代码。
用上下文管理策略,支持运行时切换
上下文(Context)不关心具体怎么算,只持有策略接口,并提供设置和执行入口:
type PaymentContext struct {strategy PaymentStrategy} func (p *PaymentContext) SetStrategy(s PaymentStrategy) {p.strategy = s} func (p *PaymentContext) Execute(amount float64, orderID string) error {if p.strategy == nil { return fmt.Errorf("no strategy set") } if !p.strategy.Validate() { return fmt.Errorf("invalid strategy config") } return p.strategy.Pay(amount, orderID) }
- 初始化时可设默认策略:
ctx := &PaymentContext{strategy: WechatPay{……}} - 根据用户选择、订单类型或灰度规则动态换策略:
ctx.SetStrategy(AlipayPay{……}) - 也可结合工厂函数,按字符串返回对应策略:
GetStrategy("alipay")
配合依赖注入,让策略更易测试和扩展
不要在上下文中 new 具体策略,而是由外部传入。这样单元测试可轻松注入 mock 策略:
// 测试用的假策略 type MockPay struct{Called bool} func (m *MockPay) Pay(_ float64, _ string) error {m.Called = true; return nil} func (m *MockPay) Validate() bool { return true} // 测试中直接替换 ctx := &PaymentContext{} ctx.SetStrategy(&MockPay{}) ctx.Execute(100.0, "test123") if !mock.Called {t.Fatal("expected Pay to be called") }
生产环境可通过配置中心或 URL 参数决定策略类型,再由工厂构造对应实例注入上下文,真正实现“逻辑不变、行为可配”。