WebSocket 长链接
目前为止基本上现有的所有http服务器，使用 同一个端口 都能够同时支持 长链接和短链接
项目地址
Websocket相关文件
index-ws.html
irisws-goroutine.go
irisws.go
README.md
- index-ws.html 文件是使用 HTML5 写的长链接客户端，静态页面双击打开即可向服务端自动发起链接
<!DOCTYPE html>
<html>
<head>
<meta http-equiv="Content-Type" content="text/html; charset=UTF-8">
<title>WebSocket测试</title>
<script type="text/javascript">
class WS {
constructor() {
// 定时器ID用来执行网络重新链接
this.intervalId = null;
this.initConnect();
}
/**
* 初始化链接配置
*/
initConnect() {
// 初始化 WebSocket 对象
this.webSocket = new WebSocket(`ws://localhost:8080/websocket`);
// 监听 WebSocket 连接异常的回调信息
this.webSocket.onerror = (evt) => {
console.log('%c%s', 'color:red;', '=============连接异常=============');
};
// 监听 WebSocket 服务端未连接的回调信息
this.webSocket.onclose = (evt) => {
console.log('%c%s', 'color:blue;', '=============网络连接已断开=============');
if (!this.intervalId) {
// 尝试重新链接
this.intervalId = setInterval(() => {
console.log('%c%s', 'color:blue;', '=============正在尝试重新连接=============');
this.initConnect();
}, 3000);
}
}
// 监听 WebSocket 连接成功的回调信息
this.webSocket.onopen = (evt) => {
console.log('%c%s', 'color:purple;', '=============服务器已连接=============');
if (this.intervalId) {
// 服务器链接成功，清除重链定时器
window.clearInterval(this.intervalId);
this.intervalId = null;
}
}
// 监听 WebSocket 接收消息
this.webSocket.onmessage = (evt) => {
this.onmessage(evt.data);
}
}
/**
* WebSocket 接收消息回调方法
* @param msg
*/
onmessage(msg) {
document.body.innerText = msg
console.log('%c%s', 'color:green;', msg);
this.send(`你好我是 Client ${new Date().toLocaleString()}`);
}
/**
* WebSocket 发送消息方法
* @param callback
*/
send(info) {
this.webSocket.send(info);
}
}
</script>
</head>
<body onload="new WS();">
</body>
</html>
- irisws.go 文件是使用 go 语言编写的基于Iris服务器的长链接服务端，使用
gobwas/ws模块实现WebSocket
- 基于回调函数的设计编写
package ws
import (
"github.com/kataras/iris/v12"
"github.com/kataras/iris/v12/websocket"
"github.com/kataras/neffos"
"log"
)
type IrisWebSocket struct {
ws *neffos.Server
}
// NewWebSocket 初始化 WebSocket
func NewWebSocket(app *iris.Application) *IrisWebSocket {
i := &IrisWebSocket{}
// 使用gobwas/ws模块实现WebSocket
i.ws = websocket.New(websocket.DefaultGobwasUpgrader, websocket.Events{
// 将回调函数暴露给使用者
websocket.OnNativeMessage: i.ReceiveMessage,
})
// 监听服务是否链接
i.ws.OnConnect = func(c *websocket.Conn) error {
log.Printf("[%s] 服务已连接!", c.ID())
return nil
}
// 监听链接是否断开
i.ws.OnDisconnect = func(c *websocket.Conn) {
log.Printf("[%s] 链接已断开!", c.ID())
}
// 设置WebSocket链接地址
app.Any("/websocket", websocket.Handler(i.ws))
return i
}
// SendMessage 发送消息
func (i *IrisWebSocket) SendMessage(message string) {
i.ws.Broadcast(nil, websocket.Message{
Body: []byte(message), // 消息体
IsNative: true, // 使用自己的格式传递数据
})
}
// ReceiveMessage 接收客户消息
func (i *IrisWebSocket) ReceiveMessage(nsConn *websocket.NSConn, msg websocket.Message) error {
log.Printf("接收到客户端消息: %s from [%s]", msg.Body, nsConn.Conn.ID())
return nil
}
- irisws-goroutine.go 文件是使用 go 语言编写的基于Iris服务器的长链接服务端，使用
gorilla/websocket模块实现WebSocket
- 基于Go协程的设计编写
package ws
import (
gorilla "github.com/gorilla/websocket"
"github.com/kataras/iris/v12"
"github.com/kataras/iris/v12/websocket"
"github.com/kataras/neffos"
irisgorilla "github.com/kataras/neffos/gorilla"
"log"
"net/http"
)
type IrisWebSocketGoroutine struct {
ws *neffos.Server
ReceiveMessage chan []byte // 接收消息的通道
SendMessage chan string // 推送消息的通道
}
// NewWebSocketGoroutine 初始化 WebSocket
func NewWebSocketGoroutine(app *iris.Application) *IrisWebSocketGoroutine {
i := &IrisWebSocketGoroutine{
ReceiveMessage: make(chan []byte),
SendMessage: make(chan string),
}
// 使用gorilla/websocket模块实现WebSocket
u := irisgorilla.Upgrader(gorilla.Upgrader{
// 允许跨域
CheckOrigin: func(r *http.Request) bool {
return true
},
})
i.ws = websocket.New(u, websocket.Events{
websocket.OnNativeMessage: func(nsConn *websocket.NSConn, msg websocket.Message) error {
// 将接收到的消息发送到通道中
i.ReceiveMessage <- msg.Body
return nil
},
})
// 监听服务是否链接
i.ws.OnConnect = func(c *websocket.Conn) error {
log.Printf("[%s] 服务已连接!", c.ID())
return nil
}
// 监听链接是否断开
i.ws.OnDisconnect = func(c *websocket.Conn) {
log.Printf("[%s] 链接已断开!", c.ID())
}
i.ws.OnUpgradeError = func(err error) {
log.Printf("Upgrade Error: %v", err)
}
// 开启协程，监听通道是否接收到消息
go func() {
for true {
select {
case msgChan := <-i.ReceiveMessage:
log.Printf("接收到客户端消息: %s", msgChan)
}
}
}()
// 开启协程，监听通道是否要推送消息
go func() {
for true {
select {
case msgChan := <-i.SendMessage:
// 推送消息
i.ws.Broadcast(nil, websocket.Message{
Body: []byte(msgChan), // 消息体
IsNative: true, // 使用自己的格式传递数据
})
}
}
}()
// 设置WebSocket链接地址
app.Get("/websocket", websocket.Handler(i.ws))
return i
}
测试使用 main_test.go
package main
import (
"github.com/kataras/iris/v12"
"iris-server/commons/tools"
"iris-server/commons/ws"
"testing"
"time"
)
// 测试WebSocket
func TestWebSocket(t *testing.T) {
app := iris.Default()
// 初始化 WebSocket
i := ws.NewWebSocket(app)
// 定时推送消息
go tools.SetInterval(1e9, func() {
i.SendMessage(time.Now().Format("2006-01-02 15:04:05"))
})
// 启动服务
app.Listen(":8080")
}
// 测试WebSocket
func TestGoroutineWebSocket(t *testing.T) {
app := iris.Default()
// 初始化 WebSocket
i := ws.NewWebSocketGoroutine(app)
// 定时推送消息
go tools.SetInterval(1e9, func() {
i.SendMessage <- time.Now().Format("2006-01-02 15:04:05")
})
// 启动服务
app.Listen(":8080")
}
go服务端程序启动后，打开index-ws.html文件会自动与go服务端建立长链接，之后查看控制台，go服务端程序会不断的向index-ws.html端发送信息。