Go：負WaitGroup計數器

我有點陌生，正在重新撰寫我在其他地方找到的代碼以滿足我的需求。正因為如此，我不完全明白這里發生了什么，雖然我明白了一般的想法。

我正在使用 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)
    }
    subscribe(conn, pair, i)
    defer conn.Close()

    var localwg sync.WaitGroup

    localwg.Add(1)
    go readHandler(ctx, conn, &localwg, server)

    <- ctx.Done()
    localwg.Wait()
    wg.Done()
    return
}

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

        select {

            case <- ctx.Done():
                wg.Done()
                return
            default:
                _, p, err :=  conn.ReadMessage()
                if err != nil {
                    wg.Done()
                    fmt.Println(err)
                }
                r, err := gzip.NewReader(bytes.NewReader(p))
                if(err == nil) {
                    result, err := ioutil.ReadAll(r)
                    if(err != nil) {
                        fmt.Println(err)
                    }
                    d := string(result)
                    fmt.Println(d)

                    var ping Ping
                    json.Unmarshal([]byte(d), &ping)
                    if (ping.Ping > 0) {
                        str := Pong{Pong: ping.Ping}
                        msg, err := json.Marshal(str)
                        if (err == nil) {
                            fmt.Println(string(msg))
                            conn.WriteMessage(websocket.TextMessage, []byte(msg))
                        }
                    }
            }
        }
    }
}

func subscribe(conn *websocket.Conn, pair string, id int) {
    sub := string(InitSub("trade.detail", pair, id))

    err := conn.WriteMessage(websocket.TextMessage, []byte(sub))
    if err != nil {
        panic(err)
    }
}

uj5u.com熱心網友回復：

• readHandler連接失敗時跳出回圈：

    _, p, err :=  conn.ReadMessage()
    if err != nil {
        wg.Done()
        fmt.Println(err)
        return // <--- add this line
    }
  

  如果沒有回傳，函式會在一個緊密的回圈中旋轉讀取錯誤，直到發生恐慌。

• defer wg.Done()在 goroutine 的開頭使用以確保 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功能。

• 因為呼叫者不會與 并發執行任何代碼readHander，所以運行readHandlergoroutine沒有任何價值。從中洗掉對等待組的所有參考readHandler并直接呼叫該函式：更改go readHandler(ctx, conn, &localwg, server)為readHandler(ctx, conn, server).

還有更多問題，但這應該會讓你走得更遠。

標籤：走 网络套接字 等候组

上一篇：無法修改函式中的矩陣

下一篇：Go映射到Json的串列