前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >golang中Map的并发写入

golang中Map的并发写入

原创
作者头像
KunkkaWu
修改2022-06-27 11:09:24
9760
修改2022-06-27 11:09:24
举报
文章被收录于专栏:算法协议

原理

golang中的map不是线程安全的,所以在并发的情况下不能直接使用map。

反面例子

代码语言:txt
复制
package main

import (
    "strconv"
    "time"
)

var m = make(map[string]interface{})

func main() {
    testMap(m)
    time.Sleep(2 * time.Second)
}

func testMap() {
    go func() {
        for i := 0; i < 10; i++ {
            name := "WangWu" + strconv.Itoa(i)
            writeMap(m, name, i)
        }
    }()

    go func() {
        for i := 0; i < 10; i++ {
            name := "ZhuoQi" + strconv.Itoa(i)
            writeMap(m, name, i)
        }
    }()
}

func writeMap(m map[string]int, key string, val int) {
    m[key] = val
}

运行时,会报错:

代码语言:txt
复制
fatal error: concurrent map writes

解决办法

1. Map加锁

代码语言:txt
复制
package main

import (
    "fmt"
    "strconv"
    "sync"
    "time"
)

// 定义锁
var locker = &sync.RWMutex{}
var m = make(map[string]interface{})

func main() {
    testMap()
    time.Sleep(2 * time.Second)
    fmt.Println(m)
}

func testMap() {
    go func() {
        for i := 0; i < 10; i++ {
            name := "WangWu" + strconv.Itoa(i)
            writeMap(name, i)
        }
    }()

    go func() {
        for i := 0; i < 10; i++ {
            name := "ZhuoQi" + strconv.Itoa(i)
            writeMap(name, i)
        }
    }()
}

func writeMap(key string, val int) {
    locker.Lock()
    m[key] = val
    locker.Unlock()
}

2. 使用sync.Map

代码语言:txt
复制
package main

import (
    "fmt"
    "strconv"
    "sync"
    "time"
)

var m = sync.Map{}

func main() {
    testMap()
    time.Sleep(2 * time.Second)
    m.Range(func(k, v interface{}) bool {
        fmt.Println("map:", k, v)
        return true
    })
}

func testMap() {
    go func() {
        for i := 0; i < 10; i++ {
            name := "WangWu" + strconv.Itoa(i)
            m.Store(name, i)
        }
    }()

    go func() {
        for i := 0; i < 10; i++ {
            name := "ZhuoQi" + strconv.Itoa(i)
            m.Store(name, i)
        }
    }()
}

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 原理
  • 反面例子
  • 解决办法
    • 1. Map加锁
      • 2. 使用sync.Map
      领券
      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档