Go设计模式(20)-策略模式

策略模式可能是很多同学学习到的第一种模式。这个模式确实适合作为开篇模式来讲，主要原因在于该模式简单、纯粹、没有太多技巧，但是很好的表达出了设计模式的理念，让读者能够直观的感受到设计模式带来的好处。

策略模式核心在于利用多态性，这是目前主流的面向对象语言都支持的功能。

UML类图位置：https://www.processon.com/view/link/60d29bf3e401fd49502afd25

本文代码链接为：https://github.com/shidawuhen/asap/blob/master/controller/design/20strategy.go

1.定义

1.1策略模式

策略模式：它定义了算法家族，分别封装起来，让它们之间可以互相替换，此模式让算法的变化，不会影响到使用算法的客户。

UML：

1.2分析

单看策略模式，主要利用多态性。但策略模式往往不单独使用，它会和Go设计模式(7)-工厂模式配合使用。此时策略模式便可解耦策略的定义、创建、使用。

策略模式加工厂模式，能够达到去除if-else和switch的效果，主要靠工厂模式加持，借助于“查表法”找到指定策略进行使用。

2.应用场景

策略模式的应用场景还是蛮广泛的，处理同一类问题如果有多种算法，就可以使用策略模式。如根据不同活动计算优惠价格、根据商品不同类型计算税率等。

最近有一个实际业务场景可以使用策略模式，计算跨境商品税费。

跨境商品的税费和两方面有关，一是商品是否含税，二是商品类型，不同类型对应税率不一样，如常规商品和酒类商品税率不一样、税的计算方式也不一样。

所以只要知道商品是否含税、商品类型就能找到对应的计算方案。如果我们使用if-else来写，会不优雅，因为商品类型比较多，而且计算逻辑也相对复杂，所以我们可以利用查表法进行优化。

税费计算接口被交易侧调用，交易调用的时候会传商品是否含税，商品类型由项目组自行维护。那我们来看一下具体实现。

3.代码实现

package main

import "fmt"

const (
   Common = "COMMON"
   Win    = "WIN"
)

/**
 * @Author: Jason Pang
 * @Description: 根据hscode获取商品类型
 * @param hscode
 * @return string
 */
func getProductType(hscode string) string {
   if hscode == "11" {
      return Common
   } else {
      return Win
   }
}

/**
 * @Author: Jason Pang
 * @Description: 税费计算函数，金额都为分
 * @param price
 * @param qty
 * @return taxPrice
 */
type TaxComputeFunc func(price int64, qty int64) (taxPrice int64)

/**
 * @Author: Jason Pang
 * @Description: 税费计算策略存储处
   0为不含税 1为含税
*/
var TaxComputeFuncMap = map[int]map[string]TaxComputeFunc{
   0: map[string]TaxComputeFunc{
      Common: common,
      Win:    win,
   },
   1: map[string]TaxComputeFunc{
      Common: common,
      Win:    win,
   },
}

/**
 * @Author: Jason Pang
 * @Description: 计算普通商品税费
 * @param price
 * @param qty
 * @return taxPrice
 */
func common(price int64, qty int64) (taxPrice int64) {
   radio := 0.1
   fmt.Println("计算普通商品税费")
   return int64(float64(price*qty) * radio)
}

/**
 * @Author: Jason Pang
 * @Description: 计算酒类税费
 * @param price
 * @param qty
 * @return taxPrice
 */
func win(price int64, qty int64) (taxPrice int64) {
   radio := 0.2
   fmt.Println("计算普酒类税费")
   return int64(float64(price*qty) * radio)
}

/**
 * @Author: Jason Pang
 * @Description: 计算税费
 * @param withTax
 * @param productType
 * @param price
 * @param qty
 */
func ComputeTaxPrice(withTax int, productType string, price int64, qty int64) {
   if taxFunc, ok := TaxComputeFuncMap[withTax][productType]; ok {
      taxPrice := taxFunc(price, qty)
      fmt.Println("税费为", taxPrice)
   } else {
      fmt.Println("输入有误，无法计算")
   }
}

func main() {
   //获取商品是否含税、商品价格、商品数量、商品类型
   withTax := 0
   var price, qty int64 = 10000, 3
   productType := getProductType("11")
   //计算税费
   ComputeTaxPrice(withTax, productType, price, qty)
}

输出：

➜ myproject go run main.go

计算普通商品税费

税费为 3000

这么写有哪些好处呢？首先代码很精简，没有一堆判断；其次无论是增加策略还是修改策略，都不会影响主框架，即main\ComputeTaxPrice，只需在TaxComputeFuncMap添加新的策略即可，很好的做到了对扩展开放，虽然对TaxComputeFuncMap有一定更改，但变动不大，能够接受。

如果大家看过我以前写的Go设计模式(18)-观察者模式，会感觉两个实现比较像。两者确实比较像，但是有一个核心地方不同，观察者模式是调用所有策略，策略模式准确找到合适的策略进行调用。

总结

策略模式很好的体现了开闭原则，也说明即使是很小的优化设计，也能给项目开发带来巨大的便利。当然，这种便利会在维护的时候得到充分体现，毕竟谁维护谁知道。

最后

大家如果喜欢我的文章，可以关注我的公众号（程序员麻辣烫）

我的个人博客为：https://shidawuhen.github.io/

往期文章回顾：

1. 设计模式

2. 招聘

3. 思考

4. 存储

5. 算法系列

6. 读书笔记

7. 小工具

8. 架构

9. 网络

10. Go语言