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/TokenController.java | 41 +++++++++++++++++++++++++++++++---------- 1 files changed, 31 insertions(+), 10 deletions(-) diff --git a/src/main/java/com/lf/server/controller/sys/TokenController.java b/src/main/java/com/lf/server/controller/sys/TokenController.java index eddfbe5..1ea83ef 100644 --- a/src/main/java/com/lf/server/controller/sys/TokenController.java +++ b/src/main/java/com/lf/server/controller/sys/TokenController.java @@ -4,6 +4,7 @@ import com.lf.server.controller.all.BaseController; import com.lf.server.entity.all.ResponseMsg; import com.lf.server.entity.sys.TokenEntity; +import com.lf.server.entity.sys.UserEntity; import com.lf.server.service.sys.TokenService; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; @@ -12,6 +13,7 @@ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; +import javax.servlet.http.HttpServletRequest; import java.util.List; /** @@ -90,12 +92,18 @@ @SysLog() @ApiOperation(value = "鎻掑叆涓�鏉�") @ApiImplicitParams({ - @ApiImplicitParam(name = "tokenEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.sys.TokenEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "com.lf.server.entity.sys.TokenEntity", paramType = "body", example = "") }) @PostMapping(value = "/insertToken", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertToken(@RequestBody TokenEntity tokenEntity) { + public ResponseMsg<Integer> insertToken(@RequestBody TokenEntity entity, HttpServletRequest req) { try { - int count = tokenService.insertToken(tokenEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setCreateUser(ue.getId()); + } + + int count = tokenService.insertToken(entity); + return success(count); } catch (Exception ex) { return fail(ex.getMessage(), -1); @@ -105,12 +113,19 @@ @SysLog() @ApiOperation(value = "鎻掑叆澶氭潯") @ApiImplicitParams({ - @ApiImplicitParam(name = "tokenEntity", value = "瀹炰綋绫婚泦鍚�", dataType = "List<TokenEntity>", paramType = "body", example = "") + @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "List<TokenEntity>", paramType = "body", example = "") }) @PostMapping(value = "/insertTokens", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertTokens(@RequestBody List<TokenEntity> tokenEntity) { + public ResponseMsg<Integer> insertTokens(@RequestBody List<TokenEntity> list, HttpServletRequest req) { try { - int count = tokenService.insertTokens(tokenEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + for (TokenEntity entity : list) { + entity.setCreateUser(ue.getId()); + } + } + + int count = tokenService.insertTokens(list); return success(count); } catch (Exception ex) { @@ -137,7 +152,7 @@ @SysLog() @ApiOperation(value = "鍒犻櫎澶氭潯") @ApiImplicitParams({ - @ApiImplicitParam(name = "ids", value = "ID闆嗗悎", dataType = "List<Integer>", paramType = "query", example = "1,2") + @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "List<Integer>", paramType = "query", example = "1,2") }) @GetMapping(value = "/deleteTokens") public ResponseMsg<Integer> deleteTokens(@RequestParam List<Integer> ids) { @@ -155,13 +170,19 @@ @SysLog() @ApiOperation(value = "鏇存柊涓�鏉�") @ApiImplicitParams({ - @ApiImplicitParam(name = "tokenEntity", value = "ID闆嗗悎", dataType = "TokenEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "TokenEntity", paramType = "body", example = "") }) @ResponseBody @PostMapping(value = "/updateToken", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> updateToken(@RequestBody TokenEntity tokenEntity) { + public ResponseMsg<Integer> updateToken(@RequestBody TokenEntity entity, HttpServletRequest req) { try { - int count = tokenService.updateToken(tokenEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setUpdateUser(ue.getId()); + } + + int count = tokenService.updateToken(entity); + return success(count); } catch (Exception ex) { return fail(ex.getMessage(), -1); -- Gitblit v1.9.3