From a9262d9b1c5b570d077c2dc7d3fdda9058f88ec7 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 25 六月 2024 11:15:07 +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