策略模式通过封装不同行为并分离策略选择与执行,替代冗长条件判断,提升代码可维护性;在Golang中可用接口或函数式编程实现,结合工厂方法根据条件动态选取策略,降低耦合,增强扩展性。

在 Golang 中,策略模式能有效替代冗长的 if-else 或 switch 条件判断,提升代码可维护性和扩展性。当业务逻辑依赖多种条件分支时,直接使用条件语句容易导致函数臃肿、难以测试和修改。通过策略模式,可以将每种“策略”封装为独立的行为,再结合条件选择对应策略执行,实现清晰解耦。
策略模式基本结构
策略模式核心包含三个部分:策略接口、具体策略实现、上下文调用者。Golang 中通过接口和函数注入实现。
定义一个策略接口,比如处理不同类型的折扣:
type DiscountStrategy interface {
Calculate(price float64) float64
}
实现多个具体策略:
立即学习“go语言免费学习笔记(深入)”;
type NormalDiscount struct{}
func (n NormalDiscount) Calculate(price float64) float64 {
return price 0.95 // 95折
}
type VIPDiscount struct{}
func (v VIPDiscount) Calculate(price float64) float64 {
return price 0.8 // 8折
}
type HolidayDiscount struct{}
func (h HolidayDiscount) Calculate(price float64) float64 {
return price 0.7 // 7折
}
条件判断与策略映射结合
传统做法可能写成一连串 if-else:
func GetDiscount(userType string, isHoliday bool, price float64) float64 {
if isHoliday {
return price * 0.7
} else if userType == "vip" {
return price * 0.8
} else {
return price * 0.95
}
}
这种写法难以扩展,新增策略需修改原函数。改用策略+映射方式更优雅:
type DiscountContext struct {
strategy DiscountStrategy
}
func (d *DiscountContext) SetStrategy(strategy DiscountStrategy) {
d.strategy = strategy
}
func (d *DiscountContext) Execute(price float64) float64 {
if d.strategy == nil {
panic("no strategy set")
}
return d.strategy.Calculate(price)
}
构建策略工厂或映射表,根据条件自动匹配策略:
func NewDiscountStrategy(userType string, isHoliday bool) DiscountStrategy {
if isHoliday {
return &HolidayDiscount{}
}
switch userType {
case "vip":
return &VIPDiscount{}
default:
return &NormalDiscount{}
}
}
使用时:
context := &DiscountContext{}
strategy := NewDiscountStrategy("vip", true)
context.SetStrategy(strategy)
result := context.Execute(100) // 输出 70
进一步优化:函数式策略
Golang 支持函数作为一等公民,可简化策略定义。将策略抽象为函数类型:
type DiscountFunc func(float64) float64func GetDiscountFunc(userType string, isHoliday bool) DiscountFunc { if isHoliday { return func(p float64) float64 { return p 0.7 } } if userType == "vip" { return func(p float64) float64 { return p 0.8 } } return func(p float64) float64 { return p * 0.95 } }
调用更简洁:
discount := GetDiscountFunc("normal", false)
finalPrice := discount(100) // 95
这种方式适合逻辑简单、无需状态的场景,减少结构体定义,提升灵活性。
总结
策略模式结合条件判断的关键在于:将“选策略”和“执行策略”分离。通过映射或工厂方法根据条件返回对应策略,避免分散的 if-else。无论是接口实现还是函数式风格,都能显著提升代码可读性和可测试性。实际项目中,可根据复杂度选择合适形式——简单场景用函数,复杂行为用结构体+接口。
基本上就这些,不复杂但容易忽略设计初衷:让变化的部分独立出来,而不是散落在逻辑深处。










