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/RoleUserController.java | 38 ++++++++++++++++++++++++++++++++++----
1 files changed, 34 insertions(+), 4 deletions(-)
diff --git a/src/main/java/com/lf/server/controller/sys/RoleUserController.java b/src/main/java/com/lf/server/controller/sys/RoleUserController.java
index 855544b..3243472 100644
--- a/src/main/java/com/lf/server/controller/sys/RoleUserController.java
+++ b/src/main/java/com/lf/server/controller/sys/RoleUserController.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.RoleUserEntity;
+import com.lf.server.entity.sys.UserEntity;
import com.lf.server.service.sys.RoleUserService;
+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 RoleUserController extends BaseController {
@Autowired
RoleUserService roleUserService;
+
+ @Autowired
+ TokenService tokenService;
@SysLog()
@ApiOperation(value = "鏌ヨ璁板綍鏁�")
@@ -123,8 +129,13 @@
@ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "RoleUserEntity", paramType = "body")
})
@PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
- public ResponseMsg<Integer> insert(@RequestBody RoleUserEntity entity) {
+ public ResponseMsg<Integer> insert(@RequestBody RoleUserEntity entity, HttpServletRequest req) {
try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue != null) {
+ entity.setCreateUser(ue.getId());
+ }
+
int count = roleUserService.insert(entity);
return success(count);
@@ -139,8 +150,15 @@
@ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "RoleUserEntity", paramType = "body")
})
@PostMapping(value = "/inserts", produces = "application/json; charset=UTF-8")
- public ResponseMsg<Integer> inserts(@RequestBody List<RoleUserEntity> list) {
+ public ResponseMsg<Integer> inserts(@RequestBody List<RoleUserEntity> list, HttpServletRequest req) {
try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue != null) {
+ for (RoleUserEntity entity : list) {
+ entity.setCreateUser(ue.getId());
+ }
+ }
+
int count = roleUserService.inserts(list);
return success(count);
@@ -192,8 +210,13 @@
})
@ResponseBody
@PostMapping(value = "/update", produces = "application/json; charset=UTF-8")
- public ResponseMsg<Integer> update(@RequestBody RoleUserEntity entity) {
+ public ResponseMsg<Integer> update(@RequestBody RoleUserEntity entity, HttpServletRequest req) {
try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue != null) {
+ entity.setUpdateUser(ue.getId());
+ }
+
int count = roleUserService.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<RoleUserEntity> list) {
+ public ResponseMsg<Integer> updates(@RequestBody List<RoleUserEntity> list, HttpServletRequest req) {
try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue != null) {
+ for (RoleUserEntity entity : list) {
+ entity.setUpdateUser(ue.getId());
+ }
+ }
+
int count = roleUserService.updates(list);
return success(count);
--
Gitblit v1.9.3