feat(base): 添加 WebSocket 支持

- 在 go.mod 中引入 gws 库
- 在 middleware.go 中实现 WebSocket 升级处理
- 添加 WebSocket 连接的授权和读取循环逻辑

modules/base/middleware/middleware.go

@@ -2,10 +2,12 @@ package middleware
 
 import (
 	"blazing/modules/base/config"
+	"fmt"
 	"net/http"
 
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/lxzan/gws"
 )
 
 func MiddlewareCORS(r *ghttp.Request) {
@@ -39,5 +41,41 @@ func init() {
 	if config.Config.Middleware.Log.Enable {
 		g.Server().BindMiddleware("/admin/*", BaseLog)
 	}
+	g.Server().BindHandler("/ws", func(r *ghttp.Request) {
 
+		upgrader := gws.NewUpgrader(new(Handler), &gws.ServerOption{
+			//CompressEnabled: true,
+
+			// 在querystring里面传入用户名
+			// 把Sec-WebSocket-Key作为连接的key
+
+			// 刷新页面的时候, 会触发上一个连接的OnClose/OnError事件, 这时候需要对比key并删除map里存储的连接
+			Authorize: func(rt *http.Request, session gws.SessionStorage) bool {
+				//r.s\\\
+				//	r.Get("t")
+				//admin := cool.GetAdmin(r.Context())
+				// var name = r.URL.Query().Get("name")
+				// if name == "" {
+				// 	return false
+				// }
+				//	t, _ := service.NewBaseSysUserService().Person(admin.UserID)
+
+				//g.Log().Debug(context.TODO(), t.Mimi)
+				//	session.Store("name", t.Mimi)
+				//session.Store("key", r.Header.Get("Sec-WebSocket-Key"))
+				return true
+			},
+		})
+
+		socket, err := upgrader.Upgrade(r.Response.Writer, r.Request)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		// ants.Submit(func() {
+		// 	socket.ReadLoop()
+		// })
+		go socket.ReadLoop()
+
+	})
 }