设计模式:策略模式

在策略模式中,我们需要创建一系列策略对象和一个能随策略对象改变而改变的Context对象,策略对象改变Context的执行方法。

仍以两个数字的加减乘除操作作为示例

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
package strategy
import "fmt"
// Context 类
type Context struct {
strategy Strategy
}
func NewContext(strategy Strategy) *Context {
return &Context{
strategy: strategy,
}
}
func (c *Context) GetResult(a float64, b float64) (float64, error) {
return c.strategy.GetResult(a, b)
}
// 策略接口
type Strategy interface {
GetResult(a float64, b float64) (float64, error)
}
// 以下类实现策略接口
// 加法
type Add struct{}
func (o *Add) GetResult(a float64, b float64) (float64, error) {
return a + b, nil
}
// 减法
type Sub struct{}
func (o *Sub) GetResult(a float64, b float64) (float64, error) {
return a - b, nil
}
// 乘法
type Mul struct{}
func (o *Mul) GetResult(a float64, b float64) (float64, error) {
return a * b, nil
}
// 除法
type Div struct{}
func (o *Div) GetResult(a float64, b float64) (float64, error) {
if b == 0 {
return 0, fmt.Errorf("除数不能为0")
}
return a / b, nil
}

源码以及测试源码下载地址:https://github.com/h1z3y3/big-talk-go-design-patterns/tree/master/01_strategy

坚持原创技术分享,您的支持将鼓励我继续创作!