golang?防緩存擊穿singleflight的實現(xiàn)
一、什么是緩存擊穿
當一個key是熱點key時,一般會做緩存來抗大量并發(fā),但當緩存失效的一瞬間,這些大量的并發(fā)請求會擊穿緩存,直接請求數據庫
為了避免緩存擊穿,一種解決方法可以設置緩存永不過期,另一種可以使用golang的包 singleflight golang.org/x/sync/singleflight
二、原理
多個并發(fā)請求對一個失效key進行數據獲取時,只會有其中一個去直接獲取數據,其它請求會阻塞等待第一個請求返回給它們結果
三、實現(xiàn)
package singleflight
import (
"sync"
)
var WaitCount int
var DirectCount int
type Caller struct {
val interface{}
err error
wg sync.WaitGroup
}
type Group struct {
mu sync.RWMutex
m map[string]*Caller
}
func (g *Group) Do(key string, fn func() (interface{}, error)) (interface{}, error) {
g.mu.Lock()
if g.m == nil {
g.m = make(map[string]*Caller)
}
c, ok := g.m[key]
if ok {
//阻塞等待其它已經執(zhí)行此操作的返回結果
g.mu.Unlock()
c.wg.Wait()
WaitCount++
return c.val, c.err
}
//直接請求獲取數據
c = &Caller{}
g.m[key] = c
c.wg.Add(1)
g.mu.Unlock()
c.val, c.err = fn()
c.wg.Done()
g.mu.Lock()
delete(g.m, key)
g.mu.Unlock()
DirectCount++
return c.val, c.err
}測試:
func TestGroup_Do(t *testing.T) {
sg := &Group{}
wg := sync.WaitGroup{}
for i := 0; i < 10000; i++ {
fn := func() (interface{}, error) {
return i, nil
}
wg.Add(1)
go func() {
defer wg.Done()
got, err := sg.Do("test-key", fn)
_, _ = got, err
//t.Log("got:", i)
}()
}
wg.Wait()
fmt.Println("waitCount:", WaitCount)
fmt.Println("DirectCount:", DirectCount)
}輸出:
waitCount: 8323
DirectCount: 1401
到此這篇關于golang 防緩存擊穿singleflight的實現(xiàn)的文章就介紹到這了,更多相關golang 防緩存擊穿singleflight 內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
詳解golang consul-grpc 服務注冊與發(fā)現(xiàn)
這篇文章主要介紹了詳解golang consul-grpc 服務注冊與發(fā)現(xiàn),小編覺得挺不錯的,現(xiàn)在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧2018-06-06
go select編譯期的優(yōu)化處理邏輯使用場景分析
select 是 Go 中的一個控制結構,類似于用于通信的 switch 語句。每個 case 必須是一個通信操作,要么是發(fā)送要么是接收。接下來通過本文給大家介紹go select編譯期的優(yōu)化處理邏輯使用場景分析,感興趣的朋友一起看看吧2021-06-06

