Go WebSocket 连接超时处理的惯用方法

package main

import (
    "fmt"
    "net"
    "net/url"
    "time"

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

func main() {
    // 设置连接超时时间为 5 秒
    dialTimeout := 5 * time.Second

    // WebSocket 服务器地址
    wsURL := "wss://remote-server"

    // 解析 URL
    u, err := url.Parse(wsURL)
    if err != nil {
        fmt.Println("URL 解析失败:", err)
        return
    }

    // 使用 net.DialTimeout 建立 TCP 连接
    conn, err := net.DialTimeout("tcp", u.Host+":443", dialTimeout) // 注意端口号，wss默认443端口
    if err != nil {
        fmt.Println("TCP 连接失败:", err)
        return
    }
    defer conn.Close()

    // 构建 Config 对象
    config := &websocket.Config{
        Location: u,
        Origin:   &url.URL{Scheme: "http", Host: "localhost"}, // 替换为你的 Origin
        Version:  websocket.ProtocolVersionHybi13,
    }

    // 使用 websocket.NewClient 函数创建 WebSocket 连接
    ws, err := websocket.NewClient(config, conn)
    if err != nil {
        fmt.Println("WebSocket 连接失败:", err)
        return
    }
    defer ws.Close()

    fmt.Println("WebSocket 连接成功!")

    // 在这里可以进行 WebSocket 通信
    // 例如：
    // var msg string = "Hello, WebSocket!"
    // err = websocket.Message.Send(ws, msg)
    // if err != nil {
    //  fmt.Println("发送消息失败:", err)
    //  return
    // }

    // var reply string
    // err = websocket.Message.Receive(ws, &reply)
    // if err != nil {
    //  fmt.Println("接收消息失败:", err)
    //  return
    // }

    // fmt.Println("接收到的消息:", reply)
}