策略模式通过将算法选择与执行分离,解决Go中复杂条件逻辑导致的代码臃肿问题。首先定义支付策略接口PaymentStrategy及其实现如Alipay、WeChatPay,再由PaymentContext上下文调用;接着使用映射表strategies存储策略,结合工厂函数GetStrategy根据输入动态选取,避免硬编码判断;进一步可从JSON配置加载用户类型到策略的映射,实现无需修改代码的灵活扩展;对于简单场景,可用函数式编程定义DiscountFunc类型和Discounts函数映射,直接执行对应折扣逻辑;最终通过接口、组合与函数式设计,将“选策略”与“用策略”解耦,提升可维护性与扩展性。

在Go语言中,策略模式常用于解耦算法实现与使用逻辑。当业务中存在大量条件判断(如if-else或switch)并伴随不同行为时,直接写条件逻辑容易导致代码臃肿、难以维护。通过将策略模式与条件逻辑结合,可以提升代码的可读性与扩展性。
策略模式基础结构
策略模式包含三个核心部分:上下文(Context)、策略接口和具体策略实现。在Go中,通常用接口表示策略,结构体实现具体行为。
例如,定义一个支付策略:
type PaymentStrategy interface {
Pay(amount float64) string
}
type Alipay struct{}
立即学习“go语言免费学习笔记(深入)”;
func (a *Alipay) Pay(amount float64) string {
return fmt.Sprintf("支付宝支付 %.2f 元", amount)
}
type WeChatPay struct{}
func (w *WeChatPay) Pay(amount float64) string {
return fmt.Sprintf("微信支付 %.2f 元", amount)
}
上下文持有策略接口,运行时动态切换:
type PaymentContext struct {
strategy PaymentStrategy
}
func (p *PaymentContext) SetStrategy(s PaymentStrategy) {
p.strategy = s
}
func (p *PaymentContext) ExecutePayment(amount float64) string {
return p.strategy.Pay(amount)
}
将条件逻辑映射为策略选择
常见场景是根据用户类型、订单金额或地区决定支付方式。若用if-else嵌套,会随着规则增加而变得复杂。此时可将条件判断封装为策略工厂函数。
定义一个策略注册与选择机制:
var strategies = map[string]PaymentStrategy{
"alipay": &Alipay{},
"wechat": &WeChatPay{},
"credit": &CreditCard{},
}
通过输入参数(如支付方式名称)直接获取对应策略:
func GetStrategy(paymentMethod string) PaymentStrategy {
if strategy, exists := strategies[paymentMethod]; exists {
return strategy
}
return nil // 或返回默认策略
}
调用时无需层层判断:
context := &PaymentContext{}
strategy := GetStrategy("alipay")
if strategy != nil {
context.SetStrategy(strategy)
result := context.ExecutePayment(99.9)
fmt.Println(result) // 输出:支付宝支付 99.90 元
}
结合配置或规则引擎动态加载策略
更高级的应用是将策略选择与外部配置结合。例如从JSON配置文件读取用户等级对应的策略:
{
"vip": "alipay",
"normal": "wechat"
}
程序启动时加载映射关系:
func LoadStrategiesFromConfig(configFile string) error {
data, _ := ioutil.ReadFile(configFile)
var config map[string]string
json.Unmarshal(data, &config)
for userType, method := range config {
if strategy, ok := strategies[method]; ok {
userStrategyMap[userType] = strategy
}
}
return nil}
这样,新增策略只需修改配置,无需改动主逻辑。
使用函数式策略简化简单场景
对于轻量级行为差异,可用函数代替接口。Go支持将函数作为值传递,使策略更简洁。
type DiscountFunc func(price float64) float64var Discounts = map[string]DiscountFunc{ "student": func(p float64) float64 { return p 0.8 }, "senior": func(p float64) float64 { return p 0.7 }, "normal": func(p float64) float64 { return p }, }
调用时直接执行:
discount := Discounts["student"] finalPrice := discount(100) // 得到80
基本上就这些。策略模式与条件逻辑结合的关键在于避免硬编码分支,把“选什么”和“做什么”分开。Go虽无类继承,但凭借接口和组合,能优雅实现这一设计思想。实际项目中,配合依赖注入或选项模式,还能进一步提升灵活性。










