Go：负的 WaitGroup 计数器

Question

我有点陌生，正在修改我在其他地方找到的代码以满足我的需求。正因为如此，我并不完全理解这里发生了什么，尽管我明白了一般的想法。我正在使用 go 例程运行一些 websocket 客户端，但是我遇到了一个导致程序崩溃的意外错误。当从 websocket 读取消息时出错（检查 readHandler 函数中的 conn.ReadMessage() 函数），我的程序似乎关闭了太多线程（如果这是错误的术语，请原谅）。关于如何解决这个问题的任何想法？我非常感谢任何花时间浏览它的人。提前致谢！package mainimport (    "context"    "fmt"    "os"    "time"    "os/signal"    "syscall"    "sync"    "net/url"    "github.com/gorilla/websocket"    "strconv"    "encoding/json"    "log"    "bytes"    "compress/gzip"    "io/ioutil")// Structstype Ping struct {    Ping    int64   `json:"ping"`}type Pong struct {    Pong        int64       `json:"pong"`}type SubParams struct {    Sub         string          `json:"sub"`    ID          string          `json:"id"`}func InitSub(subType string, pair string, i int) []byte {    var idInt string = "id" + strconv.Itoa(i)    subStr := "market." + pair + "." + subType    sub := &SubParams{        Sub: subStr,        ID: idInt,    }    out, err := json.MarshalIndent(sub, "", " ")    if err != nil {        log.Println(err);    }    //log.Println(string(out))    return out}// main funcfunc main() {    var server string = "api.huobi.pro"    pairs := []string{"btcusdt", "ethusdt", "ltcusdt"}    comms := make(chan os.Signal, 1)    signal.Notify(comms, os.Interrupt, syscall.SIGTERM)    ctx := context.Background()    ctx, cancel := context.WithCancel(ctx)    var wg sync.WaitGroup    for x, pair := range pairs {        wg.Add(1)        go control(server, "ws", pair, ctx, &wg, x+1)    }    <-comms    cancel()    wg.Wait()}func control(server string, path string, pair string, ctx context.Context, wg *sync.WaitGroup, i int) {    fmt.Printf("Started control for %s\n", server)    url := url.URL {        Scheme: "wss",        Host: server,        Path: path,    }    fmt.Println(url.String())    conn, _, err := websocket.DefaultDialer.Dial(url.String(), nil)    if err != nil {        panic(err)    }}

Answer 1

readHandler连接失败时跳出循环：

  _, p, err :=  conn.ReadMessage()

  if err != nil {

      wg.Done()

      fmt.Println(err)

      return // <--- add this line

  }

如果没有返回，函数会在一个紧密的循环中旋转，读取错误，直到出现恐慌。

在 goroutine 的开头使用defer wg.Done()以确保 Done 只被调用一次。

func readHandler(ctx context.Context, conn *websocket.Conn, wg *sync.WaitGroup, server string) {

    defer wg.Done()

    for {

      select {

      case <-ctx.Done():

          return

      default:

          _, p, err := conn.ReadMessage()

          if err != nil {

              fmt.Println(err)

              return

          }

     ...

也更新control功能。

因为调用者不会与 并发执行任何代码，所以运行goroutinereadHander没有任何价值。readHandler删除所有对等待组的引用readHandler并直接调用该函数：更改go readHandler(ctx, conn, &localwg, server)为readHandler(ctx, conn, server).

还有更多问题，但这应该会让你走得更远。