sync
sync包提供了许多功能，比如sync.WaitGroup保证所有gorountinue都能执行完
sync.Once 保证在多线程情况只执行一次等等。下面分别说下：
1. sync.WaitGroup 保证所有gorountine执行完
它主要是通过：
-
Add()gorountine计数+1 -
Done()表明 当前gorountine执行完 -
Wait()等待所有gorountine执行完
通过以上三个方法实现,看代码
package main
import (
"fmt"
"sync"
"time"
)
var w sync.WaitGroup
func main() {
w.Add(1) // 添加一个gorountine，开启gorountine前调用
go func() {
defer w.Done() // 保证执行完会标记此gorountine执行过
for i := 0; i < 10; i++ {
fmt.Println("i:", i)
time.Sleep(time.Second)
}
}()
w.Wait() // 等待所有gorountine执行完
fmt.Println("finished!")
}
2. sync.Once保证在多线程情况下也只执行一次
在真实项目中，我们会存在许多情况，要求多线程只执行一次（不用多次重复执行）；
比如：
- 更新缓存（一个线程去更新时，其它线程不要再去更新）
- 初始化配置数据
虽然通过加锁也达到只执行一次的目的；但是性能不好；go专门为上述场景，设计了sync.Once
,它内部提供了一个Do()方法，参数是一个函数，来实现。
看代码：
package main
import (
"fmt"
"sync"
)
var (
w sync.WaitGroup
once sync.Once
)
// 初始化配置文件
func initConfig() {
fmt.Println("执行初始化配置文件!")
}
func main() {
// 添加10个 gorountine
for i := 0; i < 100; i++ {
w.Add(1)
go func() {
defer w.Done()
// 虽然有多个gorountine执行，但是只有一个gorountine会真正执行initConfig函数
once.Do(initConfig) //
}()
}
w.Wait()
}
执行后你会看到只打印了一次 "执行初始化配置文件!"；而不是10次；说明once有效；
3. sync.Map 线程安全的map
通用的map它是非线程安全的，多个gorountine下会有问题;
比如下面代码：
package main
import (
"fmt"
"strconv"
"sync"
)
var w sync.WaitGroup
var m = make(map[string]int) // 通用map 非线程安全
func main() {
// 添加10个 gorountine
for i := 0; i < 10; i++ {
w.Add(1)
go func() {
defer w.Done()
for j := 0; j < 10; i++ {
key := strconv.Itoa(i) // 转整数为字符串
m[key] = j // 设置值
fmt.Printf("key: %s; val: %d\n", key, j)
}
}()
}
w.Wait()
}
会报错fatal error: concurrent map writes 不允许并发写map
sync.Map 提供了Store/Load 用于存/取等方法，正确代码如下：
package main
import (
"fmt"
"strconv"
"sync"
)
var (
w sync.WaitGroup
m sync.Map // 线程安全map
)
func main() {
// 添加10个 gorountine
for i := 0; i < 10; i++ {
w.Add(1)
go func() {
defer w.Done()
for j := 0; j < 10; j++ {
key := strconv.Itoa(j) // 转整数为字符串
m.Store(key, j) // 设置值
val, _ := m.Load(key)
fmt.Printf("key: %s; val: %d\n", key, val)
}
}()
}
w.Wait()
}
