优化结构

internal/server/handler/session.go

@@ -1,6 +1,7 @@
 package handler
 
 import (
+	"kra/internal/server/middleware"
 	"kra/internal/service"
 
 	"github.com/gin-gonic/gin"
@@ -10,9 +11,10 @@ type Session struct{ tokens *service.TokenService }
 
 func NewSession(tokens *service.TokenService) *Session { return &Session{tokens: tokens} }
 func (h *Session) Logout(c *gin.Context) {
-	token := c.GetHeader("x-token")
+	token := middleware.RequestToken(c, false)
 	if token == "" {
-		token, _ = c.Cookie("x-token")
+		NoAuth(c, "未登录或非法访问")
+		return
 	}
 	if err := h.tokens.BlacklistToken(c.Request.Context(), token); err != nil {
 		Fail(c, "jwt作废失败")