From d692b45b22f985aff4635a5bb8b8958ae9bcd6dd Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 21 十月 2022 17:50:09 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/controller/data/DownloadController.java |   53 ++++++++++++++++++++++++++++++++++++++++++-----------
 1 files changed, 42 insertions(+), 11 deletions(-)

diff --git a/src/main/java/com/lf/server/controller/data/DownloadController.java b/src/main/java/com/lf/server/controller/data/DownloadController.java
index a23dab0..f41581b 100644
--- a/src/main/java/com/lf/server/controller/data/DownloadController.java
+++ b/src/main/java/com/lf/server/controller/data/DownloadController.java
@@ -3,8 +3,11 @@
 import com.lf.server.aspect.SysLog;
 import com.lf.server.controller.all.BaseController;
 import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.data.DirEntity;
 import com.lf.server.entity.data.DownloadEntity;
+import com.lf.server.entity.sys.UserEntity;
 import com.lf.server.service.data.DownloadService;
+import com.lf.server.service.sys.TokenService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiImplicitParams;
@@ -12,6 +15,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import javax.servlet.http.HttpServletRequest;
 import java.util.List;
 
 /**
@@ -20,10 +24,13 @@
  */
 @Api(tags = "鏁版嵁绠＄悊\\涓嬭浇璁板綍")
 @RestController
-@RequestMapping("/args")
+@RequestMapping("/download")
 public class DownloadController extends BaseController {
     @Autowired
     DownloadService downloadService;
+
+    @Autowired
+    TokenService tokenService;
 
     @SysLog()
     @ApiOperation(value = "鏌ヨ璁板綍鏁�")
@@ -118,13 +125,18 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "娣诲姞鏁版嵁")
+    @ApiOperation(value = "鎻掑叆涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "DownloadEntity", paramType = "body")
     })
     @PostMapping(value = "/insert", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> insert(@RequestBody DownloadEntity entity) {
+    public ResponseMsg<Integer> insert(@RequestBody DownloadEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setCreateUser(ue.getId());
+            }
+
             int count = downloadService.insert(entity);
 
             return success(count);
@@ -134,13 +146,20 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺娣诲姞")
+    @ApiOperation(value = "鎻掑叆澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "DownloadEntity", paramType = "body")
     })
     @PostMapping(value = "/inserts", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> inserts(@RequestBody List<DownloadEntity> list) {
+    public ResponseMsg<Integer> inserts(@RequestBody List<DownloadEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (DownloadEntity entity : list) {
+                    entity.setCreateUser(ue.getId());
+                }
+            }
+
             int count = downloadService.inserts(list);
 
             return success(count);
@@ -150,7 +169,7 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鍒櫎鏁版嵁")
+    @ApiOperation(value = "鍒犻櫎涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "id", value = "ID", dataType = "Integer", paramType = "query", example = "1")
     })
@@ -166,7 +185,7 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺鍒犻櫎")
+    @ApiOperation(value = "鍒犻櫎澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "Integer", paramType = "query", example = "1,2")
     })
@@ -186,14 +205,19 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "淇敼鏁版嵁")
+    @ApiOperation(value = "鏇存柊涓�鏉�")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "entity", value = "瀹炰綋绫�", dataType = "DownloadEntity", paramType = "body")
     })
     @ResponseBody
     @PostMapping(value = "/update", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> update(@RequestBody DownloadEntity entity) {
+    public ResponseMsg<Integer> update(@RequestBody DownloadEntity entity, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                entity.setDownloadUser(ue.getId());
+            }
+
             int count = downloadService.update(entity);
 
             return success(count);
@@ -203,14 +227,21 @@
     }
 
     @SysLog()
-    @ApiOperation(value = "鎵归噺淇敼")
+    @ApiOperation(value = "鏇存柊澶氭潯")
     @ApiImplicitParams({
             @ApiImplicitParam(name = "list", value = "瀹炰綋绫婚泦鍚�", dataType = "DownloadEntity", paramType = "body")
     })
     @ResponseBody
     @PostMapping(value = "/updates", produces = "application/json; charset=UTF-8")
-    public ResponseMsg<Integer> updates(@RequestBody List<DownloadEntity> list) {
+    public ResponseMsg<Integer> updates(@RequestBody List<DownloadEntity> list, HttpServletRequest req) {
         try {
+            UserEntity ue = tokenService.getCurrentUser(req);
+            if (ue != null) {
+                for (DownloadEntity entity : list) {
+                    entity.setDownloadUser(ue.getId());
+                }
+            }
+
             int count = downloadService.updates(list);
 
             return success(count);

--
Gitblit v1.9.3