Go语学习笔记 - 读写锁 | 从零开始Go语言

package main
import (
  "fmt"
  "sync"
  "time"
)
var (
  rwCount  int
  myRWLock sync.RWMutex
  wg2      sync.WaitGroup
)
func ReadCount() int {
  myRWLock.RLock()
  fmt.Println("ReadCount readlock ")
  time.Sleep(time.Second)
  defer myRWLock.RUnlock()
  defer fmt.Println("ReadCount unreadlock ")
  return rwCount
}
func ReadCount1() int {
  myRWLock.RLock()
  fmt.Println("ReadCount1 readlock ")
  time.Sleep(time.Second)
  defer myRWLock.RUnlock()
  defer fmt.Println("ReadCount1 unreadlock ")
  return rwCount
}
func HandleCount() {
  myRWLock.Lock()
  fmt.Println("HandleCount writelock ")
  time.Sleep(time.Second)
  defer myRWLock.Unlock()
  defer fmt.Println("HandleCount unwritelock ")
  rwCount += 1
}
func main() {
  wg2.Add(3)
  go func() {
  defer wg2.Done()
  for i := 0; i < 11; i++ {
    fmt.Println("count1 -> ", ReadCount())
  }
  }()
  go func() {
  defer wg2.Done()
  for i := 0; i < 11; i++ {
    fmt.Println("count2 -> ", ReadCount1())
  }
  }()
  go func() {
  defer wg2.Done()
  for i := 0; i < 10; i++ {
    HandleCount()
    fmt.Println("count -> add 1")
  }
  }()
  wg2.Wait()
}