sync包与锁
锁
为什么我们需要锁?
比如执行以下代码的时候
func main() {
a:=int64(0)
for i := 0; i < 1000; i++ {
go func() {
a++
}()
}
time.Sleep(time.Second)
fmt.Println(a)
}
开启1000个协程,每个协程让a加1,等待1s使全部执行完毕后输出a。我们期望的a是1000,但结果总是a<1000。
因为a++其实包含了不止一步操作。假设一开始a=0,协程0先读取a=0,然后协程1也读取到a=0,然后协程0执行a=0+1,之后协程1也执行a=0+1,结果两个协程执行完成后a=1。
为了解决这种情况我们需要保证a++在同一时刻只有一个协程在执行,或者在a++没成功执行后可以重试a++。这两种解决方案对应悲观锁和乐观锁。
乐观锁
atomic包下的下提供了一系列原子操作的函数,底层使用硬件的原子指令(比如CAS Compare-And-Swap)保证操作的原子性
比如把上面例子的a++换成atomic.AddInt64(&a, 1),最后得到的结果就是a=1000。
拿int64举例,原子操作有以下几种方法:
atomic.AddInt64(addr *int64, delta int64) int64
//*addr+=delta
//return *addr
atomic.LoadInt64(addr *int64) int64
//return *addr
atomic.StoreInt64(addr *int64, value int64)
//*addr=value
atomic.SwapInt64(addr *int64, new int64) int64
//old:=*addr
//*addr=new
//return old
atomic.CompareAndSwapInt64(addr *int64, old