辨识 golang 中的并发数据争用情况

学习Golang要努力，但是不要急！今天的这篇文章《辨识 golang 中的并发数据争用情况》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

type setgetRequestInfo struct {
  mu                 sync.Mutex
  once               *sync.Once
  firstSetOccurrence time.Time
  lastSetOccurrence  time.Time
  countPerClientSet  uint64 //if it exceeds max value,it will wrap around to 0
  countPerClientGet  uint64
  cumulativeTime     time.Duration
  setType            string
}

var (
  globalSetCount, globalGetCount uint64 = 0, 0
  backendSetGetInfomap                  = &sync.Map{}
)  

func processSetRequestStatsInfo(key string, setType string, incomingSetReqTime time.Time) {
    val, loaded := backendSetGetInfomap.LoadOrStore(key, &setgetRequestInfo{setType: setType, once: &sync.Once{}})

    if !loaded {
        info := val.(*setgetRequestInfo)
        info.once.Do(func() {
           if info.countPerClientSet == 0 {
              info.firstSetOccurrence = incomingSetReqTime
           }
        })
    }
    if info, ok := val.(*setgetRequestInfo); ok {
        info.updateStats(incomingSetReqTime)
    }
}

func (info *setgetRequestInfo) updateStats(incomingSetReqTime time.Time) {
   info.mu.Lock()
   defer info.mu.Unlock()
   info.countPerClientSet++
   info.lastSetOccurrence = incomingSetReqTime
   info.cumulativeTime += time.Since(incomingSetReqTime)
}

我需要验证调用processsetrequeststatsinfo时是否有2个或更多goroutines会导致数据竞争

谁能告诉我如何验证？

正确答案

是的，有数据争用报告。这是我检查它的方法：

1. 为函数编写一个测试，并行执行几秒钟。

func test_processsetrequeststatsinfo(t *testing.t) {
    const routines = 100
    const testduration = 5 * time.second

    ctx, cancel := context.withtimeout(context.background(), testduration)
    defer cancel()

    for i := 0; i < routines; i++ {
        go func() {
            for {
                select {
                case <-ctx.done():
                    return
                default:
                }

                processsetrequeststatsinfo("key1", "sometype", time.now())
            }
        }()
    }
    <-ctx.done()
}

如果需要，可以通过使用不同的键和类型来改进测试。

1. 在启用竞争检测器的情况下运行测试：

go test -race . -count=1

-count=1 确保每次都运行测试并忽略缓存的结果。

这将（可能不是第一次，但在重复运行时）报告数据争用。 （省略，因为文件和行号适用于我的测试代码。）其要点是 updatestats 中对 info 的访问与 processsetrequeststatsinfo 中对 info 的访问发生冲突。一个受互斥体保护，但另一个则不受互斥体保护。

@kostix 已经在评论中描述了该问题，解释了发生的情况。我只想补充一点，代码使用 2 个不同的互斥体，并且需要将其归结为 1 个互斥体。目前它使用 mu 和 once 互斥体。 sync.once 也包含一个互斥锁。将sync.once 替换为mu 互斥锁即可解决该问题。

type setgetRequestInfo struct {
    mu                 sync.Mutex
    firstSetOccurrence time.Time
    lastSetOccurrence  time.Time
    countPerClientSet  uint64 // if it exceeds max value,it will wrap around to 0
    countPerClientGet  uint64
    cumulativeTime     time.Duration
    setType            string
}

var (
    globalSetCount, globalGetCount uint64 = 0, 0
    backendSetGetInfomap                  = &sync.Map{}
)

func processSetRequestStatsInfo(key string, setType string, incomingSetReqTime time.Time) {
    val, loaded := backendSetGetInfomap.LoadOrStore(key, &setgetRequestInfo{setType: setType})

    if !loaded {
        info := val.(*setgetRequestInfo)
        info.mu.Lock()
        if info.countPerClientSet == 0 {
            info.firstSetOccurrence = incomingSetReqTime
        }
        info.mu.Unlock()
    }
    if info, ok := val.(*setgetRequestInfo); ok {
        info.updateStats(incomingSetReqTime)
    }
}

func (info *setgetRequestInfo) updateStats(incomingSetReqTime time.Time) {
    info.mu.Lock()
    defer info.mu.Unlock()
    info.countPerClientSet++
    info.lastSetOccurrence = incomingSetReqTime
    info.cumulativeTime += time.Since(incomingSetReqTime)
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。