策略模式通过接口封装不同算法,使条件分支可扩展。以折扣系统为例,定义DiscountStrategy接口及多种实现,结合PricingContext执行策略,并用map映射类型与策略,消除if-else,提升维护性与扩展性。

在 Go 语言开发中,随着业务逻辑复杂度上升,if-else 或 switch 这类条件分支很容易变得难以维护。尤其当多个判断条件对应不同处理逻辑时,代码会变得冗长且不易扩展。策略模式提供了一种优雅的解决方案——将不同的算法或行为封装成独立的结构体,通过接口统一调用,从而消除复杂的条件判断。
策略模式的核心思想
策略模式定义一系列算法,把它们分别封装成类型,并使它们可以互换。客户端根据需要选择具体策略,而无需关心内部实现细节。
在 Go 中,我们通常使用接口(interface)来定义策略契约,再由不同的结构体实现该接口。这样可以在运行时动态替换行为,避免硬编码的条件分支。
用策略替代 if-else 的实际场景
假设我们要实现一个订单折扣系统,不同用户类型享受不同折扣:
立即学习“go语言免费学习笔记(深入)”;
- 普通用户:无折扣
- 会员用户:9 折
- 超级会员:8 折
传统写法可能如下:
func CalculatePrice(original float64, userType string) float64 {if userType == "normal" {
return original
} else if userType == "vip" {
return original * 0.9
} else if userType == "svip" {
return original * 0.8
}
return original
}
这种写法的问题很明显:新增用户类型需修改函数,违反开闭原则;逻辑集中不易测试;难以复用。
使用策略模式重构代码
我们先定义一个折扣策略接口:
type DiscountStrategy interface {Apply(float64) float64
}
然后为每种用户类型实现对应的策略:
func (d NormalDiscount) Apply(price float64) float64 { return price }
type VipDiscount struct{}
func (d VipDiscount) Apply(price float64) float64 { return price * 0.9 }
type SVipDiscount struct{}
func (d SVipDiscount) Apply(price float64) float64 { return price * 0.8 }
接下来创建一个上下文管理器来执行策略:
type PricingContext struct {strategy DiscountStrategy
}
func (c *PricingContext) SetStrategy(s DiscountStrategy) {
c.strategy = s
}
func (c *PricingContext) Calculate(price float64) float64 {
if c.strategy == nil {
panic("no strategy set")
}
return c.strategy.Apply(price)
}
使用方式非常清晰:
ctx := &PricingContext{}ctx.SetStrategy(NormalDiscount{})
fmt.Println(ctx.Calculate(100)) // 输出: 100
ctx.SetStrategy(VipDiscount{})
fmt.Println(ctx.Calculate(100)) // 输出: 90
此时,添加新策略不再影响原有代码,只需新增结构体并实现接口即可。
结合映射自动匹配策略
为了进一步简化调用端的判断逻辑,我们可以用 map 将用户类型与策略构造函数关联起来:
var strategies = map[string]DiscountStrategy{"normal": NormalDiscount{},
"vip": VipDiscount{},
"svip": SVipDiscount{},
}
调用时直接获取对应策略:
userType := "vip"strategy, exists := strategies[userType]
if !exists {
panic("unknown user type")
}
ctx.SetStrategy(strategy)
result := ctx.Calculate(100)
至此,所有条件分支都被移除,扩展性大幅提升。
基本上就这些。通过策略模式,Go 程序可以更灵活地应对多变的业务规则,让代码结构更清晰、更易维护。关键是设计好接口边界,并合理组织策略注册与分发机制。不复杂但容易忽略的是:保持策略的无状态性,避免在结构体中保存临时数据,确保并发安全。










