Go实现Websocket消息推送
1.websocket 简介
以往浏览器要获取服务端数据，都是通过发送 HTTP 请求，然后等待服务端回应的。也就是说浏览器端一直是整个请求的发起者，只有它主动，才能获取到数据。而要让浏览器一侧能够获取到服务端的实时数据，就需要不停地向服务端发起请求。虽然大多数情况下并没有获取到实际数据，但这大大增加了网络压力，对于服务端来说压力也直线上升。
image.png
由于 WebSocket 是全双工通信，所以当建立了 WebSocket 连接之后，接下来的通信就类似于传统的 TCP 通信了。客户端和服务端可以相互发送数据，不再有实时性的问题。
image.png
image.png
2.开发包的选择
在 Go 官方的 SDK 中，并不包含对 WebSocket 的支持，所以必须使用第三方库。
要使用 Golang 开发 WebSocket，选择基本就在 x/net/websocket 和 gorilla/websocket 之间。《Go Web 编程》一书中的例子使用了x/net/websocket作为开发包，而且貌似它也更加官方且正式。而实际根据我在网上查询得到的反馈看来，并非如此。x/net/websocket貌似 Bug 较多，且较为不稳定，问题解决也并不及时。相比之下，gorilla/websocket则更加优秀。
下载链接及手册
3.推送服务实现
server 启动以后会注册两个 Handler。
websocketHandler 用于提供浏览器端发送 Upgrade 请求并升级为 WebSocket 连接。 pushHandler 用于提供外部推送端发送推送数据的请求。
3-1.go代码的简单实现
package main
import (
"net/http"
"log"
"github.com/gorilla/websocket"
)
var (
upgrader = websocket.Upgrader {
// 读取存储空间大小
ReadBufferSize:1024,
// 写入存储空间大小
WriteBufferSize:1024,
// 允许跨域
CheckOrigin: func(r *http.Request) bool {
return true
},
}
)
func wsHandler(w http.ResponseWriter, r *http.Request) {
var (
wbsCon *websocket.Conn
err error
data []byte
)
// 完成http应答，在httpheader中放下如下参数
if wbsCon, err = upgrader.Upgrade(w, r, nil);err != nil {
return // 获取连接失败直接返回
}
for {
// 只能发送Text, Binary 类型的数据,下划线意思是忽略这个变量.
if _, data, err = wbsCon.ReadMessage();err != nil {
goto ERR // 跳转到关闭连接
}
if err = wbsCon.WriteMessage(websocket.TextMessage, data); err != nil {
goto ERR // 发送消息失败，关闭连接
}
}
ERR:
// 关闭连接
wbsCon.Close()
}
func main() {
// 当有请求访问ws时，执行此回调方法
http.HandleFunc("/ws",wsHandler)
// 监听127.0.0.1:7777
err := http.ListenAndServe("0.0.0.0:7777", nil)
if err != nil {
log.Fatal("ListenAndServe", err.Error())
}
}
3-2.前端代码
群聊天室
3-3. 结果分析
运行的结果
image.png
交互的过程
1.request请求时，要求将协议升级为websocket 参数
2.response返回时，如果成功则返回websocket 参数
image.png
数据传送形式是把数据拆成多个frame进行发送和接收，这个不用我们关心。
image.png
image.png
4.封装Websocket
image.png
image.png
image.png
image.png
1.目录结构
image.png
2.server.go
package main
import (
"net/http"
"log"
"github.com/gorilla/websocket"
"github.com/go-websocket/impl"
"time"
)
var (
upgrader = websocket.Upgrader {
// 读取存储空间大小
ReadBufferSize:1024,
// 写入存储空间大小
WriteBufferSize:1024,
// 允许跨域
CheckOrigin: func(r *http.Request) bool {
return true
},
}
)
func wsHandler(w http.ResponseWriter, r *http.Request) {
var (
wsConn *websocket.Conn
err error
// data []byte
conn *impl.Connection
data []byte
)
// 完成http应答，在httpheader中放下如下参数
if wsConn, err = upgrader.Upgrade(w, r, nil);err != nil {
return // 获取连接失败直接返回
}
if conn, err = impl.InitConnection(wsConn); err != nil {
goto ERR
}
go func() {
var (
err error
)
for {
// 每隔一秒发送一次心跳
if err = conn.WriteMessage([]byte("heartbeat")); err != nil {
return
}
time.Sleep(1 * time.Second)
}
}()
for {
if data, err = conn.ReadMessage(); err != nil {
goto ERR
}
if err = conn.WriteMessage(data); err != nil {
goto ERR
}
}
ERR:
// 关闭当前连接
}
func main() {
// 当有请求访问ws时，执行此回调方法
http.HandleFunc("/ws",wsHandler)
// 监听127.0.0.1:7777
err := http.ListenAndServe("0.0.0.0:7777", nil)
if err != nil {
log.Fatal("ListenAndServe", err.Error())
}
}
3.connection.go
package impl
import (
"github.com/gorilla/websocket"
"sync"
"errors"
)
type Connection struct {
// 存放websocket连接
wsConn *websocket.Conn
// 用于存放数据
inChan chan []byte
// 用于读取数据
outChan chan []byte
closeChan chan byte
mutex sync.Mutex
// chan是否被关闭
isClosed bool
}
// 读取Api
func (conn *Connection) ReadMessage() (data []byte, err error) {
//select是Go中的一个控制结构，类似于用于通信的switch语句。每个case必须是一个通信操作，要么是发送要么是接收。 select随机执行一个可运行的case。如果没有case可运行，它将阻塞，直到有case可运行。一个默认的子句应该总是可运行的。
select {
case data = <- conn.inChan:
case <- conn.closeChan:
err = errors.New("connection is closed")
}
return
}
// 发送Api
func (conn *Connection) WriteMessage(data []byte) (err error) {
select {
case conn.outChan <- data:
case <- conn.closeChan:
err = errors.New("connection is closed")
}
return
}
// 关闭连接的Api
func (conn *Connection) Close() {
// 线程安全的Close，可以并发多次调用也叫做可重入的Close
conn.wsConn.Close()
conn.mutex.Lock()
if !conn.isClosed {
// 关闭chan,但是chan只能关闭一次
close(conn.closeChan)
conn.isClosed = true
}
conn.mutex.Unlock()
}
// 初始化长连接
func InitConnection(wsConn *websocket.Conn) (conn *Connection, err error) {
conn = &Connection{
wsConn: wsConn,
inChan: make(chan []byte, 1000),
outChan: make(chan []byte, 1000),
closeChan: make(chan byte, 1),
}
// 启动读协程
go conn.readLoop()
// 启动写协程
go conn.writeLoop()
return
}
// 内部实现
func (conn *Connection) readLoop() {
var (
data []byte
err error
)
for {
if _, data, err = conn.wsConn.ReadMessage(); err != nil {
goto ERR
}
// 容易阻塞到这里，等待inChan有空闲的位置
select {
case conn.inChan <- data:
case <- conn.closeChan: // closeChan关闭的时候执行
goto ERR
}
}
ERR:
conn.Close()
}
func (conn *Connection) writeLoop() {
var (
data []byte
err error
)
for {
select {
case data = <- conn.outChan:
case <- conn.closeChan:
goto ERR
}
data = <- conn.outChan
if err = conn.wsConn.WriteMessage(websocket.TextMessage, data); err != nil {
goto ERR
}
}
ERR:
conn.Close()
}
5.第三方资源
https://www.imooc.com/video/17599
https://www.cnblogs.com/snowInPluto/p/8688453.html