From 7fd2c877f44b3835fbcc90e420f52090787f0249 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期二, 11 十月 2022 10:17:45 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/sys/RoleMenuController.java |   38 ++++++++++++++++++++++++++++++++++----
 1 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/sys/RoleMenuController.java b/src/main/java/com/lf/server/controller/sys/RoleMenuController.java
index c4a14f1..9b68fb9 100644
--- a/src/main/java/com/lf/server/controller/sys/RoleMenuController.java
+++ b/src/main/java/com/lf/server/controller/sys/RoleMenuController.java
@@ -4,7 +4,9 @@
 import com.lf.server.controller.all.BaseController;
 import com.lf.server.entity.all.ResponseMsg;
 import com.lf.server.entity.sys.RoleMenuEntity;
+import com.lf.server.entity.sys.UserEntity;
 import com.lf.server.service.sys.RoleMenuService;
+import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -12,6 +14,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.List;
 
 /**
@@ -24,6 +27,9 @@
 public class RoleMenuController extends BaseController {
     @Autowired
     RoleMenuService roleMenuService;
+
+    @Autowired
+    TokenService tokenService;
 
     @SysLog()
     @ApiOperation(value = "鏌ヨ璁板綍鏁�")
@@ -123,8 +129,13 @@
             @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "RoleMenuEntity", paramType = "body")
     })
     @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> insert(@RequestBody RoleMenuEntity entity) {
+    public ResponseMsg<Integer> insert(@RequestBody RoleMenuEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setCreateUser(ue.getId());
+            }
+
             int count = roleMenuService.insert(entity);
 
             return success(count);
@@ -139,8 +150,15 @@
             @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "RoleMenuEntity", paramType = "body")
     })
     @PostMapping(value = "/inserts", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> inserts(@RequestBody List<RoleMenuEntity> list) {
+    public ResponseMsg<Integer> inserts(@RequestBody List<RoleMenuEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (RoleMenuEntity entity : list) {
+                    entity.setCreateUser(ue.getId());
+                }
+            }
+
             int count = roleMenuService.inserts(list);
 
             return success(count);
@@ -192,8 +210,13 @@
     })
     @ResponseBody
     @PostMapping(value = "/update", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> update(@RequestBody RoleMenuEntity entity) {
+    public ResponseMsg<Integer> update(@RequestBody RoleMenuEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setUpdateUser(ue.getId());
+            }
+
             int count = roleMenuService.update(entity);
 
             return success(count);
@@ -209,8 +232,15 @@
     })
     @ResponseBody
     @PostMapping(value = "/updates", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> updates(@RequestBody List<RoleMenuEntity> list) {
+    public ResponseMsg<Integer> updates(@RequestBody List<RoleMenuEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (RoleMenuEntity entity : list) {
+                    entity.setUpdateUser(ue.getId());
+                }
+            }
+
             int count = roleMenuService.updates(list);
 
             return success(count);

--
Gitblit v1.9.3