如何用Golang实现策略模式_Golang 策略模式实现实践

策略模式通过接口或函数封装不同算法,使算法可互换且符合开闭原则;以折扣计算为例,定义DiscountStrategy接口及多种会员折扣实现,Order上下文通过SetStrategy动态切换策略,调用GetFinalPrice获得不同折扣价;Golang中亦可用函数类型简化实现,定义DiscountFunc类型及对应函数,OrderFunc结构体接收函数式策略,实现更简洁的动态替换;该模式适用于支付方式、排序算法等需运行时切换行为的场景,避免多重条件判断,提升扩展性与维护性。

策略模式是一种行为设计模式,它让你定义一系列算法或行为,并将它们封装在独立的类或结构体中,使得它们可以互相替换。在 Golang 中虽然没有类的概念,但通过接口和函数式编程特性,我们可以很自然地实现策略模式。

策略模式的核心思想

将不同的“策略”(即算法)抽象为统一接口,客户端通过该接口调用具体策略,而无需关心内部实现。当需要更换算法时,只需更换策略实例,不修改原有代码,符合开闭原则。

比如:支付方式、排序算法、日志输出格式等场景都适合使用策略模式。

使用接口实现策略模式

假设我们有一个任务是计算订单折扣,不同会员等级有不同的折扣策略。

定义策略接口:

type DiscountStrategy interface {
    CalculateDiscount(amount float64) float64
}

实现具体策略:

type NormalUserStrategy struct{}

func (s NormalUserStrategy) CalculateDiscount(amount float64) float64 { return amount 0.05 // 5% 折扣 }

type VipUserStrategy struct{}

func (s VipUserStrategy) CalculateDiscount(amount float64) float64 { return amount 0.15 // 15% 折扣 }

type PremiumUserStrategy struct{}

func (s PremiumUserStrategy) CalculateDiscount(amount float64) float64 { return amount 0.30 // 30% 折扣 }

上下文使用策略:

type Order struct {
    Amount   float64
    Strategy DiscountStrategy
}

func (o *Order) SetStrategy(strategy DiscountStrategy) { o.Strategy = strategy }

func (o *Order) GetFinalPrice() float64 { discount := o.Strategy.CalculateDiscount(o.Amount) return o.Amount - discount }

使用示例:

order := &Order{Amount: 100}

order.SetStrategy(&NormalUserStrategy{}) fmt.Printf("普通用户最终价格: %.2f\n", order.GetFinalPrice()) // 95.00

order.SetStrategy(&VipUserStrategy{}) fmt.Printf("VIP用户最终价格: %.2f\n", order.GetFinalPrice()) // 85.00

order.SetStrategy(&PremiumUserStrategy{}) fmt.Printf("高级用户最终价格: %.2f\n", order.GetFinalPrice()) // 70.00

使用函数式风格简化策略模式

Golang 支持函数作为一等公民,我们可以用函数替代接口,使代码更简洁。

定义函数类型作为策略:

type DiscountFunc func(amount float64) float64

实现不同策略函数:

func NormalDiscount(amount float64) float64 {
    return amount * 0.05
}

func VipDiscount(amount float64) float64 { return amount * 0.15 }

func PremiumDiscount(amount float64) float64 { return amount * 0.30 }

更新上下文以接受函数:

type OrderFunc struct {
    Amount   float64
    Strategy DiscountFunc
}

func (o *OrderFunc) SetStrategy(f DiscountFunc) { o.Strategy = f }

func (o *OrderFunc) GetFinalPrice() float64 { if o.Strategy != nil { discount := o.Strategy(o.Amount) return o.Amount - discount } return o.Amount }

调用方式类似:

orderFunc := &OrderFunc{Amount: 100}
orderFunc.SetStrategy(NormalDiscount)
fmt.Printf("函数式-普通用户: %.2f\n", orderFunc.GetFinalPrice())

适用场景与优势

策略模式适用于以下情况:

  • 有多个相似类,仅行为不同
  • 需要在运行时动态切换算法
  • 避免使用多重条件判断(如 if-else 或 switch)选择行为

优势包括:

  • 算法可自由切换
  • 扩展性强,新增策略不影响现有代码
  • 职责清晰,每种策略独立封装

基本上就这些。Golang 虽无继承,但通过接口和函数式编程能优雅实现策略模式,关键是把变化的行为抽象出来,让调用方只依赖抽象,不依赖具体实现。