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/DownloadController.java | 39 +++++++++++++++++++++++++++++++++++----
1 files changed, 35 insertions(+), 4 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 e7461cf..15cd8ee 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;
/**
@@ -24,6 +28,9 @@
public class DownloadController extends BaseController {
@Autowired
DownloadService downloadService;
+
+ @Autowired
+ TokenService tokenService;
@SysLog()
@ApiOperation(value = "鏌ヨ璁板綍鏁�")
@@ -123,8 +130,13 @@
@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);
@@ -139,8 +151,15 @@
@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);
@@ -192,8 +211,13 @@
})
@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);
@@ -209,8 +233,15 @@
})
@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