Go学习(十九):并发编程-Sync包使用

1.介绍

sync包提供了互斥锁。除了Once和WaitGroup类型，其余多数适用于低水平的程序，多数情况下，高水平的同步使用channel通信性能会更优一些。

2.并发等待组(WaitGroup)

WaitGroup，即等待一组Goroutine结束。父Goroutine调用Add()方法来设置应等待Goroutine的数量。每个被等待的Goroutine在结束时应该调用Done()方法。与此同时，主Goroutine可调用Wait()方法阻塞至所有Goroutine结束。

2.1 WaitGroup结构

// A WaitGroup must not be copied after first use.
type WaitGroup struct {
 noCopy noCopy
 state1 [3]uint32
}

2.2 方法列表

方法名	功能
(wg *WaitGroup) Add(delta int)	等待组的计数器 +1
(wg *WaitGroup) Done()	等待组的计数器 -1
(wg *WaitGroup) Wait()	当等待组计数器不等于0时,阻塞直到0

2.3 Add参数取值范围

等待组内部拥有一个计数器，计数器的值可以通过Add(delta int)方法调用实现计数器的增加和减少。该方法应该在创建新的Goroutine之前调用。

参数值x取值

取值	描述
delta < 0	x小于0时,但会报错: panic: sync: negative WaitGroup counter
delta = 0	x等于0时,会释放Wait()方法阻塞等待的所有Goroutine
delta > 0	x大于0时,Wait()方法会阻塞Goroutine直到WaitGroup计数减为0

2.4 使用示例

1. 不使用WaitGroup示例

package main
import (
 "fmt"
 "time"
)
func main() {
 // 创建通道
 intChan := make(chan int)

 // 计算1-50的和
 go func(intChan chan int) {
  sum := 0
  for i := 1; i<= 50 ; i++  {
   sum += i
  }
  intChan <- sum
 }(intChan)
 // 计算51-100的和
 go func(intChan chan int) {
  sum := 0
  for i := 51; i<= 100 ; i++  {
   sum += i
  }
  intChan <- sum
 }(intChan)
 // 另外创建个channle聚合结果
 go func(intChan chan int) {
  sum1 := <- intChan
  sum2 := <- intChan
  fmt.Printf("sum1 = %d sum2 = %d  \nsum1 + sum2 = %d \n",sum1,sum2,sum1+sum2)
 }(intChan)

 // 注意,需求手动sleep
 time.Sleep( time.Second)
 fmt.Println("运行结束")
}
/** 输出:
  sum1 = 1275 sum2 = 3775  
  sum1 + sum2 = 5050 
  运行结束
*/

2.使用WaitGroup示例

package main
import (
 "fmt"
 "sync"
)
func main() {
 // 声明等待组
 var wg sync.WaitGroup
 // 设置，需要等待3个协程执行完成
 wg.Add(3)
 // 创建通道
 intChan := make(chan int)
 // 计算1-50的和
 go func(intChan chan int, wg *sync.WaitGroup) {
  sum := 0
  for i := 1; i <= 50; i++ {
   sum += i
  }
  intChan <- sum
  // 计数器减一
  wg.Done()
 }(intChan, &wg)
 // 计算51-100的和
 go func(intChan chan int, wg *sync.WaitGroup) {
  sum := 0
  for i := 51; i <= 100; i++ {
   sum += i
  }
  intChan <- sum
  // 计数器减一
  wg.Done()
 }(intChan, &wg)
 // 另外创建个channle聚合结果
 go func(intChan chan int,wg *sync.WaitGroup) {
  sum1 := <-intChan
  sum2 := <-intChan
  fmt.Printf("sum1 = %d sum2 = %d  \nsum1 + sum2 = %d \n", sum1, sum2, sum1+sum2)
  // 计数器减一
  wg.Done()
 }(intChan,&wg)
 // 阻塞，直到等待组的计数器等于0
 wg.Wait()
 fmt.Println("运行结束")
}
/**输出
  sum1 = 3775 sum2 = 1275  
  sum1 + sum2 = 5050 
  运行结束
*/

3.互斥锁(Mutex)

Mutex是一个互斥锁，保证同时只有一个Goroutine可以访问共享资源。Mutex类型的锁和Goroutine无关，可以由不同的Goroutine加锁和解锁。也可以为其他结构体的字段，零值为解锁状态。

3.1 结构介绍

type Mutex struct {
 state int32 // state 表示当前互斥锁的状态
 sema  uint32 // sema 是用于控制锁状态的信号量
}

3.2 方法列表

方法名	描述
(m *Mutex) Lock()	方法锁住m，如果 m 已经加锁，则阻塞直到 m 解锁。
(m *Mutex) Unlock()	解锁 m，如果 m 未加锁会导致运行时错误。

3.3 使用(售票)

需求:模拟多个窗口售票

1.不作为结构体属性使用

package main
import (
 "fmt"
 "sync"
 "time"
)
// 声明全局等待组
var wg sync.WaitGroup
// 声明全局锁
var mutex sync.Mutex
// 声明全局余票
var ticket int = 10
func main() {
 // 设置等待组计数器
 wg.Add(3)
 // 窗口卖票
 go saleTicket("窗口A",&wg)
 go saleTicket("窗口B",&wg)
 go saleTicket("窗口C",&wg)
 wg.Wait()
 fmt.Println("运行结束!")
}
// 卖票流程
func saleTicket(windowName string, wg *sync.WaitGroup)  {
 // 卖票流程结束后关闭
 defer wg.Done()
 for {
  // 加锁
  mutex.Lock()
  if  ticket > 0{
   time.Sleep(10 * time.Millisecond)
   ticket--
   fmt.Printf("%s 卖出一张票，余票: %d \n",windowName,ticket)
  } else  {
   fmt.Printf("%s 票已卖完! \n",windowName)
   // 解锁
   mutex.Unlock()
   break
  }
  // 解锁
  mutex.Unlock()
 }
}
/**输出
  窗口C 卖出一张票，余票: 9 
  窗口C 卖出一张票，余票: 8 
  窗口B 卖出一张票，余票: 7 
  窗口A 卖出一张票，余票: 6 
  窗口C 卖出一张票，余票: 5 
  窗口B 卖出一张票，余票: 4 
  窗口A 卖出一张票，余票: 3 
  窗口C 卖出一张票，余票: 2 
  窗口B 卖出一张票，余票: 1 
  窗口A 卖出一张票，余票: 0 
  窗口C 票已卖完! 
  窗口B 票已卖完! 
  窗口A 票已卖完! 
  运行结束!
*/

2.作为结构体属性使用

package main
import (
 "fmt"
 "strconv"
 "sync"
 "time"
)
// 声明一个票池
type ticketPool struct {
 over int
 lock sync.Mutex
 wg   sync.WaitGroup
}
// 定义售票方法
func (t *ticketPool) sellTicket(windowName string) {
 // 等待组减一
 defer t.wg.Done()
 for {
  // 加锁
  t.lock.Lock()
  if t.over > 0 {
   time.Sleep(10 * time.Millisecond)
   t.over--
   fmt.Printf("%s 卖出一张票，余票: %d \n", windowName, t.over)
  } else {
   // 无票，跳无限循环并解锁
   t.lock.Unlock()
   fmt.Printf("%s 票已卖完! \n", windowName)
   break
  }
  // 正常售票流程解锁
  t.lock.Unlock()
 }
}
func main() {
 // 创建一个票池
 ticketP := ticketPool{over: 10}
 fmt.Printf("T:%T v: %v \n", ticketP, ticketP)
 // 设置窗口数量
 windowNum := 3
 // 设置等待组计数器
 ticketP.wg.Add(windowNum)
 // 定义3个窗口售票
 for i:= 1 ; i <= windowNum; i++ {
  go ticketP.sellTicket("窗口" + strconv.Itoa(i))
 }
 ticketP.wg.Wait()
 fmt.Println("运行结束!")
}
/**输出
  窗口3 卖出一张票，余票: 9 
  窗口3 卖出一张票，余票: 8 
  窗口1 卖出一张票，余票: 7 
  窗口2 卖出一张票，余票: 6 
  窗口3 卖出一张票，余票: 5 
  窗口1 卖出一张票，余票: 4 
  窗口2 卖出一张票，余票: 3 
  窗口3 卖出一张票，余票: 2 
  窗口1 卖出一张票，余票: 1 
  窗口2 卖出一张票，余票: 0 
  窗口1 票已卖完! 
  窗口2 票已卖完! 
  窗口3 票已卖完! 
  运行结束!
*/

4.读写锁(RWMutex)

4.1 结构介绍

RWMutex是读写互斥锁，简称读写锁。该锁可以同时被多个读取者持有或被唯一个写入者持有。RWMutex类型锁跟Goroutine无关，可以由不同的Goroutine加锁、解锁。RWMutex也可以创建为其他结构体的字段；零值为解锁状态。

1. RWMutex锁结构

type RWMutex struct {
    w  Mutex //用于控制多个写锁，获得写锁首先要获取该锁，如果有一个写锁在进行，那么再到来的写锁将会阻塞于此
    writerSem  uint32 //写阻塞等待的信号量，最后一个读者释放锁时会释放信号量
    readerSem  uint32 //读阻塞的协程等待的信号量，持有写锁的协程释放锁后会释放信号量
    readerCount int32  //记录读者个数
    readerWait  int32  //记录写阻塞时读者个数
}

2. 读写锁堵塞场景

1. 写锁需要阻塞写锁：一个协程拥有写锁时，其他协程写锁需要阻塞
2. 写锁需要阻塞读锁：一个协程拥有写锁时，其他协程读锁需要阻塞
3. 读锁需要阻塞写锁：一个协程拥有读锁时，其他协程写锁需要阻塞
4. 读锁不能阻塞读锁：一个协程拥有读锁时，其他协程也可以拥有读锁

4.2 方法列表

方法名	描述
(rw *RWMutex) RLock()	获取读锁,当一个协程拥有读锁时，其他协程写锁需要阻塞。
(rw *RWMutex) RUnlock()	释放读锁。
(rw *RWMutex) Lock()	获取写锁，与Mutex完全一致； 当一个协程拥有写锁时，其他协程读写锁都需要阻塞
(rw *RWMutex) Unlock()	释放写锁

4.3 使用(读写文件)

1.不作为结构体属性使用

package main
import (
 "fmt"
 "strconv"
 "sync"
)
// 声明全局变量，文件内容
var fileContext string
// 声明全局读写互斥锁
var rxMutex sync.RWMutex
// 声明全局等待组
var wg sync.WaitGroup
func main() {
 // 设置计数器
 wg.Add(5)
 for i := 1; i <= 5; i++ {
  name := "同学-" + strconv.Itoa(i)
  if i%2 == 0 {
   go readFile(name)
  } else {
   go writeFile(name, strconv.Itoa(i))
  }
 }
 // 等待所有计数器执行完成
 wg.Wait()
 fmt.Println("运行结束!")
}

// 读文件
func readFile(name string) {
 // 释放读锁
 defer rxMutex.RUnlock()
 // 获取读锁
 rxMutex.RLock()
 // 打印读取内容
 fmt.Printf("%s 获取读锁，读取内容为: %s \n", name, fileContext)
 // 计数器减一
 wg.Done()
}
// 写文件
func writeFile(name, s string) {
 // 释放写锁
 defer rxMutex.Unlock()
 // 获取写锁
 rxMutex.Lock()
 // 写入内容
 fileContext = fileContext + " " + s
 fmt.Printf("%s 获取写锁，写入内容: %s。 文件内容变成: %s \n", name, s, fileContext)
 // 计数器减一
 wg.Done()
}

/**输出
  同学-1 获取写锁，写入内容: 1。 文件内容变成:  1 
  同学-4 获取读锁，读取内容为:  1 
  同学-2 获取读锁，读取内容为:  1 
  同学-5 获取写锁，写入内容: 5。 文件内容变成:  1 5 
  同学-3 获取写锁，写入内容: 3。 文件内容变成:  1 5 3 
  运行结束!
*/

2.作为结构体属性使用

package main
import (
 "fmt"
 "strconv"
 "sync"
 "time"
)
// 定义一个文件结构体
type fileResource struct {
 content string
 wg sync.WaitGroup
 rwLock sync.RWMutex
}

// 读文件
func (f *fileResource)readFile(name string) {
 // 释放读锁
 defer f.rwLock.RUnlock()
 // 获取读锁
 f.rwLock.RLock()
 // 打印读取内容
 time.Sleep(time.Second)
 fmt.Printf("%s 获取读锁，读取内容为: %s \n", name, f.content)
 // 计数器减一
 f.wg.Done()
}

// 写文件
func (f *fileResource)writeFile(name, s string) {
 // 释放写锁
 defer f.rwLock.Unlock()
 // 获取写锁
 f.rwLock.Lock()
 // 写入内容
 time.Sleep(time.Second)
 f.content = f.content + " " + s
 fmt.Printf("%s 获取写锁，写入内容: %s。 文件内容变成: %s \n", name, s, f.content)
 // 计数器减一
 f.wg.Done()
}
func main() {
 // 声明结构体
 var file fileResource
 // 设置计数器
 file.wg.Add(5)
 for i := 1; i <= 5; i++ {
  name := "同学-" + strconv.Itoa(i)
  if i%2 == 0 {
   go file.readFile(name)
  } else {
   go file.writeFile(name, strconv.Itoa(i))
  }
 }
 // 等待所有计数器执行完成
 file.wg.Wait()
 fmt.Println("运行结束!")
}
/**输出
  同学-5 获取写锁，写入内容: 5。 文件内容变成:  5 
  同学-1 获取写锁，写入内容: 1。 文件内容变成:  5 1 
  同学-2 获取读锁，读取内容为:  5 1 
  同学-3 获取写锁，写入内容: 3。 文件内容变成:  5 1 3 
  同学-4 获取读锁，读取内容为:  5 1 3 
  运行结束!
*/

5.条件变量(Cond)

5.1 介绍

与互斥锁不同，条件变量的作用并不是保证在同一时刻仅有一个线程访问某一个共享数据，而是在对应的共享数据的状态发生变化时，通知其他因此而被阻塞的线程。条件变量总是与互斥锁组合使用，互斥锁为共享数据的访问提供互斥支持，而条件变量可以就共享数据的状态的变化向相关线程发出通知。

使用场景: 我需要完成一项任务，但是这项任务需要满足一定条件才可以执行，否则我就等着。

5.2 方法列表

方法名	描述
NewCond(l Locker) *Cond	生成一个cond,需要传入实现Locker接口的变量。 一般是*Mutex或*RWMutex类型的值。
(c *Cond) Wait()	等待通知
(c *Cond) Signal()	发送单个通知
(c *Cond) Broadcast()	广播(多个通知)

5.3 使用示例

package main
import (
 "fmt"
 "sync"
 "time"
)
func main() {
 // 声明互斥锁
 var mutex sync.Mutex
 // 声明条件变量
 cond := sync.NewCond(&mutex)
 for i := 1; i <= 10; i++ {
  go func(i int) {
   // 获取锁
   cond.L.Lock()
   // 释放锁
   defer cond.L.Unlock()
   // 等待通知,阻塞当前协程
   cond.Wait()
   // 等待通知后打印输出
   fmt.Printf("输出:%d ! \n", i)
  }(i)
 }
 // 单个通知
 time.Sleep(time.Second)
 fmt.Println("单个通知A！")
 cond.Signal()
 time.Sleep(time.Second)
 fmt.Println("单个通知B！")
 cond.Signal()

 // 广播通知
 time.Sleep(time.Second)
 fmt.Println("广播通知！并睡眠1秒，等待其他协程输出!")
 cond.Broadcast()
 // 等待其他协程处理完
 time.Sleep(time.Second)
 fmt.Println("运行结束！")
}
/**输出
  单个通知A！
  输出:1 ! 
  单个通知B！
  输出:4 ! 
  广播通知！并睡眠1秒，等待其他协程输出!
  输出:10 ! 
  输出:2 ! 
  输出:3 ! 
  输出:8 ! 
  输出:9 ! 
  输出:6 ! 
  输出:5 ! 
  输出:7 ! 
  运行结束！
*/

6.一次(Once)

sync.Once 是使Go方法只执行一次的对象实现，作用与 init 函数类似,但也有所不同。区别如下:

• init 函数是在文件包首次被加载的时候执行，且只执行一次
• sync.Onc 是在代码运行中需要的时候执行，且只执行一次

6.1 方法介绍

方法名	描述
(o *Once) Do(f func())	函数只会执行一次，并保证在返回时，传入Do的函数已经执行完成。 多个 goroutine 同时执行 once.Do 的时候，可以保证抢占到 once.Do 执行权的 goroutine 执行完 once.Do 后，其他goroutine才能得到返回 。

6.2 使用示例

示例1: 重复调用只执行一次

package main
import (
 "fmt"
 "strconv"
 "sync"
 "time"
)
func main() {
 echo := func() {
  t := time.Now().Unix()
  fmt.Printf("输出时间 %v ",strconv.FormatInt(t,10))
 }
 var one sync.Once
  // 虽然遍历调用，但是只会执行一次
 for i := 1; i< 10 ; i++  {
  go func(a,b int) {
   one.Do(echo)
  }(i,i+1)
 }
 time.Sleep(3 * time.Second)
 fmt.Println("运行结束！")
}
/**输出
  输出时间 1608083525 
  运行结束！
*/

微信搜索关注【猿码记】查看更多文章。

本文由 mdnice 多平台发布