Golang实现对map的并发读写

  • Post author:
  • Post category:golang


在Golang多协程的情况下使用全局map时,如果不做线程同步,会出现panic的情况。

为了解决这个问题,通常有两种方式:

  • 第一种是最常见的使用互斥锁或者读写锁的方法;
  • 第二种是比较符合Golang特色的方法,启动单个协程对map进行读写,当其他协程需要读写map时,通过channel向这个协程发送信号即可。

写了一个模拟程序对map中的一项进行读或者写,后台一直运行的协程阻塞的接受读写信号,并对map进行操作,但是读操作的时候没想好怎么返回这个值。

后来想到用传引用的方式,定义结构体,第一个参数是读写的标志,第二个参数是读成功或者写成功后的值的channel,定义的channel中传结构体指针。

ps:以后验证一下效率。简单封装了一下:

https://blog.csdn.net/liyunlong41/article/details/84259488

package main

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

type value struct {
	id int
	op int
	ret chan int
}
var dic map[int]int
var ch chan *value

func readAndWrite2Map() {
	for {
		select{
		case flag := <- ch:

			if flag.op > 0 {
				log.Printf("id: %v, op: %v, ret: %v", flag.id, flag.op, flag.op)
				dic[1] = flag.op
				flag.ret <- dic[1]
			} else if flag.op == 0 {
				log.Printf("id: %v, op: %v, ret: %v", flag.id, flag.op, dic[1])
				flag.ret <- dic[1]
			} else {
				return
			}
		}
	}
}


func out(flag, i, val int) {
	if flag == 0 {
		fmt.Println(strconv.Itoa(i) + "th goroutine read the value is ", val)
	} else {
		fmt.Println(strconv.Itoa(i)+"th goroutine write to the map  ", val)
	}
}

func main() {
	dic = make(map[int]int)
	ch = make(chan *value)
	dic[1] = -1
	go readAndWrite2Map()
	for i := 0; i <= 5; i++ {
		if (i % 2) == 0 {
				go func(i int) {
					var tmp value
					for {
						tmp.op = 0
						ch <- &tmp
						out(0, i, <-tmp.ret)
						time.Sleep(time.Millisecond)
					}
				}(i)

		} else {
				go func(i int) {
					var tmp value
					for {
						tmp.op = i
						ch <- &tmp
						out(1, i, <-tmp.ret)
						time.Sleep(time.Millisecond)

					}
				}(i)
		}
	}
	time.Sleep(time.Second * 60)
}



版权声明:本文为liyunlong41原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。