您现在的位置是：首页 > 文章详情

nettyws v1.0.3 发布，高性能 Websocket 库

日期：2023-11-29点击：229收藏

go-netty-ws 是一款基于 go-netty 编写的高性能 Websocket 服务器 & 客户端库。

基于事件回调的API设计，简单易用
高性能（压测数据参考go-websocket-benchmark）
支持压缩协议
支持从标准库升级，无缝衔接

更新记录

v1.0.3
- 修复压缩相关问题，提升与浏览器交互兼容性

安装

 go get github.com/go-netty/go-netty-ws@latest

API 预览

 type Websocket func NewWebsocket(options ...Option) *Websocket func (ws *Websocket) Close() error func (ws *Websocket) Listen(addr string) error func (ws *Websocket) Open(addr string) error func (ws *Websocket) UpgradeHTTP(w http.ResponseWriter, r *http.Request) (Conn, error) type Option func WithAsyncWrite(writeQueueSize int, writeForever bool) Option func WithBinary() Option func WithBufferSize(readBufferSize, writeBufferSize int) Option func WithCompress(compressLevel int, compressThreshold int64) Option func WithMaxFrameSize(maxFrameSize int64) Option func WithNoDelay(noDelay bool) Option func WithServeMux(serveMux *http.ServeMux) Option func WithServeTLS(tls *tls.Config) Option func WithValidUTF8() Option

简单使用案例：

服务端示例:

 // 创建一个websocket实例 var ws = nettyws.NewWebsocket() // 设置OnOpen事件处理器，当连接建立时被调用 ws.OnOpen = func(conn nettyws.Conn) { fmt.Println("OnOpen: ", conn.RemoteAddr()) } // 设置OnData事件处理器，当收到一个消息时被调用 ws.OnData = func(conn nettyws.Conn, data []byte) { fmt.Println("OnData: ", conn.RemoteAddr(), ", message: ", string(data)) conn.Write(data) } // 设置OnClose事件处理器，当收到连接关闭时被调用 ws.OnClose = func(conn nettyws.Conn, err error) { fmt.Println("OnClose: ", conn.RemoteAddr(), ", error: ", err) } fmt.Println("listening websocket connections ....") // 启动端口监听，接受新的连接 if err := ws.Listen("ws://127.0.0.1:9527/ws"); nil != err { panic(err) }

客户端示例:

 // 创建websocket实例 var ws = nettyws.NewWebsocket() // 设置OnOpen事件处理器，当连接建立时被回调 ws.OnOpen = func(conn nettyws.Conn) { fmt.Println("OnOpen: ", conn.RemoteAddr()) conn.Write([]byte("hello world")) } // 设置OnData事件处理器，当连接收到消息时被回调 ws.OnData = func(conn nettyws.Conn, data []byte) { fmt.Println("OnData: ", conn.RemoteAddr(), ", message: ", string(data)) } // 设置OnClose事件处理器，当连接关闭时被回调 ws.OnClose = func(conn nettyws.Conn, err error) { fmt.Println("OnClose: ", conn.RemoteAddr(), ", error: ", err) } fmt.Println("open websocket connection ...") // 连接到远程服务器 if err := ws.Open("ws://127.0.0.1:9527/ws"); nil != err { panic(err) }

从标准Http服务器升级:

 // 创建websocket实例 var ws = nettyws.NewWebsocket() // 设置OnOpen事件处理器，当连接建立时被调用 ws.OnOpen = func(conn nettyws.Conn) { fmt.Println("OnOpen: ", conn.RemoteAddr()) } // 设置OnData事件处理器，当连接收到消息时被调用 ws.OnData = func(conn nettyws.Conn, data []byte) { fmt.Println("OnData: ", conn.RemoteAddr(), ", message: ", string(data)) conn.Write(data) } // 设置OnClose事件处理器，当连接关闭时被回调 ws.OnClose = func(conn nettyws.Conn, err error) { fmt.Println("OnClose: ", conn.RemoteAddr(), ", error: ", err) } fmt.Println("upgrade websocket connections ....") // 绑定http路由处理函数，并从http升级到websocket serveMux := http.NewServeMux() serveMux.HandleFunc("/ws", func(writer http.ResponseWriter, request *http.Request) { ws.UpgradeHTTP(writer, request) }) // 启动端口监听，接受客户端请求 if err := http.ListenAndServe(":9527", serveMux); nil != err { panic(err) }