欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-03-18 11:58:08
在golang多协程的情况下使用全局map时,如果不做线程同步,会出现panic的情况。 为了解决这个问题,通常有两种方式: 第一种是最常见的使用互斥锁或者...

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

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

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

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

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

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

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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。