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