From 06d2c7e9e73bccba003ef64f4eb8f52c2d47ca40 Mon Sep 17 00:00:00 2001
From: aozhiwei <aozhiwei@kingsome.cn>
Date: Mon, 4 Nov 2024 18:03:50 +0800
Subject: [PATCH] 1

---
 server/wheelserver/api/v1/user/user.go   | 8 ++++++--
 server/wheelserver/constant/constant.go  | 4 ++++
 server/wheelserver/middleware/jwtauth.go | 5 +++--
 3 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/server/wheelserver/api/v1/user/user.go b/server/wheelserver/api/v1/user/user.go
index f20f0661..152c24c7 100644
--- a/server/wheelserver/api/v1/user/user.go
+++ b/server/wheelserver/api/v1/user/user.go
@@ -1,7 +1,8 @@
 package user
 
 import (
-	//"main/common"
+	"main/constant"
+	"main/common"
 	"github.com/gin-gonic/gin"
 )
 
@@ -9,5 +10,8 @@ type UserApi struct {
 }
 
 func (this *UserApi) Login(c *gin.Context) {
-	//s := c.MustGet("session").(common.Session)
+	s := c.MustGet(constant.SESSION_KEY).(common.Session)
+	if s == nil {
+
+	}
 }
diff --git a/server/wheelserver/constant/constant.go b/server/wheelserver/constant/constant.go
index 21235a19..fcc497a4 100644
--- a/server/wheelserver/constant/constant.go
+++ b/server/wheelserver/constant/constant.go
@@ -16,3 +16,7 @@ const (
 const (
 	PLAT_ID_TG = 5
 )
+
+const (
+	SESSION_KEY = "session"
+)
diff --git a/server/wheelserver/middleware/jwtauth.go b/server/wheelserver/middleware/jwtauth.go
index b1f54d47..cdea38f9 100644
--- a/server/wheelserver/middleware/jwtauth.go
+++ b/server/wheelserver/middleware/jwtauth.go
@@ -6,6 +6,7 @@ import (
 	"jccommon"
 	"main/mt"
 	"main/common"
+	"main/constant"
 	"q5"
 
 	. "main/global"
@@ -32,7 +33,7 @@ func internalJwtAuth(c *gin.Context, maybe bool) {
 	} else {
 		if maybe {
 			s := GetSessionMgr().CreateGuestSession()
-			c.Set("session", s)
+			c.Set(constant.SESSION_KEY, s)
 			c.Next()
 			return
 		}
@@ -94,7 +95,7 @@ func internalMetaMaskJwtAuth(c *gin.Context, jwtToken string) {
 				return
 			}
 			s := GetSessionMgr().CreateJwtSession(jsonRspObj.Decoded)
-			c.Set("session", s)
+			c.Set(constant.SESSION_KEY, s)
 			c.Next()
 		})
 }