✨ github oauth2 login /gitee oauth2 login

api/websocket_api/gws_api.go

@@ -0,0 +1,104 @@
+package websocket_api
+
+import (
+	"fmt"
+	"github.com/gin-gonic/gin"
+	"github.com/lxzan/gws"
+	"net/http"
+	"schisandra-cloud-album/global"
+	"time"
+)
+
+const (
+	PingInterval         = 5 * time.Second  // 客户端心跳间隔
+	HeartbeatWaitTimeout = 10 * time.Second // 心跳等待超时时间
+)
+
+func (WebsocketAPI) NewGWSServer(c *gin.Context) {
+	var handler = NewWebSocket()
+	upgrader := gws.NewUpgrader(handler, &gws.ServerOption{
+		HandshakeTimeout: 5 * time.Second, // 握手超时时间
+		ReadBufferSize:   1024,            // 读缓冲区大小
+		ParallelEnabled:  true,            // 开启并行消息处理
+		Recovery:         gws.Recovery,    // 开启异常恢复
+		CheckUtf8Enabled: true,            // 开启UTF8校验
+		PermessageDeflate: gws.PermessageDeflate{
+			Enabled: true, // 开启压缩
+		},
+		Authorize: func(r *http.Request, session gws.SessionStorage) bool {
+			var clientId = r.URL.Query().Get("client_id")
+			if clientId == "" {
+				return false
+			}
+			session.Store("client_id", clientId)
+			return true
+		},
+	})
+	socket, err := upgrader.Upgrade(c.Writer, c.Request)
+	if err != nil {
+		return
+	}
+	go func() {
+		socket.ReadLoop() // 此处阻塞会使请求上下文不能顺利被GC
+	}()
+}
+func MustLoad[T any](session gws.SessionStorage, key string) (v T) {
+	if value, exist := session.Load(key); exist {
+		v = value.(T)
+	}
+	return
+}
+
+func NewWebSocket() *WebSocket {
+	return &WebSocket{
+		sessions: gws.NewConcurrentMap[string, *gws.Conn](64, 128),
+	}
+}
+
+type WebSocket struct {
+	gws.BuiltinEventHandler
+	sessions *gws.ConcurrentMap[string, *gws.Conn] // 使用内置的ConcurrentMap存储连接, 可以减少锁冲突
+}
+
+func (c *WebSocket) OnOpen(socket *gws.Conn) {
+	name := MustLoad[string](socket.Session(), "client_id")
+	if conn, ok := c.sessions.Load(name); ok {
+		conn.WriteClose(1000, []byte("connection is replaced"))
+	}
+	c.sessions.Store(name, socket)
+	global.LOG.Printf("%s connected\n", name)
+}
+
+func (c *WebSocket) OnClose(socket *gws.Conn, err error) {
+	name := MustLoad[string](socket.Session(), "client_id")
+	sharding := c.sessions.GetSharding(name)
+	c.sessions.Delete(name)
+	sharding.Lock()
+	defer sharding.Unlock()
+
+	global.LOG.Printf("onerror, name=%s, msg=%s\n", name, err.Error())
+}
+
+func (c *WebSocket) OnPing(socket *gws.Conn, payload []byte) {
+	_ = socket.SetDeadline(time.Now().Add(PingInterval + HeartbeatWaitTimeout))
+	_ = socket.WritePong(payload)
+}
+
+func (c *WebSocket) OnPong(socket *gws.Conn, payload []byte) {}
+
+func (c *WebSocket) OnMessage(socket *gws.Conn, message *gws.Message) {
+	defer message.Close()
+	name := MustLoad[string](socket.Session(), "client_id")
+	if conn, ok := c.sessions.Load(name); ok {
+		_ = conn.WriteMessage(gws.OpcodeText, message.Bytes())
+	}
+}
+
+// SendMessageToClient 向指定客户端发送消息
+func (c *WebSocket) SendMessageToClient(clientId string, message []byte) error {
+	conn, ok := c.sessions.Load(clientId)
+	if ok {
+		return conn.WriteMessage(gws.OpcodeText, message)
+	}
+	return fmt.Errorf("client %s not found", clientId)
+}

api/websocket_api/websocket_api.go

@@ -1,11 +1,11 @@
 package websocket_api
 
 import (
-	"fmt"
 	"github.com/gin-gonic/gin"
 	"github.com/gorilla/websocket"
 	"log"
 	"net/http"
+	"schisandra-cloud-album/common/result"
 	"schisandra-cloud-album/global"
 	"sync"
 	"time"
@@ -40,9 +40,7 @@ func (WebsocketAPI) DeleteClient(context *gin.Context) {
 		}
 		deleteClient(id)
 	} else {
-		context.JSON(http.StatusOK, gin.H{
-			"mesg": "未找到该客户端",
-		})
+		result.FailWithMessage("客户端不存在", context)
 	}
 	// 关闭其消息通道
 	_, exist = getMsgChannel(id)
@@ -86,7 +84,7 @@ func WsHandler(w http.ResponseWriter, r *http.Request, id string) {
 	var err error
 	var exist bool
 	// 创建一个定时器用于服务端心跳
-	//pingTicker := time.NewTicker(time.Second * 10)
+	pingTicker := time.NewTicker(time.Second * 10)
 	conn, err = wsupgrader.Upgrade(w, r, nil)
 	if err != nil {
 		global.LOG.Println(err)
@@ -110,7 +108,7 @@ func WsHandler(w http.ResponseWriter, r *http.Request, id string) {
 	for {
 		_, _, err := conn.ReadMessage()
 		if err != nil {
-			fmt.Println(err)
+			global.LOG.Error(err)
 			return
 		}
 		select {
@@ -118,30 +116,29 @@ func WsHandler(w http.ResponseWriter, r *http.Request, id string) {
 			// 从消息通道接收消息，然后推送给前端
 			err = conn.WriteJSON(content)
 			if err != nil {
-				global.LOG.Error(err)
-				err := conn.Close()
+				err = conn.Close()
 				if err != nil {
 					return
 				}
 				deleteClient(id)
 				break
 			}
-			//case <-pingTicker.C:
-			//	// 服务端心跳:每20秒ping一次客户端，查看其是否在线
-			//	err := conn.SetWriteDeadline(time.Now().Add(time.Second * 20))
-			//	if err != nil {
-			//		return
-			//	}
-			//	err = conn.WriteMessage(websocket.PongMessage, []byte("pong"))
-			//	if err != nil {
-			//		log.Println("send pong err:", err)
-			//		err := conn.Close()
-			//		if err != nil {
-			//			return
-			//		}
-			//		deleteClient(id)
-			//		return
-			//	}
+		case <-pingTicker.C:
+			// 服务端心跳:每20秒ping一次客户端，查看其是否在线
+			err := conn.SetWriteDeadline(time.Now().Add(time.Second * 20))
+			if err != nil {
+				return
+			}
+			err = conn.WriteMessage(websocket.PingMessage, []byte{})
+			if err != nil {
+				log.Println("send pong err:", err)
+				err := conn.Close()
+				if err != nil {
+					return
+				}
+				deleteClient(id)
+				return
+			}
 		}
 	}