如何在 GO 中发出 OUTGOING websocket 连接请求

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《如何在 GO 中发出 OUTGOING websocket 连接请求》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我找到了很多关于如何在 go 中创建一个接受 ws 连接的 websocket 服务的示例，但没有一个发送它们的示例。 由于 ws 请求最初是作为正常的休息请求执行的，所以我尝试这样的事情

request, err := http.NewRequest("GET", "ws://localhost:8081/ws", nil)
if err != nil {
    fmt.Println(err)
    return
}

_, err = http.DefaultClient.Do(request)
if err != nil {
    fmt.Println(err)
    return
}
client := newClient(*websocket.Conn) // how i can get *websocket.Conn?
go client.Read()
go client.Write()
clientsMap.Add(client)

但这显然不起作用，我收到不支持的协议方案“ws”错误。 如何建立 ws 连接并获取 *websocket.conn？

正确答案

如果您使用的是 gorilla websocket，则可以使用该软件包中的 Dialer.Dial 或 Dialer.DialContext。下面是如何使用它的示例。

dial, _, err := websocket.defaultdialer.dial("ws://localhost:8081/ws", nil)
if err != nil {
    log.fatalln(err)
}
defer dial.close()

err = dial.writemessage(websocket.textmessage, []byte("hello"))
if err != nil {
    log.fatalln(err)
}

err = dial.writemessage(websocket.textmessage, []byte("world"))
if err != nil {
    log.fatalln(err)
}

有关完整代码，您可以查看此。

package main

import (
    "log"
    "net/http"
    "sync"

    "github.com/gorilla/websocket"
)

func main() {
    wg := sync.WaitGroup{}
    wg.Add(2)

    // Server
    go func() {
        defer wg.Done()

        var (
            upgrader = &websocket.Upgrader{}
            stopChan = make(chan struct{})
        )

        http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
            defer func() {
                close(stopChan)
            }()

            upgrade, err := upgrader.Upgrade(w, r, nil)
            if err != nil {
                return
            }
            for {
                msgType, msg, err := upgrade.ReadMessage()
                if err != nil {
                    log.Println("connection closed,", err)
                    return
                }
                log.Println(msgType, string(msg))
            }
        })

        go func() {
            err := http.ListenAndServe("localhost:8081", nil)
            if err != nil {
                log.Fatalln(err)
            }
        }()

        <-stopChan
    }()

    // Dialer (Client)
    go func() {
        defer wg.Done()

        dial, _, err := websocket.DefaultDialer.Dial("ws://localhost:8081/ws", nil)
        if err != nil {
            log.Fatalln(err)
        }
        defer dial.Close()

        err = dial.WriteMessage(websocket.TextMessage, []byte("hello"))
        if err != nil {
            log.Fatalln(err)
        }

        err = dial.WriteMessage(websocket.TextMessage, []byte("world"))
        if err != nil {
            log.Fatalln(err)
        }
    }()

    wg.Wait()
}

// output
// 2022/09/17 07:18:57 1 hello
// 2022/09/17 07:18:57 1 world
// 2022/09/17 07:18:57 connection closed, websocket: close 1006 (abnormal closure): unexpected EOF

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。