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

---
 controllers/product_controller.go |   15 +++++++++++++++
 middleware/utils.go               |    9 +++++++++
 router/router.go                  |    1 +
 3 files changed, 25 insertions(+), 0 deletions(-)

diff --git a/controllers/product_controller.go b/controllers/product_controller.go
index 075acb4..84d970e 100644
--- a/controllers/product_controller.go
+++ b/controllers/product_controller.go
@@ -12,6 +12,7 @@
 	"wms/constvar"
 	"wms/extend/code"
 	"wms/extend/util"
+	"wms/middleware"
 	"wms/models"
 	"wms/pkg/logx"
 	"wms/pkg/mysqlx"
@@ -746,3 +747,17 @@
 	}
 	util.ResponseFormat(c, code.Success, "鎿嶄綔鎴愬姛")
 }
+
+// GetUserInfo
+//
+//	@Tags		浜у搧
+//	@Summary	鑾峰彇鐧诲綍鐢ㄦ埛淇℃伅
+//	@Produce	application/json
+//	@Success	200	{object}	util.ResponseList{data=map[string]interface{}}	"鎴愬姛"
+//	@Router		/api-wms/v1/product/getUserInfo [get]
+func (slf ProductController) GetUserInfo(c *gin.Context) {
+	userInfo := middleware.GetUserInfo(c)
+	m := make(map[string]interface{})
+	m["userName"] = userInfo.Username
+	util.ResponseFormat(c, code.Success, m)
+}
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
+}
diff --git a/router/router.go b/router/router.go
index e7565eb..3e6ce21 100644
--- a/router/router.go
+++ b/router/router.go
@@ -119,6 +119,7 @@
 		productAPI.POST("updateDisuse", productController.UpdateDisuse)    //淇敼鎶ュ簾淇℃伅
 		productAPI.POST("listHistory", productController.ListHistory)      //浜у搧浣嶇疆鍘嗗彶璁板綍
 		productAPI.PUT("cancelDisuse/:id", productController.CancelDisuse) //鍙栨秷鎶ュ簾
+		productAPI.GET("getUserInfo", productController.GetUserInfo)       //鑾峰彇鐧诲綍鐢ㄦ埛淇℃伅
 
 	}
 

--
Gitblit v1.8.0