concurrency - 在 Go 中分配一个指针是原子的吗？

Question

在 Go 中分配一个指针是原子的吗？

我需要在锁中分配一个指针吗？假设我只想将指针分配给 nil，并希望其他线程能够看到它。我知道在 Java 中我们可以为此使用 volatile，但 Go 中没有 volatile。

Answer 1

在 go 中唯一保证是原子的是 sync.atomic 中的操作。 .

所以如果你想确定你要么需要锁，例如sync.Mutex或使用其中一种原子原语。我不建议使用原子原语，因为你必须在任何使用指针的地方使用它们，而且它们很难正确使用。

使用 mutex 是 OK 的风格——你可以定义一个函数来返回当前指针并非常容易地锁定，例如类似的东西

import "sync"

var secretPointer *int
var pointerLock sync.Mutex

func CurrentPointer() *int {
    pointerLock.Lock()
    defer pointerLock.Unlock()
    return secretPointer
}

func SetPointer(p *int) {
    pointerLock.Lock()
    secretPointer = p
    pointerLock.Unlock()
}

这些函数将指针的副本返回给它们的客户，即使主指针发生变化，该副本也将保持不变。这可能会或可能不会被接受，具体取决于您的要求的时间紧迫性。这应该足以避免任何未定义的行为 - 垃圾收集器将确保指针始终保持有效，即使指向的内存不再被您的程序使用。

另一种方法是仅从一个 go 例程进行指针访问，并使用 channel 命令该 go 例程执行操作。这将被认为是更惯用的 go，但可能不完全适合您的应用程序。

更新

Here is an example显示如何使用 atomic.SetPointer。由于使用了 unsafe.Pointer，它相当难看。然而，unsafe.Pointer 将 compile 强制转换为空，因此运行时成本很小。

import (
    "fmt"
    "sync/atomic"
    "unsafe"
)

type Struct struct {
    p unsafe.Pointer // some pointer
}

func main() {
    data := 1

    info := Struct{p: unsafe.Pointer(&data)}

    fmt.Printf("info is %d\n", *(*int)(info.p))

    otherData := 2

    atomic.StorePointer(&info.p, unsafe.Pointer(&otherData))

    fmt.Printf("info is %d\n", *(*int)(info.p))

}