From 25ee639e42cdaf5c1874fc40d8309dc898989d2b Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 25 十月 2022 13:23:40 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/all/PermsController.java |   64 +++++++++++++++++++-------------
 1 files changed, 38 insertions(+), 26 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/all/PermsController.java b/src/main/java/com/lf/server/controller/all/PermsController.java
index defdde7..ee34610 100644
--- a/src/main/java/com/lf/server/controller/all/PermsController.java
+++ b/src/main/java/com/lf/server/controller/all/PermsController.java
@@ -5,14 +5,15 @@
 import com.lf.server.entity.all.PermsAuthEntity;
 import com.lf.server.entity.all.ResAuthEntity;
 import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.sys.UserEntity;
 import com.lf.server.service.all.PermsService;
+import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiImplicitParam;
-import io.swagger.annotations.ApiImplicitParams;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.List;
 
 /**
@@ -26,15 +27,20 @@
     @Autowired
     PermsService permsService;
 
+    @Autowired
+    TokenService tokenService;
+
     @SysLog()
-    @ApiOperation(value = "鏍规嵁鐢ㄦ埛Uid鏌ヨ璧勬簮鎺堟潈")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "uid", value = "鐢ㄦ埛Uid", dataType = "String", paramType = "query", example = "admin")
-    })
+    @ApiOperation(value = "鏌ヨ褰撳墠鐢ㄦ埛鐨勮祫婧愭巿鏉�")
     @GetMapping(value = "/selectRes")
-    public ResponseMsg<List<ResAuthEntity>> selectRes(String uid) {
+    public ResponseMsg<List<ResAuthEntity>> selectRes(HttpServletRequest req) {
         try {
-            List<ResAuthEntity> rs = permsService.selectRes(uid);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return null;
+            }
+
+            List<ResAuthEntity> rs = permsService.selectRes(ue.getUid());
 
             return success(rs);
         } catch (Exception ex) {
@@ -43,14 +49,16 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鏍规嵁鐢ㄦ埛Uid鏌ヨ鑿滃崟鎺堟潈")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "uid", value = "鐢ㄦ埛Uid", dataType = "String", paramType = "query", example = "admin")
-    })
+    @ApiOperation(value = "鏌ヨ褰撳墠鐢ㄦ埛鐨勮彍鍗曟巿鏉�")
     @GetMapping(value = "/selectMenus")
-    public ResponseMsg<List<MenusAuthEntity>> selectMenus(String uid) {
+    public ResponseMsg<List<MenusAuthEntity>> selectMenus(HttpServletRequest req) {
         try {
-            List<MenusAuthEntity> rs = permsService.selectMenus(uid);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return null;
+            }
+
+            List<MenusAuthEntity> rs = permsService.selectMenus(ue.getUid());
 
             return success(rs);
         } catch (Exception ex) {
@@ -59,14 +67,16 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鏍规嵁鐢ㄦ埛Uid鏌ヨ鏉冮檺鎺堟潈")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "uid", value = "鐢ㄦ埛Uid", dataType = "String", paramType = "query", example = "admin")
-    })
+    @ApiOperation(value = "鏌ヨ褰撳墠鐢ㄦ埛鐨勬潈闄愭巿鏉�")
     @GetMapping(value = "/selectPerms")
-    public ResponseMsg<List<String>> selectPerms(String uid) {
+    public ResponseMsg<List<String>> selectPerms(HttpServletRequest req) {
         try {
-            List<String> rs = permsService.selectPerms(uid);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return null;
+            }
+
+            List<String> rs = permsService.selectPerms(ue.getUid());
 
             return success(rs);
         } catch (Exception ex) {
@@ -75,14 +85,16 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鏍规嵁鐢ㄦ埛Uid鏌ヨ鏉冮檺鎺堟潈瀹炰綋闆嗗悎")
-    @ApiImplicitParams({
-            @ApiImplicitParam(name = "uid", value = "鐢ㄦ埛Uid", dataType = "String", paramType = "query", example = "admin")
-    })
+    @ApiOperation(value = "鏌ヨ褰撳墠鐢ㄦ埛鐨勬潈闄愭巿鏉冨疄浣撻泦鍚�")
     @GetMapping(value = "/selectPermsEntity")
-    public ResponseMsg<List<PermsAuthEntity>> selectPermsEntity(String uid) {
+    public ResponseMsg<List<PermsAuthEntity>> selectPermsEntity(HttpServletRequest req) {
         try {
-            List<PermsAuthEntity> rs = permsService.selectPermsEntity(uid);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue == null) {
+                return null;
+            }
+
+            List<PermsAuthEntity> rs = permsService.selectPermsEntity(ue.getUid());
 
             return success(rs);
         } catch (Exception ex) {

--
Gitblit v1.9.3