如何使用 gorilla 连接到 websocket?

时间:2021-02-14 14:12:15

标签: go websocket gorilla

我想看一个关于如何使用 gorilla 连接到 websocket 的最小和简单示例。

我发现连接到 websocket 的唯一示例是 this, 但我无法理解它,也找不到它是如何工作的解释。

编辑:

第 20 行:
为什么他把 websocket 地址保存在 flag.String 中而不是变量字符串中?

第 26 行:
这部分是否创建了一个信号,当用户按下 crtl + C 时会通知程序?

interrupt: = make(chan os.Signal, 1)
signal.Notify (interrupt, os.Interrupt)

第 32 行:
这条线有什么作用?

websocket.DefaultDialer.Dial (u.String (), nil)

第 60 行:
因为 []byte(t.String()) 而不是 t.string()

1 个答案:

答案 0 :(得分:0)

样本的 README 说:

<块引用>

客户端每秒发送一条消息并打印所有收到的消息。

看代码,后面的部分(打印收到的所有消息)发生在这里:

go func() {
    defer close(done)
    for {
        _, message, err := c.ReadMessage()
        if err != nil {
            log.Println("read:", err)
            return
        }
        log.Printf("recv: %s", message)
    }
}()

这会启动一个 goroutine,它循环读取来自 c(websocket 连接)的消息并打印出来。它在读取失败时停止,这会发生在错误和连接关闭时。

主协程这样做:

ticker := time.NewTicker(time.Second)
defer ticker.Stop()

for {
    select {
    case <-done:
        return
    case t := <-ticker.C:
        err := c.WriteMessage(websocket.TextMessage, []byte(t.String()))
        if err != nil {
            log.Println("write:", err)
            return
        }
    case <-interrupt:
        log.Println("interrupt")

        // Cleanly close the connection by sending a close message and then
        // waiting (with timeout) for the server to close the connection.
        err := c.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
        if err != nil {
            log.Println("write close:", err)
            return
        }
        select {
        case <-done:
        case <-time.After(time.Second):
        }
        return
    }
}

它每秒都会触发一个自动收报机,然后向 websocket 发送一条消息。当读取器 goroutine 完成后,它关闭 done 通道,该通道也用 select 通知此循环退出。

最后,有一个“中断”选择来处理 Ctrl+C,干净地停止客户端。


回答您的具体问题:

第 20 行:此标志使您可以在执行程序时从命令行设置地址,而不是对其进行硬编码。例如,您可以像 client -addr localhost:9044 一样调用它来设置不同的端口。另见https://gobyexample.com/command-line-flags

第 26 行:是的,这是针对 Ctrl+C;另见https://gobyexample.com/signals

第 32 行:DefaultDialerwebsocket 包中的 preconfigured 拨号程序。根据文档,它相当于

var DefaultDialer = &Dialer{
    Proxy:            http.ProxyFromEnvironment,
    HandshakeTimeout: 45 * time.Second,
}

第 60 行:WriteMessage 采用 []byte,因此您必须将字符串转换为 []byte