今天看了一篇博客,使用策略加工厂模式消灭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)
}