101 lines
2.4 KiB
Go
101 lines
2.4 KiB
Go
package websocketServer
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/gorilla/websocket"
|
||
"net/http"
|
||
webServer "Framework/webServer"
|
||
"strconv"
|
||
"sync"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// 处理HTTP请求
|
||
func httpHandler(cxt *webServer.Context) {
|
||
msg := "test msg"
|
||
cxt.GetResponseWriter().Write([]byte(msg))
|
||
}
|
||
|
||
// websocket连接事件
|
||
func onConnFunc(ctx *Context) {
|
||
fmt.Printf("new conn, ip: %s %p\n", ctx.GetWebServerContext().GetRequestIP(), ctx)
|
||
ctx.SendMessage(websocket.TextMessage, []byte("Hello")) // 返回数据
|
||
}
|
||
|
||
// websocket关闭事件
|
||
func onCloseFunc(ctx *Context) {
|
||
fmt.Printf("close conn, ip: %s %p\n", ctx.GetWebServerContext().GetRequestIP(), ctx)
|
||
}
|
||
|
||
// websocket接收事件
|
||
func onMsgFunc(ctx *Context, msgType int, msgData []byte) {
|
||
fmt.Printf("%p ", ctx)
|
||
fmt.Println("msg ip:", ctx.GetWebServerContext().GetRequestIP(), "msg type:", msgType, "msg:", string(msgData))
|
||
ctx.SendMessage(msgType, []byte("SVR: "+string(msgData))) // 返回数据
|
||
|
||
// 主动关闭连接测试
|
||
if string(msgData) == "close" {
|
||
ctx.SendMessage(msgType, []byte("close")) // 返回数据
|
||
//time.Sleep(time.Microsecond * 100)
|
||
ctx.Close()
|
||
}
|
||
}
|
||
|
||
func TestWsServer(t *testing.T) {
|
||
var ssl bool = false
|
||
var server IServer
|
||
if ssl {
|
||
server = NewWssServer(":22222", "cert.pem", "key.pem", false)
|
||
} else {
|
||
server = NewWsServer(":22222", false)
|
||
}
|
||
|
||
server.SetIfDelegate(false)
|
||
|
||
// 注册HTTP处理函数
|
||
server.RegisterHandler("/http", httpHandler, &webServer.HandlerConfig{})
|
||
|
||
// 设置websocket允许跨域访问
|
||
server.GetUpgrader().CheckOrigin = func(r *http.Request) bool {
|
||
return true
|
||
}
|
||
//upgrader := &websocket.Upgrader{
|
||
// // 允许跨域
|
||
// CheckOrigin: func(r *http.Request) bool {
|
||
// return true
|
||
// },
|
||
//}
|
||
//server.SetUpgrader(upgrader)
|
||
|
||
// 注册websocket处理事件
|
||
server.RegisterWebsocketHandler("/websocket", &EventCallbackFuncs{
|
||
OnConnFunc: onConnFunc,
|
||
OnCloseFunc: onCloseFunc,
|
||
OnMsgFunc: onMsgFunc,
|
||
}, &webServer.HandlerConfig{})
|
||
|
||
// 设置广播并发数
|
||
server.SetBroadcastConcurrent(2)
|
||
|
||
// 设置接收到Ping消息时，自动回复Pong信息
|
||
server.SetAutoPong(true)
|
||
|
||
// 设置心跳检测信息（心跳检测周期10秒；缺少2个心跳周期断开连接）
|
||
server.SetHeartbeatDetectInfo(2, time.Second*10)
|
||
|
||
// 广播测试
|
||
go func() {
|
||
var i = 0
|
||
for {
|
||
time.Sleep(time.Second * 10)
|
||
// 广播测试
|
||
server.BroadcastMessage(websocket.TextMessage, []byte(strconv.Itoa(i)))
|
||
i++
|
||
}
|
||
}()
|
||
|
||
wg := sync.WaitGroup{}
|
||
server.Start(&wg)
|
||
}
|