Go：负的 WaitGroup 计数器

Question

我有点陌生，正在修改我在其他地方找到的代码以满足我的需求。正因为如此，我并不完全理解这里发生了什么，尽管我明白了一般的想法。

我正在使用 go 例程运行一些 websocket 客户端，但是我遇到了一个导致程序崩溃的意外错误。当从 websocket 读取消息时出错（检查 readHandler 函数中的 conn.ReadMessage() 函数），我的程序似乎关闭了太多线程（如果这是错误的术语，请原谅）。关于如何解决这个问题的任何想法？我非常感谢任何花时间浏览它的人。提前致谢！

package main

import (

    "context"

    "fmt"

    "os"

    "time"

    "os/signal"

    "syscall"

    "sync"

    "net/url"

    "github.com/gorilla/websocket"

    "strconv"

    "encoding/json"

    "log"

    "bytes"

    "compress/gzip"

    "io/ioutil"

)

// Structs

type 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 func

func 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连接失败时跳出循环：&nbsp; _, p, err :=&nbsp; conn.ReadMessage()&nbsp; if err != nil {&nbsp; &nbsp; &nbsp; wg.Done()&nbsp; &nbsp; &nbsp; fmt.Println(err)&nbsp; &nbsp; &nbsp; return // <--- add this line&nbsp; }如果没有返回，函数会在一个紧密的循环中旋转，读取错误，直到出现恐慌。在 goroutine 的开头使用defer wg.Done()以确保 Done 只被调用一次。func readHandler(ctx context.Context, conn *websocket.Conn, wg *sync.WaitGroup, server string) {&nbsp; &nbsp; defer wg.Done()&nbsp; &nbsp; for {&nbsp; &nbsp; &nbsp; select {&nbsp; &nbsp; &nbsp; case <-ctx.Done():&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; return&nbsp; &nbsp; &nbsp; default:&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; _, p, err := conn.ReadMessage()&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; if err != nil {&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; fmt.Println(err)&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; return&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; }&nbsp; &nbsp; &nbsp;...也更新control功能。因为调用者不会与 并发执行任何代码，所以运行goroutinereadHander没有任何价值。readHandler删除所有对等待组的引用readHandler并直接调用该函数：更改go readHandler(ctx, conn, &localwg, server)为readHandler(ctx, conn, server).还有更多问题，但这应该会让你走得更远。