我在Go中编写了很多并行计算软件。我想从工作线程中收集数据,而我并不确定如何以安全的方式执行此操作。我知道我可以使用频道,但在我的场景中,它们使它变得更复杂,因为我必须以某种方式在主线程中同步消息(等到每个线程发送一些东西)。
方案
主线程创建 n Worker
个实例,并在goroutine中启动它们的work()
方法,以便每个工作者都在自己的线程中运行。每10秒,主线程应从工作者处收集一些简单值(例如迭代计数)并打印统一统计量。
问题
从工人那里读取价值是否安全?主线程只读取值,每个单独的线程都会写入自己的值。如果读取时值为几纳秒就可以了。
关于如何以简单的方式实现这一点的任何其他想法?
答案 0 :(得分:8)
如果至少有一个访问是写入,则在没有同步的情况下,没有值对于多个goroutine的并发访问是安全的。您的案例符合列出的条件,因此您必须使用某种同步,否则行为将是未定义的。
如果goroutine想要将值发送给另一个人,则使用频道。您的情况并非如此:您不希望您的员工每10秒发送一次更新,您希望主要goroutine每隔10秒获取状态。
所以在这个例子中我只用sync.RWMutex
来保护数据:当工人想要修改这些数据时,他们必须获得写锁。当主goroutine想要读取这些数据时,它必须获得读锁定。
一个简单的实现可能如下所示:
type Worker struct {
iter int
iterMux sync.RWMutex
}
func (w *Worker) Iter() int {
w.iterMux.RLock()
defer w.iterMux.RUnlock()
return w.iter
}
func (w *Worker) setIter(n int) {
w.iterMux.Lock()
w.iter = n
w.iterMux.Unlock()
}
func (w *Worker) incIter() {
w.iterMux.Lock()
w.iter++
w.iterMux.Unlock()
}
使用此示例Worker
,主goroutine可以使用Worker.Iter()
获取迭代,并且worker本身可以使用Worker.setIter()
或Worker.incIter()
更改/更新迭代时间,没有任何额外的同步。正确使用Worker.iterMux
确保同步。
或者对于迭代计数器,您也可以使用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)
}