Golang实现对map的并发读写的方法示例

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

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

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

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

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

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)
}

相关推荐