Goroutines、回调和 sync.WaitGroup答案

【问题标题】：Goroutines, Callbacks and sync.WaitGroupGoroutines、回调和 sync.WaitGroup
【发布时间】：2017-08-13 13:17:03
【问题描述】：

使用以下代码：

package main

import (
    "github.com/davecgh/go-spew/spew"
    "sync"
    "time"
)
func callbackWithTimeout(cbFunc func() ([]byte, error), timeout time.Duration) {
    defer wg.Done() //I don't want this function to know about sync.WaitGroup
    time.Sleep(timeout)
    d, e := cbFunc()
    spew.Dump(d)
    spew.Dump(e)
}
var wg sync.WaitGroup
func main() {
    wg.Add(1)
    go func() {
        cbFunc := func() ([]byte, error) {
            //I feel like I should be able to defer here instead
            return nil, nil
        }
        callbackWithTimeout(cbFunc, time.Duration(4*time.Second))
    }()
    println("some line")
    wg.Wait()
}

在函数callbackWithTimeout中，我不想使用defer wg.Done()，因为callbackWithTimeout()与wg.Done()无关。我该如何实施这样的事情？即，删除callbackWithTimeout 中的任何sync.WaitGroup？我在理解这里的关注点分离时有点问题，因为回调函数不应该知道等待组，但在这种情况下，我似乎别无选择？

我觉得这应该是调用者的责任到wg.Done()（在这种情况下是cbFunc），但缺乏对文档的简明参考或关于如何在 Go 中实现它的想法因为根据定义，回调函数所做的只是回调函数。那么，我哪里做错了？

在重构过程中，您确实做出了愚蠢的假设。下面的工作代码。非常感谢。

package main

import (
    "errors"
    "github.com/davecgh/go-spew/spew"
    "sync"
    "time"
)

func callbackWithTimeout(cbFunc func() ([]byte, error), timeout time.Duration) {
    time.Sleep(timeout)
    d, e := cbFunc()
    spew.Dump(d)
    spew.Dump(e)
}

func main() {
    var wg sync.WaitGroup
    wg.Add(1)
    go func() {
        defer wg.Done()
        callbackWithTimeout(func() ([]byte, error) {
            b := []byte{1, 2, 3, 4}
            e := errors.New("error123")
            return b, e
        }, time.Duration(2*time.Second))
    }()
    println("some line")
    wg.Wait()
}

【问题讨论】：

为什么不直接去 func() { defer wg.Done()
这绝对是疯了。谢谢你..我的愚蠢错误。我的原始代码有go func() { defer wg.Done() .......，但在重构过程中，我做了一些愚蠢的假设。

标签： go callback goroutine

【解决方案1】：

可能是这样的？

package main

import (
    "sync"
    "time"

    "github.com/davecgh/go-spew/spew"
)

func callbackWithTimeout(cbFunc func() ([]byte, error), timeout time.Duration) {
    time.Sleep(timeout)
    d, e := cbFunc()
    spew.Dump(d)
    spew.Dump(e)
}

func main() {
    var wg sync.WaitGroup

    wg.Add(1)

    go func() {
        defer wg.Done() // move it here
        cbFunc := func() ([]byte, error) {
            //I feel like I should be able to defer here instead
            return nil, nil
        }
        callbackWithTimeout(cbFunc, time.Duration(4*time.Second))
    }()

    println("some line")

    wg.Wait()
}

【讨论】：