Go設(shè)計模式之策略模式講解和代碼示例

更新時間：2023年08月22日 09:31:34 作者：demo007x

策略是一種行為設(shè)計模式,?它將一組行為轉(zhuǎn)換為對象,?并使其在原始上下文對象內(nèi)部能夠相互替換,本文就將通過代碼示例給大家詳細(xì)的介紹一下Go的策略模式,需要的朋友可以參考下

Go 策略模式講解和代碼示例

策略是一種行為設(shè)計模式，它將一組行為轉(zhuǎn)換為對象，并使其在原始上下文對象內(nèi)部能夠相互替換。

原始對象被稱為上下文，它包含指向策略對象的引用并將執(zhí)行行為的任務(wù)分派給策略對象。為了改變上下文完成其工作的方式，其他對象可以使用另一個對象來替換當(dāng)前鏈接的策略對象。

概念示例

思考一下構(gòu)建內(nèi)存緩存的情形。由于處在內(nèi)存中，故其大小會存在限制。在達(dá)到其上限后，一些條目就必須被移除以留出空間。此類操作可通過多種算法進(jìn)行實現(xiàn)。一些流行的算法有：

最少最近使用（LRU）：移除最近使用最少的一條條目。
先進(jìn)先出（FIFO）：移除最早創(chuàng)建的條目。
最少使用（LFU）：移除使用頻率最低一條條目。

問題在于如何將我們的緩存類與這些算法解耦，以便在運行時更改算法。此外，在添加新算法時，緩存類不應(yīng)改變。

這就是策略模式發(fā)揮作用的場景。可創(chuàng)建一系列的算法，每個算法都有自己的類。這些類中的每一個都遵循相同的接口，這使得系列算法之間可以互換。假設(shè)通用接口名稱為 evictionAlgo移除算法。

現(xiàn)在，我們的主要緩存類將嵌入至 evictionAlgo接口中。緩存類會將全部類型的移除算法委派給 evictionAlgo接口，而不是自行實現(xiàn)。鑒于 evictionAlgo是一個接口，我們可在運行時將算法更改為 LRU、 FIFO 或者 LFU，而不需要對緩存類做出任何更改。

evictionAlgo.go: 策略接口

package main
// 策略接口
type EvictionAlgo interface {
	evict(c *Cache)
}

fifo.go:具體策略

package main
import "fmt"
type Fifo struct{}
func (l *Fifo) evict(c *Cache) {
	fmt.Println("Evicting by fifo strtegy")
}

lru.go:具體策略

package main
import "fmt"
type Lru struct{}
func (l *Lru) evict(c *Cache) {
	fmt.Println("Evicting by lru strtegy")
}

lfu.go:具體策略

package main
import "fmt"
type Lfu struct{}
func (l *Lfu) evict(c *Cache) {
	fmt.Println("Evicting by lfu strtegy")
}

cache.go:背景

package main
type Cache struct {
	storage      map[string]string
	evictionAlgo EvictionAlgo
	capacity     int
	macCapacity  int
}
// 初始化的時候?qū)⒉呗宰⑷氲?cache 中
func initCache(e EvictionAlgo) *Cache {
	storage := make(map[string]string)
	return &Cache{
		storage:      storage,
		evictionAlgo: e,
		capacity:     0,
		macCapacity:  2,
	}
}
// 動態(tài)修改策略
func (c *Cache) setEvictionAlgo(e EvictionAlgo) {
	c.evictionAlgo = e
}
func (c *Cache) add(key, value string) {
	// 如果緩存中的容量等于了最大容量，則需要執(zhí)行策略來移除 s
	if c.capacity == c.macCapacity {
		c.evict()
	}
	c.capacity++
	c.storage[key] = value
}
func (c *Cache) evict() {
	c.evictionAlgo.evict(c)
	c.capacity--
}
func (c *Cache) get(key string) {
	delete(c.storage, key)
}

main.go:客戶端代碼

package main
func main() {
	lfu := &Lfu{}
	cache := initCache(lfu)
	cache.add("a", "1")
	cache.add("b", "2")
	cache.add("c", "3")
	lru := &Lru{}
	cache.setEvictionAlgo(lru)
	cache.add("d", "4")
	fifo := &Fifo{}
	cache.setEvictionAlgo(fifo)
	cache.add("e", "5")
}