一个使用sync.Once特性的缓存示例

直接上代码：

package main

// 一个使用sync.Once的缓存示例，它不会出现缓存击穿，还缺少缓存过期机制可进一步优化

import (

"log"

"sync"

"time"

)

type CacheEntry struct {

data []byte // 存放实际数据

err error

once *sync.Once

}

type OrderSever struct {

cache map[string]*CacheEntry

mutex sync.Mutex

}

// Query 查询缓存第一次会去后端获取因为加了锁其他并发线程只能等待

func (order *OrderSever) Query(key string) ([]byte, error) {

order.mutex.Lock()

if order.cache == nil {

order.cache = make(map[string]*CacheEntry) // 初始化缓存池

}

entry, ok := order.cache[key]

// 找不到就初始化一个

if !ok {

entry = &CacheEntry{

data: make([]byte, 0),

err: nil,

once: new(sync.Once), // 注意这里是new

}

order.cache[key] = entry

}

order.mutex.Unlock()

// 只执行一次

entry.once.Do(func() {

entry.data, entry.err = getOrder() // 填充缓存去后端获取数据如数据库等

})

// 数据被赋值了返回

return entry.data, entry.err

}

func getOrder() ([]byte, error) {

time.Sleep(50 * time.Millisecond)

log.Println("后端获取数据")

return []byte("hello world"), nil

}

func main() {

var order OrderSever

for i := 0; i < 10; i++ {

query, err := order.Query("test")

if err != nil {

return

}

log.Println(string(query))

}

转载请注明：轻风博客 » 一个使用sync.Once特性的缓存示例