Golang实现对map的并发读写

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liyunlong41/article/details/82958971

在Golang多协程的情况下使用全局map时,如果不做线程同步,会出现panic的情况。
为了解决这个问题,通常有两种方式:

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

写了一个模拟程序对map中的一项进行读或者写,后台一直运行的协程阻塞的接受读写信号,并对map进行操作,但是读操作的时候没想好怎么返回这个值。
后来想到用传引用的方式,定义结构体,第一个参数是读写的标志,第二个参数是读成功或者写成功后的值,定义的channel中传结构体指针。
ps:以后验证一下效率。
现在发现读取时有问题,暂时还没改,因为给ch发了信号之后可能还没读取完dic中的内容,但是main函数中就使用了ret参数,这时ret参数是以前的。后面考虑改成等待channel的形式。

package main

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

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


func readAndWrite2Map() {
	for {
		select{
		case flag := <- ch:
			if flag.op > 0 {
				dic[1] = flag.op
				flag.ret = dic[1]
			} else if flag.op == 0 {
				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)
}

猜你喜欢

转载自blog.csdn.net/liyunlong41/article/details/82958971
今日推荐