Go语言实现WebSocket客户端的连接等待与重连策略

package main

import (
    "fmt"
    "log"
    "time"

    "golang.org/x/net/websocket" // 推荐使用此包，原问题中的 "websocket" 是旧版本或自定义包
)

func main() {
    origin := "http://localhost:8080/"
    url := "ws://localhost:8080/ws"

    var err error
    var ws *websocket.Conn // 声明 ws 变量，使其作用域覆盖整个循环
    for {
        fmt.Println("尝试连接WebSocket服务器...")
        ws, err = websocket.Dial(url, "", origin)
        if err != nil {
            fmt.Printf("连接失败: %v，将在1秒后重试...\n", err)
            time.Sleep(1 * time.Second) // 暂停一段时间，避免忙循环
            continue                   // 继续下一次循环，再次尝试连接
        }
        fmt.Println("WebSocket连接成功！")
        break // 连接成功，跳出循环
    }

    // 连接成功后，可以进行数据发送或接收
    if _, err := ws.Write([]byte("Hello from client!")); err != nil {
        log.Fatalf("发送数据失败: %v", err) // 使用 Fatalf 确保在发送失败时程序退出
    }
    fmt.Println("数据发送成功！")

    // 实际应用中，这里会有一个持续的数据读写循环
    // 为了演示，我们简单地等待几秒然后关闭连接
    time.Sleep(5 * time.Second)
    if err := ws.Close(); err != nil {
        log.Printf("关闭连接失败: %v", err)
    }
    fmt.Println("连接已关闭。")
}

go get golang.org/x/net/websocket

go run main.go

package main

import (
    "fmt"
    "log"
    "time"

    "golang.org/x/net/websocket"
)

// connectAndListen 负责连接和监听WebSocket消息
func connectAndListen(url, origin string) *websocket.Conn {
    var ws *websocket.Conn
    var err error
    for {
        fmt.Println("尝试连接WebSocket服务器...")
        ws, err = websocket.Dial(url, "", origin)
        if err != nil {
            fmt.Printf("连接失败: %v，将在1秒后重试...\n", err)
            time.Sleep(1 * time.Second)
            continue
        }
        fmt.Println("WebSocket连接成功！")
        return ws
    }
}

func main() {
    origin := "http://localhost:8080/"
    url := "ws://localhost:8080/ws"

    for {
        ws := connectAndListen(url, origin) // 建立或重连连接

        // 启动一个goroutine来处理数据接收
        go func(conn *websocket.Conn) {
            var msg []byte
            for {
                err := websocket.Message.Receive(conn, &msg)
                if err != nil {
                    // 接收失败，通常意味着连接已断开
                    log.Printf("接收消息失败，连接可能已断开: %v", err)
                    conn.Close() // 显式关闭连接
                    return       // 退出接收goroutine
                }
                fmt.Printf("收到消息: %s\n", msg)
            }
        }(ws)

        // 示例：发送一条消息
        if _, err := ws.Write([]byte("Hello from client!")); err != nil {
            log.Printf("发送数据失败: %v", err)
            // 如果发送失败，也可能意味着连接已断开，需要重连
            ws.Close()
            time.Sleep(1 * time.Second) // 短暂等待，避免立即重连导致忙循环
            continue // 继续主循环，触发重连
        }
        fmt.Println("数据发送成功！")

        // 模拟客户端工作一段时间，如果连接断开，上面的接收goroutine会退出，
        // 外部的for循环会检测到并重连。
        // 在实际应用中，这里会有更多的业务逻辑和心跳机制。
        select {
        case <-time.After(30 * time.Second): // 假设连接可以存活30秒
            fmt.Println("30秒已过，模拟连接可能断开，尝试重连...")
            ws.Close() // 模拟连接断开
            time.Sleep(1 * time.Second)
            // continue // 外部for循环会处理重连
        }
    }
}