代码之家 › 专栏 › 技术社区 › user5682282

从其他线程读取值

goroutine thread-safety concurrency go multithreading

user5682282 · 技术社区 · 8 年前

我正在用Go编写一个可以进行大量并行计算的软件。我想从工作线程收集数据,但我不确定如何以安全的方式进行。我知道我可以使用通道,但在我的场景中,通道会使事情变得更加复杂,因为我必须以某种方式在主线程中同步消息(等待每个线程都发送一些消息)。

脚本

主线程创建 n Worker 实例并启动其 work() 方法,以便每个工作线程都在自己的线程中运行。主线程应该每隔10秒从工作线程收集一些简单的值(例如迭代计数),并打印一个合并的统计数据。

问题

从工人那里读取值是否安全?主线程将只读取值,每个线程将写入自己的值。读取时,如果值为几纳秒,就可以了。

关于如何轻松实现这一点,还有其他想法吗?

1 回复 | 直到 8 年前

icza 6 年前

在Go中,如果至少有一个访问是写访问,那么对于来自多个GOroutine的并发访问,没有任何值是安全的,而无需同步。您的案例满足列出的条件,因此必须使用某种同步,否则行为将无法定义。

如果goroutine要将值发送给另一个,则使用通道。你的情况并非如此:你不希望你的员工每10秒钟发送一次更新,你希望你的主要goroutine 取来每10秒显示一次状态。

因此,在本例中,我将使用 sync.RWMutex :当工作人员想要修改此数据时,他们必须获取写锁。当主goroutine想要读取此数据时,它必须获取读取锁。

一个简单的实现可以如下所示:

type Worker struct {
    iterMu sync.RWMutex
    iter   int
}

func (w *Worker) Iter() int {
    w.iterMu.RLock()
    defer w.iterMu.RUnlock()
    return w.iter
}

func (w *Worker) setIter(n int) {
    w.iterMu.Lock()
    w.iter = n
    w.iterMu.Unlock()
}

func (w *Worker) incIter() {
    w.iterMu.Lock()
    w.iter++
    w.iterMu.Unlock()
}

使用此示例 Worker ,主goroutine可以使用 Worker.Iter() ,并且工作程序本身可以使用 Worker.setIter() 或 Worker.incIter() 在任何时候,无需任何额外的同步。通过正确使用 Worker.iterMu 。

或者,对于迭代计数器,您也可以使用 sync/atomic 包裹如果选择此选项,则只能使用的函数读取/修改迭代计数器 atomic 包装如下:

type Worker struct {
    iter int64
}

func (w *Worker) Iter() int64 {
    return atomic.LoadInt64(&w.iter)
}

func (w *Worker) setIter(n int64) {
    atomic.StoreInt64(&w.iter, n)
}

func (w *Worker) incIter() {
    atomic.AddInt64(&w.iter, 1)
}