并发获取数据的singleflight使用问题

在使用singleflight并发获取数据时，多个请求同时访问数据库的问题。

问题代码：

package main

import (
    "errors"
    "log"
    "sync"
    "golang.org/x/sync/singleflight"
)

var errornotexist = errors.new("not exist")
var g singleflight.group

func main() {
    var wg sync.waitgroup
    cpu_amount := 4
    wg.add(cpu_amount)

    for i := 0; i < cpu_amount; i++ {
        go func() {
            defer wg.done()
            data, err := getdata("key")
            if err != nil {
                log.print(err)
                return
            }
            log.println(data)
        }()
    }
    wg.wait()
}

func getdata(key string) (string, error) {
    data, err := getdatafromcache(key)
    if err == errornotexist {
        v, err, _ := g.do(key, func() (interface{}, error) {
            return getdatafromdb(key)
        })
        if err != nil {
            log.println(err)
            return "", err
        }
        data = v.(string)
    } else if err != nil {
        return "", err
    }
    return data, nil
}

func getdatafromcache(key string) (string, error) {
    return "", errornotexist
}

func getdatafromdb(key string) (string, error) {
    log.printf("get %s from database", key)
    return "data", nil
}

问题原因：

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

singleflight仅避免大量并发的访问，即在多个协程同时运行的情况下才满足“同时”的条件。

在给定代码中，getdatafromdb函数在输出信息后就结束了，这意味着如果前面三个协程都已执行完毕，第四个协程来执行时，就不能满足“同时”性，因此不会进入singleflight的do函数，从而导致访问穿透。

解决方案：

为了解决此问题，可以修改getdatafromdb函数，在返回数据之前引入延迟，以确保所有协程同时运行。例如：

func getDataFromDB(key string) (string, error) {
    log.Printf("get %s from database", key)
    time.Sleep(time.Millisecond * 500) // 引入延迟
    return "data", nil
}