Go中的同步与锁
Mutex 互斥锁
- 互斥锁是传统并发程序对共享资源进行访问控制的主要手段。是sync包中的Mutex结构体。
type Mutex struct {}
- 该结构体包括了两个方法，可以说是非常简单使用的
func (m *Mutex) Lock() {}func (m *Mutex) Unlock() {}
- 我们通过一个简单的例子来说明他的用法：
package main
import (
"sync"
"fmt"
)
type safeCounter struct {
number int
sync.Mutex
}
func (sc *safeCounter) Increment() {
sc.Lock()
sc.number++
sc.Unlock()
}
func (sc *safeCounter) Decrement() {
sc.Lock()
sc.number--
sc.Unlock()
}
func (sc *safeCounter) getNumber() int {
sc.Lock()
number := sc.number
sc.Unlock()
return number
}
func main() {
sc := new(safeCounter)
for i := 0; i < 100; i++ {
go sc.Increment()
go sc.Decrement()
}
fmt.Println(sc.getNumber())
}- 上述例子通过互斥锁来保证number整型变量在使用goroutine的情况下，持续稳定地输出0的结果。当然你可以试一下，去掉锁的保护，是否还可以稳定输出0？
RWMutex 读写锁
- 读写锁是针对读写操作的互斥锁。遵循两个原则即可
- 可以随便地读，并且支持多个goroutine同时读取
- 写的时候，既不能读也不能写。
- RWMutex 提供了四个方法
- func (*RWMutex) Lock
- func (*RWMutex) Unlock
- func (*RWMutex) RLock
- func (*RWMutex) RUnlock
- 通过读写map的一个例子来说明RWMutex的使用方法：
package main
import (
"fmt"
"math/rand"
"sync"
"time"
)
type MapCounter struct {
m map[int]int
sync.RWMutex
}
func (mapCounter *MapCounter) Reader(n int) {
for {
mapCounter.RLock()
v := mapCounter.m[rand.Intn(n)]
mapCounter.RUnlock()
fmt.Println(v)
time.Sleep(1 * time.Second)
}
}
func (mapCounter *MapCounter) Writer(n int) {
for i := 0; i < n; i++ {
mapCounter.Lock()
mapCounter.m[i] = i * 10
mapCounter.Unlock()
time.Sleep(1 * time.Second)
}
}
func main() {
mc := MapCounter{m: make(map[int]int)}
go mc.Writer(10)
go mc.Reader(10)
go mc.Reader(10)
time.Sleep(15 * time.Second)
}Once
- Once 顾名思义，就是只执行一次 。通过一个简单的例子就可以说明：
package main
import (
"fmt"
"sync"
)
func main() {
once := sync.Once{}
done := make(chan bool)
for i := 0; i < 10; i++ {
go func() {
once.Do(sayHello)
done <- true
}()
}
for i := 0; i < 10; i++ {
fmt.Println(<-done)
}
}
func sayHello() {
fmt.Println("Hello")
}- 以上的例子只输出了一个“Hello” ，其余的都是true。
WaitGroup
- WaitGroup 是可以一个goroutine集合，当集合里的goroutine执行完毕后，这个WaitGroup就会自动结束。你可以定义这个WaitGroup的大小。他里面只有三个方法：
- func (wg *WaitGroup) Add(delta int) {} //定义大小
- func (wg *WaitGroup) Done() {}
- func (wg *WaitGroup) Wait() {}
- 具体用法，通过下面的小例子说明。
package main
import (
"fmt"
"sync"
)
func main() {
wg := sync.WaitGroup{}
for i := 0; i <= 5; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
fmt.Println("Work done for ", i)
}(i)
}
wg.Wait()
}- 建议大家可以把上面的代码都跑一遍，方便自己理解代码的含义。结合官方文档来理解会更加容易。
参考
- https://gowalker.org/sync 更多的信息请大家参考文档
本文参与 腾讯云自媒体分享计划，分享自微信公众号。
原始发表：2017-05-30，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录