关闭

sync/atomic 库使用小结

标签: golang
3124人阅读 评论(0) 收藏 举报
分类:

sync/atomic 库提供了原子操作的支持,原子操作直接有底层CPU硬件支持,因而一般要比基于操作系统API的锁方式效率高些。本文对 sync/atomic 中的基本操作进行一个简单的介绍。

原子增值

用于对变量值进行原子增操作,并返回增加后的值。

e.g.
    var sum uint32 = 100
    var wg sync.WaitGroup
    for i := 0; i < 50; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            //sum += 1                 //1
            atomic.AddUint32(&sum, 1)  //2
        }()
    }
    wg.Wait()
    fmt.Println(sum)

通过对比//1与//2的结果,可以很清楚的看到原子操作起到的作用。使用//1时,可以看到sum的值是不定的,取决于sum的同步访问情况;使用//2时,结果是确定而且正确的,同一时间只有一个goroutine修改sum。

函数原型:

    atomic.AddUint32(addr *uint32, delta uint32) uint32
    atomic.AddUint64(addr *uint64, delta uint64) uint64
    atomic.AddInt32(addr *int32, delta int32) int32
    atomic.AddInt64(addr *int64, delta int64) int64
    atomic.AddUintptr(addr *uintptr, delta uintptr) uintptr

CAS

先比较变量的值是否等于给定旧值,等于旧值的情况下才赋予新值,最后返回新值是否设置成功。

e.g.
    var sum uint32 = 100
    var wg sync.WaitGroup
    for i := uint32(0); i < 100; i++ {
        wg.Add(1)
        go func(t uint32) {
            defer wg.Done()
            atomic.CompareAndSwapUint32(&sum, 100, sum+1)
        }(i)
    }
    wg.Wait()
    fmt.Println(sum)

可以看到sum的值只改变了一次,只有当sum值为100的时候,CAS才将sum的值修改为了sum+1。

函数原型:

atomic.CompareAndSwapUint32(addr *uint32, old, new uint32) bool
atomic.CompareAndSwapUint64(addr *uint64, old, new uint64) bool
atomic.CompareAndSwapInt32(addr *int32, old, new int32) bool
atomic.CompareAndSwapInt64(addr *int64, old, new int64) bool
atomic.CompareAndSwapUintptr(addr *uintptr, old, new uintptr) bool
atomic.CompareAndSwapPointer(addr *unsafe.Pointer, old, new unsafe.Pointer) bool

原子导出值

导出变量当前的值。

函数原型:

atomic.LoadUint32(addr *uint32) uint32
atomic.LoadUint64(addr *uint64) uint64
atomic.LoadInt32(addr *int32) int32
atomic.LoadInt64(addr *int64) int64
atomic.LoadUintptr(addr *uintptr) uintptr
atomic.LoadPointer(addr *unsafe.Pointer) unsafe.Pointer

原子导入值

赋予变量新值,而不管它原来是什么值。

函数原型:

atomic.StoreUint32(addr *uint32, val uint32)
atomic.StoreUint64(addr *uint64, val uint64)
atomic.StoreInt32(addr *int32, val int32)
atomic.StoreInt64(addr *int64, val int64)
atomic.StoreUintptr(addr *uintptr, val uintptr)
atomic.StorePointer(addr *unsafe.Pointer, val unsafe.Pointer)

原子交换值

赋予变量新值,同时返回变量的旧值。

函数原型:

atomic.SwapUint32(addr *uint32,  new uint32) old uint32
atomic.SwapUint64(addr *uint64,  new uint64) old uint64
atomic.SwapInt32(addr *int32,  new int32) old int32
atomic.SwapInt64(addr *int64,  new int64) old int64
atomic.SwapUintptr(addr *uintptr,  new uintptr) old uintptr
atomic.SwapPointer(addr *unsafe.Pointer,  new unsafe.Pointer) old unsafe.Pointer
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:55888次
    • 积分:1376
    • 等级:
    • 排名:千里之外
    • 原创:59篇
    • 转载:0篇
    • 译文:26篇
    • 评论:4条
    文章分类
    最新评论