feat：完善项目

app/api/websocket.go

@@ -3,6 +3,8 @@ package api
 import (
 	"ArmedPolice/app/handle"
 	"ArmedPolice/app/service"
+	"ArmedPolice/config"
+	"errors"
 	"github.com/gin-gonic/gin"
 	"github.com/gorilla/websocket"
 	"net/http"
@@ -25,8 +27,18 @@ func (*Websocket) Ws(c *gin.Context) {
 		APIFailure(err)(c)
 		return
 	}
-	session := getSession()(c).(*service.Session)
+	token := c.Query(config.APIRequestToken)
 
+	if token == "" {
+		APIFailure(errors.New("Token异常"))(c)
+		return
+	}
+	session := new(service.Session)
+
+	if session, err = service.NewAuthToken(token).Auth(); err != nil {
+		APIFailure(err)(c)
+		return
+	}
 	client := service.NewWebsocket(session.UIDToString(), conn)
 	service.HubMessage.RegisterHandle(client)
 	go client.Write()