2024年Go最全Golang Websocket框架：实时通信的新选择(1)，2024年最新2024Golang面经

最新推荐文章于 2024-07-27 17:47:02 发布

2401_84911322

最新推荐文章于 2024-07-27 17:47:02 发布

阅读量745

点赞数 7

分类专栏：程序员文章标签： go 学习面试

本文链接：https://blog.csdn.net/2401_84911322/article/details/138691163

版权

程序员专栏收录该内容

57 篇文章 0 订阅

订阅专栏

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

使用示例

下面是一个使用Gorilla Websocket的简单示例：

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func echo(w http.ResponseWriter, r \*http.Request) {
    // 将HTTP请求升级为Websocket连接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        // 读取客户端发送的消息
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }

        // 向客户端发送消息
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            break
        }
    }
}

func main() {
    http.HandleFunc("/echo", echo)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

在上面的示例中，我们首先创建了一个upgrader对象，该对象用于将HTTP请求升级为Websocket连接。然后，我们定义了一个echo函数，用于处理Websocket连接的读写操作。最后，在main函数中，我们将echo函数注册为/echo路径上的HTTP处理函数，并通过http.ListenAndServe启动了一个简单的Web服务器。

nhooyr/websocket

nhooyr/websocket是一个使用纯粹Golang实现的Websocket库，它的设计目标是提供一种更简洁和易用的API。

特点

纯粹的Golang实现，无需任何第三方依赖
简洁而灵活的API设计
支持标准的Websocket协议
提供了高级特性，例如分片、压缩等

使用示例

下面是一个使用nhooyr/websocket的简单示例：

package main

import (
    "log"
    "net/http"
    "time"

    "github.com/nhooyr/websocket"
)

func echo(w http.ResponseWriter, r \*http.Request) {
    // 升级HTTP请求为Websocket连接
    conn, err := websocket.Accept(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close(websocket.StatusInternalError, "Internal Server Error")

    for {
        // 从客户端读取消息
        messageType, message, err := conn.Read(nil)
        if err != nil {
            log.Println(err)
            return
        }

        // 向客户端发送消息
        err = conn.Write(r.Context(), messageType, message)
        if err != nil {
            log.Println(err)
            return
        }
    }
}

func main() {
    http.HandleFunc("/echo", echo)
    srv := &http.Server{
        Addr:         ":8080",
        ReadTimeout:  10 \* time.Second,
        WriteTimeout: 10 \* time.Second,
    }
    log.Fatal(srv.ListenAndServe())
}

在上面的示例中，我们首先使用websocket.Accept函数将HTTP请求升级为Websocket连接。然后，我们定义了一个echo函数，用于处理Websocket连接的读写操作。最后，我们创建了一个带有超时设置的http.Server对象，并通过ListenAndServe方法启动了一个Web服务器。