Go：使用策略模式

运维开发王义杰

发布于 2023-08-10 19:25:50

1300

发布于 2023-08-10 19:25:50

文章被收录于专栏：运维开发王义杰运维开发王义杰

策略模式是一种常见的行为设计模式，它能够在运行时改变对象的行为。这一设计模式的主要目的是定义一系列算法，并将每种算法封装起来，使得它们可以互相替换。

在这篇文章中，我们将探讨如何使用Go语言来实现策略模式，并通过一个实例进行演示。我们选择的实例是一个简单的数学问题：根据用户的选择，我们将实现一个计算器，支持加法、减法、乘法和除法。

定义策略接口

首先，我们需要定义一个策略接口，该接口将被所有具体的策略类实现。在这个例子中，我们的策略是进行数学运算。

type Strategy interface {
    DoOperation(num1 int, num2 int) int
}

创建具体策略类

然后，我们为每一种数学运算创建一个具体的策略类。这些类都将实现上述定义的Strategy接口。

type OperationAdd struct{}

func (o *OperationAdd) DoOperation(num1 int, num2 int) int {
    return num1 + num2
}

type OperationSubtract struct{}

func (s *OperationSubtract) DoOperation(num1 int, num2 int) int {
    return num1 - num2
}

type OperationMultiply struct{}

func (m *OperationMultiply) DoOperation(num1 int, num2 int) int {
    return num1 * num2
}

type OperationDivide struct{}

func (d *OperationDivide) DoOperation(num1 int, num2 int) int {
    if num2 != 0 {
        return num1 / num2
    }
    fmt.Println("Cannot divide by 0")
    return 0
}

创建上下文

接下来，我们创建一个上下文类。上下文使用了一些策略对象，该策略对象会改变上下文的执行算法。

type Context struct {
    strategy Strategy
}

func NewContext(strategy Strategy) *Context {
    return &Context{
        strategy: strategy,
    }
}

func (c *Context) ExecuteStrategy(num1 int, num2 int) int {
    return c.strategy.DoOperation(num1, num2)
}

客户端代码

最后，我们来看一下客户端代码是如何使用这些类的。用户可以根据他们的需要选择合适的策略。

func main() {
    context := NewContext(new(OperationAdd))
    fmt.Println("10 + 5 =", context.ExecuteStrategy(10, 5))

    context = NewContext(new(OperationSubtract))
    fmt.Println("10 - 5 =", context.ExecuteStrategy(10, 5))

    context = NewContext(new(OperationMultiply))
    fmt.Println("10 * 5 =", context.ExecuteStrategy(10, 5))

    context = NewContext(new(OperationDivide))
    fmt.Println("10 / 5 =", context.ExecuteStrategy(10, 5))
}

当运行上述代码时，它会根据选择的策略执行相应的操作。

总结

策略模式为我们提供了一种方式，可以将一系列算法封装起来，并根据需要进行更换。这种方式不仅使代码更加清晰，而且也让算法更易于复用。这对于需要处理不同种类的业务逻辑的大型系统来说，尤为重要。

当然，每种设计模式都有其适用场景，它们并非万能的。在使用设计模式时，我们应该根据实际的需求和场景来选择合适的模式，而不是强行套用。

希望本文能够帮助你理解策略模式，并能在实际的开发中找到其应用。如果你觉得本文对你有帮助，欢迎分享给你的朋友。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2023-08-02，如有侵权请联系 cloudcommunity@tencent.com 删除