16

在他对这个问题的回答中： Golang for Windows 不稳定的行为？ 用户@distributed 建议在并发 goroutine 上锁定/同步对共享变量的访问。

我怎样才能做到这一点？

更多关于这个问题：

views我同时在多个 goroutine上运行此代码（返回的带有闭包的函数）：

func makeHomeHandler() func(c *http.Conn, r *http.Request) {
    views := 1
    return func(c *http.Conn, r *http.Request) {
        fmt.Fprintf(c, "Counting %s, %d so far.", r.URL.Path[1:], views)
        views++
    }
}

看起来 IO 函数需要时间，结果我得到了这种输出：

Counting monkeys, 5 so far.
Counting monkeys, 5 so far.
Counting monkeys, 5 so far.
Counting monkeys, 8 so far.
Counting monkeys, 8 so far.
Counting monkeys, 8 so far.
Counting monkeys, 11 so far.

它递增得很好，但是当它被打印出来时，我可以看到打印+递增操作根本不是原子的。

如果我将其更改为：

func makeHomeHandler() func(c *http.Conn, r *http.Request) {
    views := 0
    return func(c *http.Conn, r *http.Request) {
        views++
        // I can only hope that other goroutine does not increment the counter 
        // at this point, i.e., right after the previous line and before the 
        // next one are executed!
        views_now := views
        fmt.Fprintf(c, "Counting %s, %d so far.", r.URL.Path[1:], views_now)
    }
}

它似乎工作正常，但我不完全确定它是否最终不会失败......

4

2 回答 2

25

如果您想要一个同步的计数器，那么使用sync.Mutex是规范的解决方案。sync/atomic 包应该只用于低级别的东西，或者当您测量到严重的性能问题时。

type Counter struct {
    mu  sync.Mutex
    x   int64
}

func (c *Counter) Add(x int64) {
    c.mu.Lock()
    c.x += x
    c.mu.Unlock()
}

func (c *Counter) Value() (x int64) {
    c.mu.Lock()
    x = c.x
    c.mu.Unlock()
    return
}

func makeHomeHandler() func(c http.ResponseWriter, r *http.Request) {
    var views Counter
    return func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Counting %s, %d so far.", r.URL.Path[1:], views.Value())
        views.Add(1)
    }
}

对于您的特定问题，我建议定义一个满足 http.Handler 接口的新类型，而不是返回一个闭包。这看起来也更简单：

type homeHandler struct {
    mu  sync.Mutex
    views   int64
}

func (h *homeHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    h.mu.Lock()
    defer h.mu.Unlock()
    fmt.Fprintf(w, "Counting %s, %d so far.", r.URL.Path[1:], h.views)
    h.views++
}

func init() {
    http.Handle("/", new(homeHandler))
}
于 2012-05-24T10:34:12.113 回答
10

同步包有一些同步原语。根据问题，您可以使用 RWMutex 或普通的 Mutex。

如果您想要更具体的答案，请提供有关其用途的更多信息。

编辑：阅读链接问题后，您可能正在寻找sync/atomic，尽管 Mutex 也可以。

Edit2：我看到你用一个例子更新了你的帖子。这是使用同步/原子的代码。

func makeHomeHandler() func(w http.ResponseWriter, r *http.Request) {
    var views *uint64 = new(uint64)
    atomic.StoreUint64(views, 0) // I don't think this is strictly necessary
    return func(w http.ResponseWriter, r *http.Request) {
        // Atomically add one to views and get the new value
        // Perhaps you want to subtract one here
        views_now := atomic.AddUint64(views, 1) 
        fmt.Fprintf(w, "Counting %s, %d so far.", r.URL.Path[1:], views_now)
    }
}

（注意：我没有测试过上面的，所以可能有错别字/brainfarts）我现在测试了。

于 2012-05-23T23:07:54.830 回答