From 9c5c93a36294b3f14536689d548928196c367de8 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 11 十月 2022 09:18:20 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/sys/MenuController.java |   55 +++++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 43 insertions(+), 12 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/sys/MenuController.java b/src/main/java/com/lf/server/controller/sys/MenuController.java
index e37257c..b606928 100644
--- a/src/main/java/com/lf/server/controller/sys/MenuController.java
+++ b/src/main/java/com/lf/server/controller/sys/MenuController.java
@@ -5,8 +5,10 @@
 import com.lf.server.entity.all.ResponseMsg;
 import com.lf.server.entity.data.DirEntity;
 import com.lf.server.entity.sys.MenuEntity;
+import com.lf.server.entity.sys.UserEntity;
 import com.lf.server.helper.StringHelper;
 import com.lf.server.service.sys.MenuService;
+import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -14,6 +16,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.List;
 
 /**
@@ -28,15 +31,24 @@
     @Autowired
     MenuService menuService;
 
+    @Autowired
+    TokenService tokenService;
+
     @SysLog()
     @ApiOperation(value = "鎻掑叆涓�鏉℃暟鎹�")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "MenusEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.sys.MenusEntity", paramType = "body", example = "")
+            @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "com.lf.server.entity.sys.MenusEntity", paramType = "body", example = "")
     })
     @PostMapping(value = "/insertMenu", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> insertMenu(@RequestBody MenuEntity menuEntity) {
+    public ResponseMsg<Integer> insertMenu(@RequestBody MenuEntity entity, HttpServletRequest req) {
         try {
-            int count = menuService.insertMenu(menuEntity);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setCreateUser(ue.getId());
+            }
+
+            int count = menuService.insertMenu(entity);
+
             return success(count);
         } catch (Exception ex) {
             return fail(ex.getMessage(), -1);
@@ -46,12 +58,19 @@
     @SysLog()
     @ApiOperation(value = "鎻掑叆澶氭潯鏁版嵁")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "MenusEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.sys.MenusEntity", paramType = "body", example = "")
+            @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "com.lf.server.entity.sys.MenusEntity", paramType = "body", example = "")
     })
     @PostMapping(value = "/insertMenus", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> insertMenus(@RequestBody List<MenuEntity> menuEntity) {
+    public ResponseMsg<Integer> insertMenus(@RequestBody List<MenuEntity> list, HttpServletRequest req) {
         try {
-            int count = menuService.insertMenus(menuEntity);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (MenuEntity entity : list) {
+                    entity.setCreateUser(ue.getId());
+                }
+            }
+
+            int count = menuService.insertMenus(list);
 
             return success(count);
         } catch (Exception ex) {
@@ -98,13 +117,18 @@
     @SysLog()
     @ApiOperation(value = "鏇存柊涓�鏉℃暟鎹�")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "MenusEntity", value = "鑿滃崟ID闆嗗悎", dataType = "MenusEntity", paramType = "body", example = "")
+            @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "MenusEntity", paramType = "body", example = "")
     })
     @ResponseBody
     @PostMapping(value = "/updateMenu", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> updateMenu(@RequestBody MenuEntity menuEntity) {
+    public ResponseMsg<Integer> updateMenu(@RequestBody MenuEntity entity, HttpServletRequest req) {
         try {
-            int count = menuService.updateMenu(menuEntity);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setUpdateUser(ue.getId());
+            }
+
+            int count = menuService.updateMenu(entity);
 
             return success(count);
         } catch (Exception ex) {
@@ -115,13 +139,20 @@
     @SysLog()
     @ApiOperation(value = "鏇存柊澶氭潯鏁版嵁")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "MenusEntity", value = "涓婚敭ID闆嗗悎", dataType = "MenusEntity", paramType = "body", example = "")
+            @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "MenusEntity", paramType = "body", example = "")
     })
     @ResponseBody
     @PostMapping(value = "/updateMenus", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> updateMenus(@RequestBody List<MenuEntity> menuEntity) {
+    public ResponseMsg<Integer> updateMenus(@RequestBody List<MenuEntity> list, HttpServletRequest req) {
         try {
-            int count = menuService.updateMenus(menuEntity);
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (MenuEntity entity : list) {
+                    entity.setUpdateUser(ue.getId());
+                }
+            }
+
+            int count = menuService.updateMenus(list);
 
             return success(count);
         } catch (Exception ex) {

--
Gitblit v1.9.3