Go 快速起步：创建 WebSocket 服务器（聊天室）

最新推荐文章于 2024-09-25 20:17:13 发布

ato'ng

最新推荐文章于 2024-09-25 20:17:13 发布

阅读量1.2k

点赞数 3

分类专栏： Go 文章标签： websocket 网络协议网络

本文链接：https://blog.csdn.net/Phplayers/article/details/125471726

版权

本文介绍了如何使用Go语言和gorilla/websocket库创建WebSocket服务器，以实现即时聊天室。首先，讲解了WebSocket协议与HTTP协议的区别，强调其双向通信特性。接着，详细阐述了简单即时聊天室的实现，包括建立连接、推送用户信息以及消息传递。然后，进阶讨论了并发即时聊天室的设计，利用goroutines和channels进行通信。最后，提到了前端实现及与其它实时聊天解决方案的对比总结。

摘要由CSDN通过智能技术生成

先了解 WebSocket 协议

和 HTTP 协议一样，WebSocket 协议也建立在 TCP/IP 协议基础上，但不一样的是 HTTP 协议为单向协议，即只能客户端向服务器请求资源，服务器才能将数据传送给浏览器。

而 WebSocket 协议是一种双向通信协议，在建立连接后，WebSocket 服务器和 Browser/UA 都能主动的向对方发送或接收数据。

WebSocket 可以认为是与 HTTP 属于同一个层级的兄弟协议，但实际上它又有些依赖 HTTP 协议，因为 WebSocket 的初始连接是通过 HTTP 协议的三次握手实现，完成连接之后再向服务器请求升级成为 websocket 协议；

简单即时聊天室的实现思路

引入了 github.com/gorilla/websocket 包，先创建 http 服务器，然后升级为 websocket 服务器；
当用户打开聊天页面，则建立连接，并将用户信息推送给ws服务端，服务端注册该用户到连接池；
用户点击发送消息时，循环用户连接池拿到用户的连接，主动将消息推送给每个用户；
这个思路比较简单，只通过 ReadMessage 接口通讯，所以需要为每个消息约定一下消息类型，比如注册、注销、消息）；

简单版的 serv.go

package main

// go get -u -v github.com/gorilla/websocket 安装第三方包
import (
	"encoding/json"
	"fmt"
	"github.com/gorilla/websocket"
	"net/http"
)

func main() {
	// http请求，跳转到一个简单的前端页面
	http.HandleFunc("/", func(writer http.ResponseWriter, request *http.Request) {
		http.ServeFile(writer, request, "home.html")
	})
	// http请求，连接成功后申请升级为websocket服务
	http.HandleFunc("/ws", indexHandler)

	err := http.ListenAndServe("0.0.0.0:8888", nil)
	if err != nil {
		return
	}
}

// 消息体
type replyMsg struct {
	Type string `json:"type,omitempty"`
	Uid  string `json:"uid,omitempty"`
	Msg  string `json:"msg,omitempty"`
}

var (
	upgrader = websocket.Upgrader{
		// 允许所有CORS跨域请求
		CheckOrigin: func(r *http.Req

最低0.47元/天解锁文章