17370845950

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

在 Golang 中,策略模式能有效替代冗长的 if-else 或 switch 条件判断,提升代码可维护性和扩展性。当业务逻辑依赖多种条件分支时,直接使用条件语句容易导致函数臃肿、难以测试和修改。通过策略模式,可以将每种“策略”封装为独立的行为,再结合条件选择对应策略执行,实现清晰解耦。

策略模式基本结构

策略模式核心包含三个部分:策略接口、具体策略实现、上下文调用者。Golang 中通过接口和函数注入实现。

定义一个策略接口,比如处理不同类型的折扣:

type DiscountStrategy interface {
    Calculate(price float64) float64
}

实现多个具体策略:

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) float64

func 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。无论是接口实现还是函数式风格,都能显著提升代码可读性和可测试性。实际项目中,可根据复杂度选择合适形式——简单场景用函数,复杂行为用结构体+接口。

基本上就这些,不复杂但容易忽略设计初衷:让变化的部分独立出来,而不是散落在逻辑深处。