使用Go语言实现策略模式

运维开发王义杰

发布于 2023-08-10 19:30:00

3310

发布于 2023-08-10 19:30:00

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

策略模式是一种行为设计模式，允许我们在运行时改变对象的行为。通过这种方式，我们可以选择算法并改变对象的行为。在这篇文章中，我们将探讨如何在Go语言中实现策略模式，并通过一个排序算法选择器实例来进行说明。

定义策略接口

首先，我们定义策略接口。

// Strategy 是策略接口
type Strategy interface {
    Sort(data []int) []int
}

实现具体的策略

然后，我们实现具体的策略，例如冒泡排序策略和快速排序策略。

type BubbleSort struct {
}

func (bs *BubbleSort) Sort(data []int) []int {
    // 实现冒泡排序
    // 这里仅作示意，省略具体实现
}

type QuickSort struct {
}

func (qs *QuickSort) Sort(data []int) []int {
    // 实现快速排序
    // 这里仅作示意，省略具体实现
}

实现上下文

接下来，我们实现上下文，也就是执行策略的环境。

type Sorter struct {
    strategy Strategy
}

func (s *Sorter) SetStrategy(strategy Strategy) {
    s.strategy = strategy
}

func (s *Sorter) Sort(data []int) []int {
    return s.strategy.Sort(data)
}

客户端代码

最后，我们的客户端代码看起来像这样：

func main() {
    sorter := &Sorter{}
    data := []int{5, 2, 3, 1, 4}

    sorter.SetStrategy(&BubbleSort{})
    sortedData := sorter.Sort(data)
    fmt.Println(sortedData)

    sorter.SetStrategy(&QuickSort{})
    sortedData = sorter.Sort(data)
    fmt.Println(sortedData)
}

总结

策略模式为我们提供了一种方式，可以根据情况选择不同的算法或策略。希望本文能够帮助你理解策略模式，并在实际开发中找到其应用场景。如果你觉得本文有价值，欢迎分享给你的朋友。

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

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