我可以在 Go 中使用特定值锁定吗？

目前golang学习网上已经有很多关于Golang的文章了，自己在初次阅读这些文章中，也见识到了很多学习思路；那么本文《我可以在 Go 中使用特定值锁定吗？》，也希望能帮助到大家，如果阅读完后真的对你学习Golang有帮助，欢迎动动手指，评论留言并分享~

问题内容

在回答另一个问题时，我写了一个sync.Map用于缓存 API 请求的小结构。

type PostManager struct {
    sync.Map
}

func (pc PostManager) Fetch(id int) Post {
    post, ok := pc.Load(id)
    if ok {
        fmt.Printf("Using cached post %v\n", id)
        return post.(Post)
    }
    fmt.Printf("Fetching post %v\n", id)
    post = pc.fetchPost(id)
    pc.Store(id, post)

    return post.(Post)
}

不幸的是，如果两个 goroutine 同时获取同一个未缓存的 Post，它们都会发出请求。

var postManager PostManager

wg.Add(3)

var firstPost Post
var secondPost Post
var secondPostAgain Post

go func() {
    // Fetches and caches 1
    firstPost = postManager.Fetch(1)
    defer wg.Done()
}()

go func() {
    // Fetches and caches 2
    secondPost = postManager.Fetch(2)
    defer wg.Done()
}()

go func() {
    // Also fetches and caches 2
    secondPostAgain = postManager.Fetch(2)
    defer wg.Done()
}()

wg.Wait()

我需要确保当同时提取相同 ID 时，只允许一个实际发出请求。另一个必须等​​待并将使用缓存的 Post。但也不要锁定不同 ID 的提取。

在上面的示例中，我希望只有一个调用pc.fetchPost(1)andpc.fetchPost(2)并且它们应该是同时的。

链接到完整代码。

正确答案

golang.org/x/sync/singleflight包正是为此目的而编写的。

请注意，所有缓存访问都应该发生在传递给 Do 的回调函数中。在您在评论中链接到的代码中，您在外部进行查找；这有点违背了目的。

此外，您必须使用指向 singleflight.Group 的指针。这就是你的数据竞赛的来源，请去 vet 指出：

./foo.go:41:10：fetchPost 按值传递锁：命令行参数。PostManager 包含 golang.org/x/sync/singleflight.Group 包含 sync.Mutex

这是我的写法（操场上的完整示例：https: //play.golang.org/p/2hE721uA88S）：

import (
    "strconv"
    "sync"

    "golang.org/x/sync/singleflight"
)

type PostManager struct {
    sf    *singleflight.Group
    cache *sync.Map
}

func (pc *PostManager) Fetch(id int) Post {
    x, _, _ := pc.sf.Do(strconv.Itoa(id), func() (interface{}, error) {
        post, ok := pc.cache.Load(id)
        if !ok {
            post = pc.fetchPost(id)
            pc.cache.Store(id, post)
        }

        return post, nil
    })

    return x.(Post)
}

到这里，我们也就讲完了《我可以在 Go 中使用特定值锁定吗？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于golang的知识点！