go使用map加策略模式减少代码if else

今天看了一篇博客，使用策略加工厂模式消灭if else
博客地址https://juejin.im/post/5dad23685188251d2c4ea2b6#comment
为了巩固印象，使用了go实现了策略模式消灭if else

首先初始化工作，定义了一个接口和2个实现结构体，和Context上下文

type Operation interface {//定义接口
	count(a,b int)
}
type Add struct {//定义2个结构体实现接口
}
func(Add) count(a,b int){
	fmt.Println(a+b)
}
type Sub struct {
}
func(Sub) count(a,b int){
	fmt.Println(a-b)
}
type Context struct {
	Operation
}
func NewContext(operation Operation)*Context{//工厂模式，实现构造函数
	return &Context{
		operation,
	}
}
var m =make(map[int]Operation)//定义map为工厂

首先是没有使用策略模式的时候,需要使用if else来判断接下来的逻辑。

//假设现在有a,b2个参数，根据不同的状态执行不同的计算，1是加法 2是减法 state代表状态
func main() {
	state:=1
	a:=3
	b:=3
	if state==1{
		add:=Add{}
		add.count(a,b)
		
	}else {
		sub:=Sub{}
		sub.count(a,b)
		
	}
}

使用策略模式的代码

func init() { //初始化，创建策略类放入map中。
	a:=Add{}//这里有一个方法集的问题。由于是Add结构体实现了Operation接口
	m[1]=&a//所以这里a和*a(*a=&a)都是实现了Operation
	s:=Sub{}
	m[2]=&s
}

func main() {
	state:=1
	a:=3
	b:=3
	c:=NewContext(m[state]) //根据state从map中取对应的策略类
	c.count(a,b)
}