sync/atomic包

Value

在GO语言中, 对于小于64位的基本类型和指针类型, 其赋值操作是原子的, 可以不加锁的进行并发赋值. 但对于复杂的数据结构, 无法保证原子的赋值. Go提供了Value对象来原子的对任意类型赋值. 其Store函数源码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
func (v *Value) Store(val any) {
    if val == nil {
        panic("sync/atomic: store of nil value into Value")
    }
    // Value对象实际就是一个interface{}, 而interface{}可以视为一个包含两个指针字段的结构体
    // 其中字段typ表示类型, data表示实际的值
    // 无论val是什么类型, 进入Store函数后, val都是指向实际数据的interface{}对象
    vp := (*ifaceWords)(unsafe.Pointer(v))
    vlp := (*ifaceWords)(unsafe.Pointer(&val))
    for {
        // 由于interface{}对象包含两个字段, 实际上这个函数要做的就是一致性的写入两个字段
        typ := LoadPointer(&vp.typ)
        if typ == nil {
            // 首先获取当前对象的type字段, 如果为空, 说明之前从未写入过任何数据, 这是第一次尝试写入数据
            runtime_procPin()   // 此函数关闭Go语言的协程调度
            // 首先尝试CAS更新type字段, 将其设置为更新的标记
            if !CompareAndSwapPointer(&vp.typ, nil, unsafe.Pointer(&firstStoreInProgress)) {
                // CAS失败, 等价于其他协程获得锁, 当前协程进入自旋
                runtime_procUnpin()
                continue
            }
            // CAS成功等价于当前协程获得锁, 因此执行更新操作
            StorePointer(&vp.data, vlp.data)
            StorePointer(&vp.typ, vlp.typ)
            runtime_procUnpin()
            return
        }
        // 如果当前的type标记为更新状态
        // 说明其他协程正在更新, 因此当前协程进行自旋等待
        // 等待其他协程更新完毕后, 当前协程可直接更新data部分
        if typ == unsafe.Pointer(&firstStoreInProgress) {
            continue
        }
        // 如果type不是nil, 说明之前已经设置过类型, 此时进行判断
        // Value对象不能存储不同类型的数据
        if typ != vlp.typ {
            panic("sync/atomic: store of inconsistently typed value into Value")
        }
        // 类型相同时, 更新操作值需要更新一个data字段, 此时直接更新即可
        StorePointer(&vp.data, vlp.data)
        return
    }
}


// ifaceWords is interface{} internal representation.
type ifaceWords struct {
    typ  unsafe.Pointer
    data unsafe.Pointer
}

上述接口非常巧妙地利用了Go语言中interface{}对象的结构, 用一种简单的方式实现了对象的原子赋值

最后更新： 2024年04月29日 13:25

版权声明：本文为原创文章，转载请注明出处

原始链接： https://lizec.top/2023/07/30/Go%E8%AF%AD%E8%A8%80%E7%AC%94%E8%AE%B0%E4%B9%8B%E6%BA%90%E7%A0%81%E5%88%86%E6%9E%90/