
singleflight库的并发访问控制
并发场景下,多个请求同时访问同一个资源会造成资源的重复获取。为了解决这一问题,singleflight库提供了并发访问控制机制。
在给定的代码示例中,问题在于getdatafromcache函数总是返回errornotexist,导致singleflight库的do方法无法缓存从数据库获取的数据。因此,所有请求都直接访问了数据库,造成并发访问问题。
要解决这个问题,需要确保在第一个请求成功获取数据并写入缓存后,后续请求都能从缓存中读取数据。具体实现方法如下:
- 在getdatafromcache函数中,如果缓存中存在数据,则直接返回数据,而不返回errornotexist。
- 在getdata函数中,如果数据存在于缓存中,则直接从缓存中读取数据,而不调用do方法。
修改后的代码如下:
func getDataFromCache(key string) (string, error) {
// 检查缓存中是否存在数据
if data, ok := cache[key]; ok {
return data, nil
}
return "", errorNotExist
}
func getData(key string) (string, error) {
// 从缓存中获取数据
data, err := getDataFromCache(key)
if err == nil {
return data, nil
}
// 使用 singleflight 获取数据
v, err, _ := g.Do(key, func() (interface{}, error) {
return getDataFromDB(key)
})
if err != nil {
log.Println(err)
return "", err
}
data = v.(string)
// 将数据写入缓存
cache[key] = data
return data, nil
}通过这些修改,singleflight库能够有效地控制并发访问,确保每个请求都从缓存中获取数据,从而避免数据库并发访问问题。










