WebSocket 协议本质上是一个基于 TCP 的协议。
为了建立一个 WebSocket 连接,客户端浏览器首先要向服务器发起一个 HTTP 请求,这个请求和通常的 HTTP 请求不同,包含了一些附加头信息,其中附加头信息"Upgrade: WebSocket"表明这是一个申请协议升级的 HTTP 请求,服务器端解析这些附加的头信息然后产生应答信息返回给客户端,客户端和服务器端的 WebSocket 连接就建立起来了,双方就可以通过这个连接通道自由的传递信息,并且这个连接会持续存在直到客户端或者服务器端的某一方主动的关闭连接。
import (
"encoding/json"
"github.com/gorilla/websocket"
"net/http"
)
// 抽象出需要的数据结构
// ws连接器 数据 管道 // 已经不是http连接,管道传输数据的事情
type connection struct {
// ws 连接器
ws *websocket.Conn
// 管道
send chan []byte
// 数据
data *Data
}
// 抽象ws连接器
// 处理ws中的各种逻辑
type hub struct {
// connections注册了连接器
connections map[*connection]bool
// 从服务器发送的信息
broadcast chan []byte
// 从连接器注册请求
register chan *connection
// 销毁请求
unregister chan *connection
}
var user_list = []string{
}
// 将连接器对象初始化
var h = hub{
// connections注册了连接器
connections: make(map[*connection]bool),
// 从服务器发送的信息
broadcast: make(chan []byte),
// 从连接器注册请求
register: make(chan *connection),
// 销毁请求
unregister: make(chan *connection),
}
// 处理ws的逻辑实现
func (h *hub)run() {
// 监听数据管道,在后端不断处理管道数据
for true {
// 根据不同的数据管道,处理不同的逻辑
select {
// 注册
case c:= <- h.register