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()
+
+	})
 }

modules/base/middleware/websocket.go

@@ -0,0 +1,71 @@
+package middleware
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/lxzan/gws"
+)
+
+const PingInterval = 10 * time.Second
+
+type Handler struct {
+	gws.BuiltinEventHandler
+}
+
+func (c *Handler) OnOpen(socket *gws.Conn) {
+
+}
+
+func (c *Handler) OnPing(socket *gws.Conn, payload []byte) {
+	_ = socket.SetDeadline(time.Now().Add(2 * PingInterval))
+	_ = socket.WritePong(nil)
+}
+
+func (c *Handler) OnPong(socket *gws.Conn, payload []byte) {}
+
+func (c *Handler) OnMessage(socket *gws.Conn, gwsmessage *gws.Message) {
+	fmt.Println(gwsmessage.Bytes())
+
+}
+
+// RemoveSocket 移除WebSocket连接
+
+func messeunpack(id int64, x []byte) {
+
+	// json := gjson.ParseBytes(x)
+	// json.ForEach(func(key, value gjson.Result) bool {
+	// 	fmt.Println(key, value)
+	// 	numeric := regexp.MustCompile(`\d`).MatchString(value.String())
+
+	// 	var ob int
+	// 	if numeric { //如果正确  则使用常量
+
+	// 	} else { //如果错误，则使用变量，即从数据库查询常量定义值是多少
+
+	// 	}
+
+	// 	work.User_script(uint(id), strings.Trim(message.Data.String(), " "))
+	// 	// switch s:=value.String() {
+	// 	// case regexp.MustCompile(`\d`).MatchString(s):
+
+	// 	// }
+
+	// 	return true
+	// })
+	// json.Get("1").String()
+	// var ob int
+	// numeric := regexp.MustCompile(`\d`).MatchString(json.Get("1").String())
+	// if numeric { //如果正确  则使用常量
+	// 	ob, _ = strconv.Atoi(json.Get("1").String())
+	// } else { //如果错误，则使用变量，即从数据库查询常量定义值是多少
+	// 	ob, _ = strconv.Atoi(json.Get("1").String())
+	// }
+	// switch ob {
+	// case 1: //发包函数
+	// 	work.User_script(uint(id), strings.Trim(message.Data.String(), " "))
+	// }
+
+	// Go 1.9.7 onwards only.
+
+}