使用超时机制优雅地建立Go WebSocket连接

package main

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

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

func main() {
    origin := "http://localhost/"
    urlStr := "ws://remote-server:1234" // 替换为你的WebSocket服务器地址

    // 设置连接超时时间
    timeout := 5 * time.Second

    // 使用net.DialTimeout建立连接
    conn, err := net.DialTimeout("tcp", getHostPort(urlStr), timeout)
    if err != nil {
        fmt.Println("连接超时:", err)
        return
    }
    defer conn.Close()

    // 将字符串URL转换为URL结构
    u, err := url.Parse(urlStr)
    if err != nil {
        fmt.Println("URL解析失败:", err)
        return
    }

    // 使用websocket.NewClient创建WebSocket客户端
    config := &websocket.Config{
        Location: u,
        Origin:   origin,
        Version:  websocket.ProtocolVersionHybi13,
    }

    ws, err := websocket.NewClient(config, conn)
    if err != nil {
        fmt.Println("创建WebSocket客户端失败:", err)
        return
    }
    defer ws.Close()

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

    // 在这里进行WebSocket通信
    // 例如：发送消息、接收消息等

}

//getHostPort 从URL中提取主机和端口
func getHostPort(urlStr string) string {
    u, err := url.Parse(urlStr)
    if err != nil {
        return ""
    }
    return u.Host
}