Golang中Gorilla Websocket并发使用时的故障排查

Golang中Gorilla Websocket并发使用时的故障排查 我有一个使用Gorilla WebSocket的应用程序，地址是：“GitHub - gorilla/websocket: gorilla/websocket包是一个快速、经过良好测试且广泛使用的Go语言WebSocket实现。” 我的应用程序会打开连接，发送一个股票代码列表，然后在一个循环中读取通过该连接返回的消息，直到连接被取消或出错。 使用Go协程，我希望同时监控多个股票列表。当打开第二个连接且第一个连接正在读取消息时，会发生错误。在我的代码中，我通过迭代编号来标识协程，由于Go协程的工作方式，无法保证哪个连接会先被尝试。运行此代码时出现错误，因此我提取了必要的代码来演示该问题。我对WebSocket还比较陌生，所以很可能是我操作有误。我将发送示例代码和结果：

package main

import (
	"fmt"
	"github.com/gorilla/websocket"
	"net/url"
	"sync"
)

type Stock struct {
	Symbol string `json:"symbol"`
	Type   string `json:"type"`
}

func main() {
	// WebSocket连接的URL和令牌

	var wg *sync.WaitGroup = &sync.WaitGroup{}

	for i := 0; i < 2; i++ {
		wg.Add(1)
		go processStocks(i, wg)
	}

	wg.Wait()
}

func processStocks(interationNumber int, wg *sync.WaitGroup) {
	iterationStr := fmt.Sprintf("%d", interationNumber)

	u := url.URL{Scheme: "wss", Host: "ws.finnhub.io", RawQuery: "token=cbbb00iad3ibhoa1vbcg"}

	fmt.Printf("begin processStocks() for iteration: %s\n", iterationStr)

	// 股票代码列表
	stocks := []Stock{
		{"IBM", "subscribe"},
		// 在此处添加更多股票
	}

	defer wg.Done()

	// 创建新的WebSocket连接
	fmt.Printf("begin Dial() for iteration: %s\n", iterationStr)
	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
	if err != nil {
		//cAddress := fmt.Sprintf("%p", c)
		fmt.Printf("failed dial error for iteration:%s address:%p: %v\n", iterationStr, c, err)
	} else {
		fmt.Printf("successful connection at address: %p for iteration: %s \n", c, iterationStr)
	}
	defer func(c *websocket.Conn) {
		err := c.Close()
		if err != nil {
			fmt.Printf("iteration %s Close error: %v\n", iterationStr, err)
		}
	}(c)

	// 发送股票代码列表
	for _, stock := range stocks {
		fmt.Printf("begin WriteJSON() for iteration: %s address: %p\n", iterationStr, c)
		err := c.WriteJSON(stock)
		if err != nil {
			fmt.Printf("iteration %s write error: %v\n", iterationStr, err)
			return
		} else {
			fmt.Printf("successful write at address: %p for iteration: %s \n", c, iterationStr)
		}

	}

	// 无限循环读取结果
	for {
		fmt.Printf("begin ReadMessage() for iteration:%s address:%p\n", iterationStr, c)
		_, message, err := c.ReadMessage()
		if err != nil {
			fmt.Printf("iteration %s read error: %v address:%p: \n", iterationStr, err, c)
			return
		} else {
			fmt.Printf("iteration %s successful read: [%s] address:%p: \n", iterationStr, message, c)
		}
	}
}

输出：

begin processStocks() for iteration: 1
begin Dial() for iteration: 1
begin processStocks() for iteration: 0
begin Dial() for iteration: 0
successful connection at address: 0x14000012580 for iteration: 0 
begin WriteJSON() for iteration: 0 address: 0x14000012580
successful write at address: 0x14000012580 for iteration: 0 
begin ReadMessage() for iteration:0 address:0x14000012580

*iteration 0 read error: websocket: close 1006 (abnormal closure): unexpected EOF address:0x14000012580:* 

successful connection at address: 0x140000126e0 for iteration: 1 
begin WriteJSON() for iteration: 1 address: 0x140000126e0
successful write at address: 0x140000126e0 for iteration: 1 
begin ReadMessage() for iteration:1 address:0x140000126e0
iteration 1 successful read: [{"data":[{"c":["1","12"],"p":144.995,"s":"IBM","t":1691083530560,"v":1}],"type":"trade"}] address:0x140000126e0: 
begin ReadMessage() for iteration:1 address:0x140000126e0

顺便提一下，这里面包含我的密钥，你可以用它来重现问题，不过我很快就会更改它，反正它是免费的。