Go singleflight使用以及原理
這個(gè)東西很重要，可以經(jīng)常用在項(xiàng)目當(dāng)中，所以我們單獨(dú)拿出來(lái)進(jìn)行講解。
在使用它之前我們需要導(dǎo)包：
go get golang.org/x/sync/singleflight
golang/sync/singleflight.Group 是 Go 語(yǔ)言擴(kuò)展包中提供了另一種同步原語(yǔ)，它能夠在一個(gè)服務(wù)中抑制對(duì)下游的多次重復(fù)請(qǐng)求。一個(gè)比較常見(jiàn)的使用場(chǎng)景是：我們?cè)谑褂?Redis 對(duì)數(shù)據(jù)庫(kù)中的數(shù)據(jù)進(jìn)行緩存，發(fā)生緩存擊穿時(shí)，大量的流量都會(huì)打到數(shù)據(jù)庫(kù)上進(jìn)而影響服務(wù)的尾延時(shí)。
但是 golang/sync/singleflight.Group 能有效地解決這個(gè)問(wèn)題，它能夠限制對(duì)同一個(gè)鍵值對(duì)的多次重復(fù)請(qǐng)求，減少對(duì)下游的瞬時(shí)流量。
使用方法
singleflight類(lèi)的使用方法就新建一個(gè)singleflight.Group，使用其方法Do或者DoChan來(lái)包裝方法，被包裝的方法在對(duì)于同一個(gè)key，只會(huì)有一個(gè)協(xié)程執(zhí)行，其他協(xié)程等待那個(gè)協(xié)程執(zhí)行結(jié)束后，拿到同樣的結(jié)果。
Group結(jié)構(gòu)體
代表一類(lèi)工作，同一個(gè)group中，同樣的key同時(shí)只能被執(zhí)行一次
Do方法
func (g *Group) Do(key string, fn func() (interface{<!--{cke_protected}{C}%3C!%2D%2D%20%2D%2D%3E-->}, error)) (v interface{<!--{cke_protected}{C}%3C!%2D%2D%20%2D%2D%3E-->}, err error, shared bool)key：同一個(gè)key，同時(shí)只有一個(gè)協(xié)程執(zhí)行
fn：被包裝的函數(shù)
v：返回值，即執(zhí)行結(jié)果。其他等待的協(xié)程都會(huì)拿到
shared：表示是否由其他協(xié)程得到了這個(gè)結(jié)果v
DoChan方法
func (g *Group) DoChan(key string, fn func() (interface{<!--{cke_protected}{C}%3C!%2D%2D%20%2D%2D%3E-->}, error)) <-chan Result和Do差不多其實(shí)，因此我們就只講解Do的實(shí)際應(yīng)用場(chǎng)景了。
具體應(yīng)用場(chǎng)景
var singleSetCache singleflight.Group
func GetAndSetCache(r *http.Request, cacheKey string) (string, error) {
log.Printf("request %s start to get and set cache...", r.URL)
value, err, _ := singleSetCache.Do(cacheKey, func() (interface{}, error) {
log.Printf("request %s is getting cache...", r.URL)
time.Sleep(3 * time.Second)
log.Printf("request %s get cache success!", r.URL)
return cacheKey, nil
})
return value.(string), err
}
func main() {
r := gin.Default()
r.GET("/sekill/:id", func(context *gin.Context) {
ID := context.Param("id")
cache, err := GetAndSetCache(context.Request, ID)
if err != nil {
log.Println(err)
}
log.Printf("request %s get value: %v", context.Request.URL, cache)
})
r.Run()
}來(lái)看一下執(zhí)行結(jié)果：
2022/12/29 16:21:18 request /sekill/5 start to get and set cache...
2022/12/29 16:21:18 request /sekill/5 is getting cache...
2022/12/29 16:21:18 request /sekill/9 start to get and set cache...
2022/12/29 16:21:18 request /sekill/9 is getting cache...
2022/12/29 16:21:18 request /sekill/9 start to get and set cache...
2022/12/29 16:21:18 request /sekill/5 start to get and set cache...
2022/12/29 16:21:18 request /sekill/5 start to get and set cache...
2022/12/29 16:21:18 request /sekill/9 start to get and set cache...
2022/12/29 16:21:18 request /sekill/9 start to get and set cache...
2022/12/29 16:21:18 request /sekill/5 start to get and set cache...
2022/12/29 16:21:19 request /sekill/9 start to get and set cache...
2022/12/29 16:21:19 request /sekill/5 start to get and set cache...
2022/12/29 16:21:21 request /sekill/9 get cache success!
2022/12/29 16:21:21 request /sekill/5 get cache success!
2022/12/29 16:21:21 request /sekill/5 get value: 5
2022/12/29 16:21:21 request /sekill/5 get value: 5
[GIN] 2022/12/29 - 16:21:21 | 200 | 3.0106529s | 127.0.0.1 | GET "/sekill/5"
2022/12/29 16:21:21 request /sekill/9 get value: 9
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.8090881s | 127.0.0.1 | GET "/sekill/5"
2022/12/29 16:21:21 request /sekill/9 get value: 9
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.2166003s | 127.0.0.1 | GET "/sekill/9"
2022/12/29 16:21:21 request /sekill/9 get value: 9
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.6064069s | 127.0.0.1 | GET "/sekill/9"
2022/12/29 16:21:21 request /sekill/9 get value: 9
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.4178652s | 127.0.0.1 | GET "/sekill/9"
2022/12/29 16:21:21 request /sekill/9 get value: 9
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.8101267s | 127.0.0.1 | GET "/sekill/9"
2022/12/29 16:21:21 request /sekill/5 get value: 5
[GIN] 2022/12/29 - 16:21:21 | 200 | 3.0116892s | 127.0.0.1 | GET "/sekill/9"
2022/12/29 16:21:21 request /sekill/5 get value: 5
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.6074537s | 127.0.0.1 | GET "/sekill/5"
2022/12/29 16:21:21 request /sekill/5 get value: 5
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.4076473s | 127.0.0.1 | GET "/sekill/5"
[GIN] 2022/12/29 - 16:21:21 | 200 | 2.218686s | 127.0.0.1 | GET "/sekill/5"
可以看到確實(shí)只有一個(gè)協(xié)程執(zhí)行了被包裝的函數(shù)，并且其他協(xié)程都拿到了結(jié)果。
接下來(lái)我們來(lái)看一下它的原理吧！
原理
首先來(lái)看一下Group結(jié)構(gòu)體：
type Group struct {
mu sync.Mutex // 鎖保證并發(fā)安全
m map[string]*call //保存key對(duì)應(yīng)的函數(shù)執(zhí)行過(guò)程和結(jié)果的變量。
}
然后我們來(lái)看一下call結(jié)構(gòu)體：
type call struct {
wg sync.WaitGroup //用WaitGroup實(shí)現(xiàn)只有一個(gè)協(xié)程執(zhí)行函數(shù)
val interface{} //函數(shù)執(zhí)行結(jié)果
err error
forgotten bool
dups int //含義是duplications，即同時(shí)執(zhí)行同一個(gè)key的協(xié)程數(shù)量
chans []chan<- Result
}
然后我們來(lái)看一下Do方法：
func (g *Group) Do(key string, fn func() (interface{}, error)) (v interface{}, err error, shared bool) {
// 寫(xiě)Group的m字段時(shí)，加鎖保證寫(xiě)安全
g.mu.Lock()
if g.m == nil {
g.m = make(map[string]*call)
}
if c, ok := g.m[key]; ok {
// 如果key已經(jīng)存在，說(shuō)明已經(jīng)由協(xié)程在執(zhí)行，則dups++并等待其執(zhí)行結(jié)果，執(zhí)行結(jié)果保存在對(duì)應(yīng)的call的val字段里
c.dups++
g.mu.Unlock()
c.wg.Wait()
if e, ok := c.err.(*panicError); ok {
panic(e)
} else if c.err == errGoexit {
runtime.Goexit()
}
return c.val, c.err, true
}
// 如果key不存在，則新建一個(gè)call，并使用WaitGroup來(lái)阻塞其他協(xié)程，同時(shí)在m字段里寫(xiě)入key和對(duì)應(yīng)的call
c := new(call)
c.wg.Add(1)
g.m[key] = c
g.mu.Unlock()
g.doCall(c, key, fn) // 進(jìn)來(lái)的第一個(gè)協(xié)程就來(lái)執(zhí)行這個(gè)函數(shù)
return c.val, c.err, c.dups > 0
}
然后我們來(lái)分析一下doCall函數(shù)：
func (g *Group) doCall(c *call, key string, fn func() (interface{}, error)) {
c.val, c.err = fn()
c.wg.Done()
g.mu.Lock()
delete(g.m, key)
for _, ch := range c.chans {
ch <- Result{c.val, c.err, c.dups > 0}
}
g.mu.Unlock()
}
- 運(yùn)行傳入的函數(shù)
fn，該函數(shù)的返回值會(huì)賦值給c.val和c.err； - 調(diào)用
sync.WaitGroup.Done方法通知所有等待結(jié)果的Goroutine— 當(dāng)前函數(shù)已經(jīng)執(zhí)行完成，可以從call結(jié)構(gòu)體中取出返回值并返回了； - 獲取持有的互斥鎖并通過(guò)管道將信息同步給使用
golang/sync/singleflight.Group.DoChan方法的Goroutine；
問(wèn)題分析
分析了源碼之后，我們得出了一個(gè)結(jié)論，這個(gè)東西是用阻塞來(lái)實(shí)現(xiàn)的，這就引發(fā)了一個(gè)問(wèn)題：如果我們處理的那個(gè)請(qǐng)求剛好遇到問(wèn)題了，那么后面的所有請(qǐng)求都會(huì)被阻塞，也就是，我們應(yīng)該加上適合的超時(shí)控制，如果在一定時(shí)間內(nèi)，沒(méi)有獲得結(jié)果，那么就當(dāng)作超時(shí)處理。
于是這個(gè)適合我們應(yīng)該使用DoChan()。兩者實(shí)現(xiàn)上完全一樣，不同的是， DoChan() 通過(guò) channel 返回結(jié)果。因此可以使用 select 語(yǔ)句實(shí)現(xiàn)超時(shí)控制。
var singleSetCache singleflight.Group
func GetAndSetCache(r *http.Request, cacheKey string) (string, error) {
log.Printf("request %s start to get and set cache...", r.URL)
retChan := singleSetCache.DoChan(cacheKey, func() (interface{}, error) {
log.Printf("request %s is getting cache...", r.URL)
time.Sleep(3 * time.Second)
log.Printf("request %s get cache success!", r.URL)
return cacheKey, nil
})
var ret singleflight.Result
timeout := time.After(2 * time.Second)
select {
case <-timeout:
log.Println("time out!")
return "", errors.New("time out")
case ret = <-retChan: // 從chan中獲取結(jié)果
return ret.Val.(string), ret.Err
}
}
func main() {
r := gin.Default()
r.GET("/sekill/:id", func(context *gin.Context) {
ID := context.Param("id")
cache, err := GetAndSetCache(context.Request, ID)
if err != nil {
log.Println(err)
}
log.Printf("request %s get value: %v", context.Request.URL, cache)
})
r.Run()
}
補(bǔ)充
這里其實(shí)還有一個(gè)Forget方法，它可以在映射表中刪除某個(gè)鍵，接下來(lái)對(duì)鍵的調(diào)用就不會(huì)等待前面的函數(shù)返回了。
總結(jié)
當(dāng)然，如果單次的失敗無(wú)法容忍，在高并發(fā)的場(chǎng)景下更好的處理方案是：
放棄使用同步請(qǐng)求，犧牲數(shù)據(jù)更新的實(shí)時(shí)性
“緩存” 存儲(chǔ)準(zhǔn)實(shí)時(shí)的數(shù)據(jù) + “異步更新” 數(shù)據(jù)到緩存
到此這篇關(guān)于Go singleflight使用以及原理的文章就介紹到這了,更多相關(guān)Go singleflight內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
GoLang OS包以及File類(lèi)型詳細(xì)講解
go中對(duì)文件和目錄的操作主要集中在os包中，下面對(duì)go中用到的對(duì)文件和目錄的操作，做一個(gè)總結(jié)筆記。在go中的文件和目錄涉及到兩種類(lèi)型，一個(gè)是type File struct，另一個(gè)是type Fileinfo interface2023-03-03
這篇文章主要介紹了Golang通脈之流程控制，流程控制是每種編程語(yǔ)言控制邏輯走向和執(zhí)行次序的重要部分，Go語(yǔ)言中最常用的流程控制有if和for，而switch和goto主要是為了簡(jiǎn)化代碼，下面文章將詳細(xì)介紹改該內(nèi)容,需要的朋友可以參考一下2021-10-10
Go語(yǔ)言開(kāi)發(fā)發(fā)送Get和Post請(qǐng)求的示例
這篇文章主要介紹了Go語(yǔ)言開(kāi)發(fā)發(fā)送Get和Post請(qǐng)求的示例,小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-07-07
這篇文章主要介紹了golang指數(shù)運(yùn)算操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-12-12
golang根據(jù)生日計(jì)算星座和屬相實(shí)例
這篇文章主要為大家介紹了golang根據(jù)生日計(jì)算星座和屬相的示例代碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
這篇文章主要為大家介紹了Go語(yǔ)言實(shí)現(xiàn)二分查找方法示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-12-12
這篇文章給大家介紹了Golang中如何使用errors返回調(diào)用堆棧信息,文章通過(guò)代碼示例給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下2023-12-12
go后端模版語(yǔ)法是通過(guò) {{}} ,vue也是通過(guò)雙花括號(hào)來(lái)渲染的,如果使用go渲染vue的html頁(yè)面的時(shí)候就會(huì)報(bào)錯(cuò),本文主要介紹了Go模板后端渲染時(shí)vue單頁(yè)面沖突,感興趣的可以了解一下2024-01-01
三種Golang數(shù)組拷貝的實(shí)現(xiàn)方式與性能分析
在?Golang?中，有多種方式可以進(jìn)行數(shù)組的拷貝，本文將對(duì)其中的三種方式進(jìn)行性能分析，并比較它們的優(yōu)缺點(diǎn)，感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-06-06
Go語(yǔ)言中break label與goto label的區(qū)別
這篇文章主要介紹了Go語(yǔ)言中break label與goto label的區(qū)別，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04