策略模式通过接口定义算法族,实现算法与客户端解耦。在golang中,定义SortStrategy接口及BubbleSort、QuickSort等具体策略,通过Sorter上下文动态切换排序算法,利用接口多态性在运行时注入不同策略,使算法可扩展且易于维护。
策略模式是一种行为设计模式,它让你定义一系列算法或行为,并将每种行为分别放入独立的类中,使得它们可以互相替换。在golang中,虽然没有继承的概念,但通过接口和组合可以非常自然地实现策略模式。
定义策略接口
首先,定义一个公共接口来表示策略的行为。比如我们有一个排序需求,不同的排序算法可以作为不同策略:
type SortStrategy interface { Sort([]int) []int }
实现具体策略
type BubbleSort struct{} <p>func (b *BubbleSort) Sort(data []int) []int { result := make([]int, len(data)) copy(result, data) n := len(result) for i := 0; i < n-1; i++ { for j := 0; j < n-i-1; j++ { if result[j] > result[j+1] { result[j], result[j+1] = result[j+1], result[j] } } } return result }</p><p>type QuickSort struct{}</p><p><span>立即学习</span>“<a href="https://pan.quark.cn/s/00968c3c2c15" style="text-decoration: underline !important; color: blue; font-weight: bolder;" rel="nofollow" target="_blank">go语言免费学习笔记(深入)</a>”;</p> <div class="aritcle_card"> <a class="aritcle_card_img" href="/ai/%E5%A6%82%E7%9F%A5ai%E7%AC%94%E8%AE%B0"> <img src="https://img.php.cn/upload/ai_manual/000/000/000/175679994166405.png" alt="如知AI笔记"> </a> <div class="aritcle_card_info"> <a href="/ai/%E5%A6%82%E7%9F%A5ai%E7%AC%94%E8%AE%B0">如知AI笔记</a> <p>如知笔记——支持markdown的在线笔记,支持ai智能写作、AI搜索,支持DeepseekR1满血大模型</p> <div class=""> <img src="/static/images/card_xiazai.png" alt="如知AI笔记"> <span>27</span> </div> </div> <a href="/ai/%E5%A6%82%E7%9F%A5ai%E7%AC%94%E8%AE%B0" class="aritcle_card_btn"> <span>查看详情</span> <img src="/static/images/cardxiayige-3.png" alt="如知AI笔记"> </a> </div> <p>func (q *QuickSort) Sort(data []int) []int { result := make([]int, len(data)) copy(result, data) quickSortHelper(result, 0, len(result)-1) return result }</p><p>func quickSortHelper(arr []int, low, high int) { if low < high { pi := partition(arr, low, high) quickSortHelper(arr, low, pi-1) quickSortHelper(arr, pi+1, high) } }</p><p>func partition(arr []int, low, high int) int { pivot := arr[high] i := low - 1 for j := low; j < high; j++ { if arr[j] <= pivot { i++ arr[i], arr[j] = arr[j], arr[i] } } arr[i+1], arr[high] = arr[high], arr[i+1] return i + 1 }
使用上下文管理策略
创建一个上下文结构体,用于设置和执行当前策略:
type Sorter struct { strategy SortStrategy } <p>func (s *Sorter) SetStrategy(strategy SortStrategy) { s.strategy = strategy }</p><p>func (s *Sorter) Sort(data []int) []int { if s.strategy == nil { panic("未设置排序策略") } return s.strategy.Sort(data) }
这样就可以在运行时动态切换算法:
func main() { data := []int{64, 34, 25, 12, 22, 11, 90} <pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">sorter := &Sorter{} // 使用冒泡排序 sorter.SetStrategy(&BubbleSort{}) sorted1 := sorter.Sort(data) fmt.Println("冒泡排序结果:", sorted1) // 切换为快速排序 sorter.SetStrategy(&QuickSort{}) sorted2 := sorter.Sort(data) fmt.Println("快速排序结果:", sorted2)
}
策略模式的核心在于解耦算法与使用它的客户端。通过接口抽象行为,让调用方无需关心具体实现,只需关注行为契约。Golang的接口机制天然支持这种多态性,使策略模式实现简洁清晰。
基本上就这些。只要掌握接口定义、具体实现和运行时注入,就能灵活运用策略模式。