Golang如何使用策略模式管理算法_Golang 策略模式算法管理实践-Golang-PHP中文网

Golang如何使用策略模式管理算法_Golang 策略模式算法管理实践

P粉602998670

发布： 2025-11-23 17:17:02

原创

479人浏览过

策略模式通过接口定义统一行为，如SortStrategy接口声明Sort方法；2. 具体策略如BubbleSort和QuickSort实现该接口；3. 不同算法可动态切换，实现解耦。

golang如何使用策略模式管理算法_golang 策略模式算法管理实践

在 Go 语言中，策略模式是一种行为设计模式，适用于需要在运行时动态切换算法或行为的场景。通过将算法封装为独立的类型，并利用接口进行调用，可以实现算法与使用逻辑的解耦，提升代码的可维护性和扩展性。

定义策略接口

策略模式的核心是定义一个统一的接口，用于声明所有支持的算法共有的操作。每个具体算法都实现这个接口。

例如，假设我们需要对数据进行不同方式的排序：

type SortStrategy interface {
    Sort(data []int) []int
}

登录后复制

这个接口只有一个方法 Sort，任何实现了该方法的类型都可以作为排序策略被使用。

立即学习“go语言免费学习笔记（深入）”；

实现具体策略

接下来编写具体的算法实现。比如实现冒泡排序和快速排序两种策略：

type BubbleSort struct{}

func (b *BubbleSort) Sort(data []int) []int {
    sorted := make([]int, len(data))
    copy(sorted, data)
    n := len(sorted)
    for i := 0; i < n-1; i++ {
        for j := 0; j < n-i-1; j++ {
            if sorted[j] > sorted[j+1] {
                sorted[j], sorted[j+1] = sorted[j+1], sorted[j]
            }
        }
    }
    return sorted
}

type QuickSort struct{}

func (q *QuickSort) Sort(data []int) []int {
    if len(data) <= 1 {
        return data
    }
    pivot := data[0]
    var less, greater []int
    for _, v := range data[1:] {
        if v <= pivot {
            less = append(less, v)
        } else {
            greater = append(greater, v)
        }
    }
    result := append(QuickSort{}.Sort(less), pivot)
    result = append(result, QuickSort{}.Sort(greater)...)
    return result
}

登录后复制

每种排序算法都独立实现 SortStrategy 接口，便于替换和测试。

Supercreator

AI视频创作编辑器，几分钟内从构思到创作。

查看详情

上下文管理策略选择

创建一个上下文结构体来持有当前策略，并提供切换和执行的方法：

type Sorter struct {
    strategy SortStrategy
}

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

func (s *Sorter) ExecuteSort(data []int) []int {
    if s.strategy == nil {
        panic("no strategy set")
    }
    return s.strategy.Sort(data)
}

登录后复制

这样可以在运行时根据需求更换策略：

data := []int{5, 2, 9, 1, 5}

sorter := &Sorter{}
sorter.SetStrategy(&BubbleSort{})
fmt.Println("Bubble Sorted:", sorter.ExecuteSort(data))

sorter.SetStrategy(&QuickSort{})
fmt.Println("Quick Sorted:", sorter.ExecuteSort(data))

登录后复制