从不同的线程读取值

时间:2018-03-23 07:48:33

标签: multithreading go concurrency thread-safety goroutine

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

方案

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

问题

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

关于如何以简单的方式实现这一点的任何其他想法?

1 个答案:

答案 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)
}