Golang 共享内存
Golang语言中,进程间通信除了使用channel,还可以使用共享内存,即sync包。
Q-1
package main
import (
"fmt"
"time"
)
func main() {
for i := 1; i < 6; i++ {
go func(x int) {
time.Sleep(time.Second)
fmt.Println("第", x, "个子goroutine执行结束")
}(i)
}
fmt.Println("主goroutine执行结束")
}
上面的代码毫无疑问的只会输出:“主goroutine执行结束”,而不会输出子goroutine执行结束,这是因为,主协程一旦执行完毕,整个进程就会结束。
sync组等待
sync包中常用的有三个方法:Add(num int) 、 Done()、 Wait()
sync中会在内部维持一个类似于计数器的东西,通过Add方法来增加计数器的值,增加参数num;通过Done来减少计数器的值,一次只能减少1;Wait方法会使主协程阻塞,只有到内部计数器的值为0时,阻塞才会解开。
用法如下:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
for i := 1; i < 6; i++ {
wg.Add(1) //计数器加1
go func(x int) {
time.Sleep(time.Second)
fmt.Println("第", x, "个子goroutine执行结束")
wg.Done() //goroutine执行完毕,计数器减1
}(i)
}
wg.Wait() // 计数器不为0的时候,一直阻塞;计数器变为0的时候,阻塞解开
fmt.Println("主goroutine执行结束")
}
注意:添加的计时器数量必须比减少计时器的数量要少,否则会引发死锁。
比如,将计数器值增加了10,但是只执行了9次计数器减少,剩余一个计数器,计数器的值始终不为0,则主协程一直等待,形成死锁。但是较少计数器的次数可以比计数器的值要大,可以提前结束主协程的阻塞。
Q-2
请看下面一段代码,尝试在两个循环中,各自使sum加1,执行1000次,乍一看,觉得结果是sum的最终结果是2000,但是真正执行之后,结果不是2000。
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
wg.Add(2)
var sum int = 0
go func() {
for i := 0; i < 1000; i++ {
sum++
}
wg.Done()
}()
go func() {
for i := 0; i < 1000; i++ {
sum++
}
wg.Done()
}()
wg.Wait()
fmt.Println(sum)
}
sync互斥锁
sync互斥锁有两个常用的方法,Lock()加锁,Unlock()解锁。使用Lock加锁后,不能再进行加锁,只有当对其进行Unlock解锁之后,才能对其加锁。这个很好理解。
如果对一个未加锁的资源进行解锁,会引发panic异常。
可以在一个goroutine中对一个资源加锁,而在另外一个goroutine中对该资源进行解锁。
使用示例如下:
package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
var mutex sync.Mutex
wg.Add(2)
var sum int = 0
go func() {
for i := 0; i < 1000; i++ {
mutex.Lock() //上锁
sum++
mutex.Unlock()//解锁
}
wg.Done()
}()
go func() {
for i := 0; i < 1000; i++ {
mutex.Lock() //加锁
sum++
mutex.Unlock() //解锁
}
wg.Done()
}()
wg.Wait()
fmt.Println(sum)
}
sync读写互斥锁(多读单写锁)
允许对同一资源进行同时读,但是不能进行同时写,也不能进行一边读,一边写。
如果一个资源被加了写锁,那么就不能再给他加其他锁,包括写锁,只能等写锁解开后,才能继续加其他锁。如果一个资源被加了读锁,那么,仍然可以给该资源加读锁,但是不能加写锁,必须等读锁释放后,才能加写锁。
sync.RWMutex有四个常用的方法:
Lock()加写锁 、 Unlock()取消写锁 、 RLock()加读锁 、 RUnlock()释放读锁
使用示例:
package main
import (
"fmt"
"sync"
"time"
)
var rwMutex sync.RWMutex //定义一个读写互斥锁
//注意传递的组等待是指针形式
func WriteData(wg *sync.WaitGroup, id int) {
rwMutex.Lock() //加 写锁
fmt.Println("增加第", id, "个写锁")
for i := 0; i < 5; i++ {
fmt.Println("进行写操作")
time.Sleep(time.Second)
}
fmt.Println("释放第", id, "个写锁")
rwMutex.Unlock() //释放 写锁
wg.Done()
}
func ReadData(wg *sync.WaitGroup, id int) {
rwMutex.RLock() //加 读锁
fmt.Println("增加第", id, "个读锁")
for i := 0; i < 5; i++ {
fmt.Println("进行读操作")
time.Sleep(time.Second)
}
fmt.Println("释放第", id, "个读锁")
rwMutex.RUnlock() //释放 读锁
wg.Done()
}
func main() {
var wg sync.WaitGroup
for i := 1; i < 3; i++ {
wg.Add(1)
go WriteData(&wg, i)
}
for i := 1; i < 4; i++ {
wg.Add(1)
go ReadData(&wg, i)
}
wg.Wait()
fmt.Println("主程序执行完毕")
}
运行之后,就会发现,只有一个写加锁,只有写解锁之后,才允许加其他锁;如果是加读锁之后,可以再加读锁。
Q-3
package main
import (
"fmt"
"sync"
)
func demo() {
fmt.Println("运行demo函数")
}
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i++ {
wg.Add(1)
go func() {
demo()
wg.Done()
}()
}
wg.Wait()
fmt.Println("主程序结束")
}
这个示例代码运行之后,会输出10次“运行demo函数”,因为有10个goroutine,每个goroutine都运行了1次。
现在的需求是:虽然有10个goroutine,但是希望只有一个goroutine中执行demo函数,如果一个goroutine执行过了demo函数,其他的goroutine就会忽略demo函数,不去执行demo,保证demo函数只被执行一次。
sync.once初始化
sync.Once就是用来解决上面这个问题的:Once的作用就是多次调用,但只执行一次,Once只有一个方法->Once.Do(func),向Do中传入一个函数,在第一次执行Once.Do()的时候,才执行传入的func,以后在执行Once.Do(),就不会再执行传入的func。
示例如下:
package main
import (
"fmt"
"sync"
)
func demo() {
fmt.Println("运行demo函数")
}
func main() {
var wg sync.WaitGroup
var once sync.Once
for i := 0; i < 10; i++ {
wg.Add(1)
go func() {
once.Do(demo)
//注意不要写成once.Do(demo()),这样写是把demo()的运行结果传入Do
wg.Done()
}()
}
wg.Wait()
fmt.Println("主程序结束")
}
这样就不会执行多次demo(),而只输出一次“运行demo函数”
浙公网安备 33010602011771号