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/data/StyleController.java | 41 ++++++++++++++++++++++++++++++++--------- 1 files changed, 32 insertions(+), 9 deletions(-) diff --git a/src/main/java/com/lf/server/controller/data/StyleController.java b/src/main/java/com/lf/server/controller/data/StyleController.java index 0458fc2..3c5e19f 100644 --- a/src/main/java/com/lf/server/controller/data/StyleController.java +++ b/src/main/java/com/lf/server/controller/data/StyleController.java @@ -4,7 +4,9 @@ import com.lf.server.controller.all.BaseController; import com.lf.server.entity.all.ResponseMsg; import com.lf.server.entity.data.StyleEntity; +import com.lf.server.entity.sys.UserEntity; import com.lf.server.service.data.StyleService; +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 StyleController extends BaseController { @Autowired StyleService styleService; + + @Autowired + TokenService tokenService; @SysLog() @ApiOperation(value = "鏌ヨ璁板綍鏁�") @@ -95,12 +101,17 @@ @SysLog() @ApiOperation(value = "鎻掑叆瀛楀吀") @ApiImplicitParams({ - @ApiImplicitParam(name = "styleEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.data.StyleEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.data.StyleEntity", paramType = "body", example = "") }) @PostMapping(value = "/insertStyle", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertStyle(@RequestBody StyleEntity styleEntity) { + public ResponseMsg<Integer> insertStyle(@RequestBody StyleEntity entity, HttpServletRequest req) { try { - int count = styleService.insertStyle(styleEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setCreateUser(ue.getId()); + } + + int count = styleService.insertStyle(entity); return success(count); } catch (Exception ex) { @@ -111,12 +122,19 @@ @SysLog() @ApiOperation(value = "鎻掑叆澶氭潯瀛楀吀") @ApiImplicitParams({ - @ApiImplicitParam(name = "StyleEntity", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.data.StyleEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "list", value = "瀛楀吀瀹炰綋绫�", dataType = "com.lf.server.entity.data.StyleEntity", paramType = "body", example = "") }) @PostMapping(value = "/insertStyles", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> insertStyles(@RequestBody List<StyleEntity> styleEntity) { + public ResponseMsg<Integer> insertStyles(@RequestBody List<StyleEntity> list, HttpServletRequest req) { try { - int count = styleService.insertStyles(styleEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + for (StyleEntity entity : list) { + entity.setCreateUser(ue.getId()); + } + } + + int count = styleService.insertStyles(list); return success(count); } catch (Exception ex) { @@ -163,13 +181,18 @@ @SysLog() @ApiOperation(value = "鏇存柊涓�鏉″瓧鍏�") @ApiImplicitParams({ - @ApiImplicitParam(name = "styleEntity", value = "瀛楀吀ID闆嗗悎", dataType = "StyleEntity", paramType = "body", example = "") + @ApiImplicitParam(name = "entity", value = "瀛楀吀ID闆嗗悎", dataType = "StyleEntity", paramType = "body", example = "") }) @ResponseBody @PostMapping(value = "/updateStyle", produces = "application/json; charset=UTF-8") - public ResponseMsg<Integer> updateStyle(@RequestBody StyleEntity styleEntity) { + public ResponseMsg<Integer> updateStyle(@RequestBody StyleEntity entity, HttpServletRequest req) { try { - int count = styleService.updateStyle(styleEntity); + UserEntity ue = tokenService.getCurrentUser(req); + if (ue != null) { + entity.setUpdateUser(ue.getId()); + } + + int count = styleService.updateStyle(entity); return success(count); } catch (Exception ex) { -- Gitblit v1.9.3