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/AuthController.java | 41 ++++++++++++++++++++++++++++++++--------- 1 files changed, 32 insertions(+), 9 deletions(-) diff --git a/src/main/java/com/lf/server/controller/sys/AuthController.java b/src/main/java/com/lf/server/controller/sys/AuthController.java index df9513c..3d0be1b 100644 --- a/src/main/java/com/lf/server/controller/sys/AuthController.java +++ b/src/main/java/com/lf/server/controller/sys/AuthController.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.AuthEntity; +import com.lf.server.entity.sys.UserEntity; import com.lf.server.service.sys.AuthService; +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; /** @@ -25,6 +28,9 @@ public class AuthController extends BaseController { @Autowired AuthService authService; + + @Autowired + TokenService tokenService; @SysLog() @ApiOperation(value = "鏌ヨ璁板綍鏁�") @@ -92,12 +98,17 @@ @SysLog() @ApiOperation(value = "鎻掑叆瀛楀吀") @ApiImplicitParams({ - @ApiImplicitParam(name = "authEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.sys.AuthEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.sys.AuthEntity", paramType = "body", example = "") }) @PostMapping(value = "/insertAuth", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertAuth(@RequestBody AuthEntity authEntity) { + public ResponseMsg<Integer> insertAuth(@RequestBody AuthEntity entity, HttpServletRequest req) { try { - int count = authService.insertAuth(authEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setCreateUser(ue.getId()); + } + + int count = authService.insertAuth(entity); return success(count); } catch (Exception ex) { @@ -108,12 +119,19 @@ @SysLog() @ApiOperation(value = "鎻掑叆澶氭潯瀛楀吀") @ApiImplicitParams({ - @ApiImplicitParam(name = "authEntity", value = "瀛楀吀瀹炰綋绫婚泦鍚�", dataType = "List<AuthEntity>", paramType = "body", example = "") + @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "List<AuthEntity>", paramType = "body", example = "") }) @PostMapping(value = "/insertAuths", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertAuths(@RequestBody List<AuthEntity> authEntity) { + public ResponseMsg<Integer> insertAuths(@RequestBody List<AuthEntity> list, HttpServletRequest req) { try { - int count = authService.insertAuths(authEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + for (AuthEntity entity : list) { + entity.setCreateUser(ue.getId()); + } + } + + int count = authService.insertAuths(list); return success(count); } catch (Exception ex) { @@ -158,13 +176,18 @@ @SysLog() @ApiOperation(value = "鏇存柊涓�鏉″瓧鍏�") @ApiImplicitParams({ - @ApiImplicitParam(name = "loginEntity", value = "瀛楀吀ID闆嗗悎", dataType = "LoginEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "LoginEntity", paramType = "body", example = "") }) @ResponseBody @PostMapping(value = "/updateAuth", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> updateAuth(@RequestBody AuthEntity authEntity) { + public ResponseMsg<Integer> updateAuth(@RequestBody AuthEntity entity, HttpServletRequest req) { try { - int count = authService.updateAuth(authEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setUpdateUser(ue.getId()); + } + + int count = authService.updateAuth(entity); return success(count); } catch (Exception ex) { -- Gitblit v1.9.3