小编典典

从其他线程读取值

go

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

情境

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

从工人那里读取价值是否安全?主线程将仅读取值,而每个单独的线程将写入其自己的值。如果读取时这些值相差几纳秒就可以了。

关于如何轻松实现此目标的其他想法?


阅读 254

收藏
2020-07-02

共1个答案

小编典典

在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)
}
2020-07-02