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