探索Golang中的策略模式:一种灵活的编程艺术

发表时间: 2024-06-04 11:05

先扯点儿废话,向往的诗和远方:

独库公路的建设历时10年,于1983年正式通车。由于公路穿越高山、峡谷、河流等复杂地形,施工难度极大,因此被誉为“天路”。独库公路每年冬季由于积雪和冰冻封闭,通常在6月至10月之间开放。

这条公路沿途风景秀丽,经过天山天池、那拉提草原、巴音布鲁克草原等多个著名景点,是自驾游和户外爱好者的理想线路。此外,独库公路的修建也极大地促进了新疆南北区域经济和文化的交流。


书归正传,好好搬砖,才能更好的去欣赏世界。


需求:

获取一个csgo饰品的定价,需要从多个渠道(steam、buff、igxe……)等等。

需求很简单,下班前搞定吧


策略模式实现

定义一个传输对象:dto.go

package price

type GetPriceReq struct {
GoodsId int64 `json:"goods_id"` // 饰品ID
}

具体实现:interface.go

package price

import (
"context"
"strings"
)

// GoodsStrategyPrice 饰品定价策略
type GoodsStrategyPrice interface {
// GetPrice 根据饰品ID获取系统自定义价格
GetPrice(ctx context.Context, req *GetPriceReq) int64
}

// GoodsStrategy 用于选择和执行策略
type GoodsStrategy struct {
strategy GoodsStrategyPrice
}

// SetStrategy 选择策略
func (s *GoodsStrategy) SetStrategy(strategy GoodsStrategyPrice) {
s.strategy = strategy
}

// GetPrice 执行策略
func (s *GoodsStrategy) GetPrice(ctx context.Context, req *GetPriceReq) int64 {
return s.strategy.GetPrice(ctx, req)
}

// GetPriceByGoods 根据策略获取价格
func GetPriceByGoods(ctx context.Context, req *GetPriceReq, strategy string) int64 {
// 策略
var strategyObj GoodsStrategyPrice

// 去除空格、转小写
strategy = strings.ToLower(strings.TrimSpace(strategy))

switch strategy {
case "steam":
strategyObj = NewGoodsSteamPrice()
case "buff":
strategyObj = NewGoodsBuffPrice()
case "igxe":
strategyObj = NewGoodsIgxePrice()
default: // 默认策略
strategyObj = NewGoodsSteamPrice()
}

obj := &GoodsStrategy{}
obj.SetStrategy(strategyObj)

return obj.GetPrice(ctx, req)
}

// GoodsSteamPrice Steam定价
type GoodsSteamPrice struct{}

// NewGoodsSteamPrice 定价策略:Steam定价
func NewGoodsSteamPrice() *GoodsSteamPrice {
return &GoodsSteamPrice{}
}

// GetPrice 根据饰品ID获取系统自定义价格:Steam定价
func (p *GoodsSteamPrice) GetPrice(ctx context.Context, req *GetPriceReq) int64 {
return 100
}

// GoodsBuffPrice Buff定价
type GoodsBuffPrice struct{}

// NewGoodsBuffPrice 定价策略: Buff定价
func NewGoodsBuffPrice() *GoodsBuffPrice {
return &GoodsBuffPrice{}
}

// GetPrice 根据饰品ID获取系统自定义价格:Buff定价
func (p *GoodsBuffPrice) GetPrice(ctx context.Context, req *GetPriceReq) int64 {
return 200
}

// GoodsIgxePrice Igxe定价
type GoodsIgxePrice struct{}

// NewGoodsIgxePrice 定价策略: Igxe定价
func NewGoodsIgxePrice() *GoodsIgxePrice {
return &GoodsIgxePrice{}
}

// GetPrice 根据饰品ID获取系统自定义价格:Igxe定价
func (p *GoodsIgxePrice) GetPrice(ctx context.Context, req *GetPriceReq) int64 {
return 300
}

什么是策略模式

这段代码展示了策略模式(Strategy Pattern)的设计思想。策略模式是一种行为设计模式,它使你能在运行时改变对象的行为。

以下是策略模式的关键组件在代码中的体现:

  1. 策略接口 (GoodsStrategyPrice): 定义了所有支持的策略或行为的公共接口。在这个例子中,接口声明了一个方法 GetPrice,用于根据饰品ID获取系统自定义价格。
  2. 具体策略类 (GoodsSteamPrice, GoodsBuffPrice, GoodsIgxePrice): 这些类实现了GoodsStrategyPrice接口,每种都代表了一种具体的定价策略。例如,GoodsSteamPrice 类的 GetPrice 方法返回100,模拟了Steam平台的定价规则。
  3. 上下文类 (GoodsStrategy): 此类用来封装使用哪种策略的上下文,并且提供了设置策略(SetStrategy)和执行策略(GetPrice)的方法。它持有一个GoodsStrategyPrice类型的策略对象,使得在运行时可以动态切换策略。
  4. 策略选择函数 (GetPriceByGoods): 这个函数根据传入的策略字符串参数,选择并实例化相应的策略对象,然后通过GoodsStrategy来执行选定的策略,获取价格。

通过这种方式,该设计允许程序在运行时选择不同的算法或策略来处理同一类型的问题,而无需知道具体策略是如何实现的,提高了代码的灵活性和可扩展性。如果未来需要添加新的定价策略,只需实现GoodsStrategyPrice接口并相应地修改策略选择逻辑即可。

main.go

package main

import (
"fmt"
"strategy/price"
)

func main() {

strategy := "steam"

priceData := price.GetPriceByGoods(nil, &price.GetPriceReq{GoodsId: 1}, strategy)

fmt.Printf("定价:%d\n", priceData)
}

输出结果:






我为人人,人人为我,美美与共,天下大同。