kobe6258 1 settimana fa
parent
commit
84bc102187
3 ha cambiato i file con 2 aggiunte e 15 eliminazioni
  1. 1 8
      services/llm/facade/llm_service.go
  2. 0 5
      services/ws_service.go
  3. 1 2
      utils/ws/session.go

+ 1 - 8
services/llm/facade/llm_service.go

@@ -11,17 +11,10 @@ var (
 	deepseekService, _ = llm.GetInstance(llm.LLM_DEEPSEEK)
 )
 
-
-
 func GenerateSessionCode() (code string) {
 	return fmt.Sprintf("%s%s", "llm_session_", uuid.NewUUID().Hex32())
 }
 
-func GetSession(userId int, sessionId string) (session *ws.Session, ok bool) {
-	token := fmt.Sprintf("%d_%s", userId, sessionId)
-	return ws.Manager().Get(token)
-}
-
 func AddSession(session *ws.Session) {
-	ws.Manager().Add(session)
+	ws.Manager().AddSession(session)
 }

+ 0 - 5
services/ws_service.go

@@ -4,19 +4,14 @@ import (
 	"eta/eta_api/models"
 	"eta/eta_api/models/system"
 	"eta/eta_api/utils"
-	"eta/eta_api/utils/ws"
 	"fmt"
 	"github.com/beego/beego/v2/server/web"
 	"github.com/beego/beego/v2/server/web/context"
-	"github.com/gorilla/websocket"
 	"net/http"
 	"strings"
 	"time"
 )
 
-func HandleWebSocketConnection(conn *websocket.Conn) {
-	ws.Manager().HandleWebSocketConnection(conn)
-}
 func WsAuthenticate() web.FilterFunc {
 	return func(ctx *context.Context) {
 		method := ctx.Input.Method()

+ 1 - 2
utils/ws/session.go

@@ -32,13 +32,12 @@ func (s *Session) readPump() {
 			handleCloseError(err)
 			return
 		}
-
 		// 更新活跃时间
 		s.mu.Lock()
 		s.LastActive = time.Now()
 		s.mu.Unlock()
 		// 处理消息
-		if err = manager.HandleMessage(s.ID, message); err != nil {
+		if err = manager.HandleMessage(s.UserID, s.ID, message); err != nil {
 
 		}
 	}