时间戳

首页 > 解决方案 > 如何锁定地图并修复数据竞争

go - 如何锁定地图并修复数据竞争

问题描述

我正在尝试解决 WARNING: DATA RACE 这是代码：

package models

import (
  "sync"
  "time"
)
type Stats struct {
    sync.Mutex
    request map[int64]int
}

func (s *Stats) PutRequest() {
  s.Lock()
  s.request[time.Now().Unix()]++
  s.Unlock()
}
func (s *Stats) GetRequests() map[int64]int {
    s.Lock()
    m := s.request
    s.Unlock()
    return m
 }
var Requests = Stats{
    sync.Mutex{},
    make(map[int64]int),
}

如果我将 Stats 字段请求更改为整数，则一切正常，但不适用于地图。如何在 Go 中正确锁定地图？

标签： gomutex

解决方案

使用 sync.RWMutex

func (s *Stats) PutRequest(ut int64) {
  s.Lock()
  defer s.Unlock()
  s.request[ut]++
}

func (s *Stats)  GetRequests() map[int64]int {
    s.RLock()
    defer s.RUnlock()
    m := make(map[int64]int, len(s.request))
    for k, v := range s.request {
        m[k] = v
    }
    return m
}

在这种情况下，以下通道示例可能会很有趣。举个例子——有状态的 goroutines

无论如何，您需要在返回之前复制地图。

推荐阅读