From 254bfe0c2b385517ab3a60952845bb0ab35d1bf3 Mon Sep 17 00:00:00 2001
From: liujiandao <274878379@qq.com>
Date: 星期一, 08 一月 2024 17:55:18 +0800
Subject: [PATCH] 产品获取登录用户信息

---
 middleware/utils.go |    9 +++++++++
 1 files changed, 9 insertions(+), 0 deletions(-)

diff --git a/middleware/utils.go b/middleware/utils.go
index 58defa0..cf98138 100644
--- a/middleware/utils.go
+++ b/middleware/utils.go
@@ -2,6 +2,7 @@
 
 import (
 	"errors"
+	"github.com/gin-gonic/gin"
 	"github.com/golang-jwt/jwt/v4"
 	"wms/request"
 )
@@ -58,3 +59,11 @@
 		return nil, TokenInvalid
 	}
 }
+
+func GetUserInfo(c *gin.Context) *request.CustomClaims {
+	if claims, exists := c.Get("claims"); exists {
+		waitUse := claims.(*request.CustomClaims)
+		return waitUse
+	}
+	return nil
+}

--
Gitblit v1.8.0