我正在用Go编写执行大量并行计算的软件。我想从工作线程中收集数据,但我不太确定如何以安全的方式进行数据收集。我知道我可以使用通道,但是在我的场景中,它们使它变得更加复杂,因为我必须以某种方式同步主线程中的消息(等到每个线程发送了一些东西)。
情境
主线程创建 n个 Worker实例,并work()在goroutine中启动其方法,以便每个工作线程都在各自的线程中运行。主线程每10秒应从工作线程中收集一些简单值(例如,迭代计数)并打印合并的统计信息。
Worker
work()
题
从工人那里读取价值是否安全?主线程将仅读取值,而每个单独的线程将写入其自己的值。如果读取时这些值相差几纳秒就可以了。
关于如何轻松实现此目标的其他想法?
在Go中,如果至少有一次访问是写操作,则没有值可安全地从多个goroutine进行并发访问而无需同步。您的案例符合列出的条件,因此您必须使用某种同步,否则行为将是不确定的。
如果goroutine要向另一个发送值,则使用通道。您的情况不完全是这样:您不希望您的工作人员每10秒发送一次更新,您希望主goroutine 每10秒 获取一次 状态。
因此,在此示例中,我仅使用sync.RWMutex:来保护数据:当工作人员想要修改此数据时,他们必须获得写锁定。当主goroutine想要读取此数据时,它必须获得一个读取锁。
sync.RWMutex
一个简单的实现可能看起来像这样:
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。
Worker.Iter()
Worker.setIter()
Worker.incIter()
Worker.iterMu
另外,对于迭代计数器,您也可以使用该sync/atomic包。如果选择此选项,则只能使用以下atomic软件包的功能读取/修改迭代计数器:
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) }