From 298a420d8fa2435e2ca2fd7f4afa47118695f6bf Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期三, 23 十一月 2022 20:33:35 +0800
Subject: [PATCH] 1
---
src/main/java/com/lf/server/controller/data/MetaController.java | 69 ++++++++++++++++++++++++++++++++++
src/main/java/com/lf/server/controller/show/ExportController.java | 20 ++++++---
2 files changed, 82 insertions(+), 7 deletions(-)
diff --git a/src/main/java/com/lf/server/controller/data/MetaController.java b/src/main/java/com/lf/server/controller/data/MetaController.java
index 7149087..0725947 100644
--- a/src/main/java/com/lf/server/controller/data/MetaController.java
+++ b/src/main/java/com/lf/server/controller/data/MetaController.java
@@ -2,10 +2,14 @@
import com.lf.server.annotation.SysLog;
import com.lf.server.controller.all.BaseController;
+import com.lf.server.entity.all.HttpStatus;
import com.lf.server.entity.all.ResponseMsg;
+import com.lf.server.entity.data.DownloadEntity;
import com.lf.server.entity.data.MetaEntity;
import com.lf.server.entity.sys.UserEntity;
+import com.lf.server.helper.StringHelper;
import com.lf.server.helper.WebHelper;
+import com.lf.server.service.data.DownloadService;
import com.lf.server.service.data.MetaService;
import com.lf.server.service.sys.TokenService;
import io.swagger.annotations.Api;
@@ -16,6 +20,7 @@
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
import java.util.List;
/**
@@ -31,6 +36,9 @@
@Autowired
TokenService tokenService;
+
+ @Autowired
+ DownloadService downloadService;
@SysLog()
@ApiOperation(value = "鏌ヨ璁板綍鏁�")
@@ -252,4 +260,65 @@
return fail(ex.getMessage(), -1);
}
}
+
+ @SysLog()
+ @ApiOperation(value = "璇锋眰涓嬭浇")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "pwd", value = "瀵嗙爜", dataType = "String", paramType = "body", example = ""),
+ @ApiImplicitParam(name = "ids", value = "ID鏁扮粍", dataType = "Integer", paramType = "body", example = "1,2")
+ })
+ @PostMapping(value = "/downloadReq")
+ public ResponseMsg<Object> downloadReq(@RequestParam String pwd, @RequestParam List<Integer> ids, HttpServletRequest req, HttpServletResponse res) {
+ try {
+ if (StringHelper.isEmpty(pwd)) {
+ return fail("瀵嗙爜涓嶈兘涓虹┖");
+ }
+ if (null == ids || ids.isEmpty()) {
+ return fail("璇烽�夋嫨瑕佷笅杞界殑鏂囦欢");
+ }
+
+ return success(true);
+ } catch (Exception ex) {
+ return fail(ex.getMessage(), null);
+ }
+ }
+
+ @SysLog()
+ @ApiOperation(value = "涓嬭浇鏂囦欢")
+ @ApiImplicitParams({
+ @ApiImplicitParam(name = "guid", value = "鏂囦欢GUID", dataType = "String", paramType = "query")
+ })
+ @RequestMapping(value = "/downloadFile", method = RequestMethod.GET)
+ public void downloadFile(String guid, HttpServletRequest req, HttpServletResponse res) {
+ try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue == null) {
+ writeInfo(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�", res);
+ return;
+ }
+
+ DownloadEntity de = downloadService.selectByGuid(guid);
+ if (de == null) {
+ writeInfo(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�", res);
+ return;
+ }
+
+ de.setDcount(de.getDcount() + 1);
+ de.setDownloadUser(ue.getId());
+ int rows = downloadService.update(de);
+
+ //String filePath = exportService.getDownloadFilePath(de);
+ //WebHelper.download(filePath, de.getName(), res);
+ } catch (Exception ex) {
+ writeInfo(HttpStatus.ERROR, ex.getMessage(), res);
+ }
+ }
+
+ private void writeInfo(HttpStatus status, String info, HttpServletResponse res) {
+ try {
+ WebHelper.write2Page(res, WebHelper.getErrJson(status, info));
+ } catch (Exception e) {
+ log.error(e.getMessage(), e);
+ }
+ }
}
diff --git a/src/main/java/com/lf/server/controller/show/ExportController.java b/src/main/java/com/lf/server/controller/show/ExportController.java
index abd0c49..5bf50b2 100644
--- a/src/main/java/com/lf/server/controller/show/ExportController.java
+++ b/src/main/java/com/lf/server/controller/show/ExportController.java
@@ -114,12 +114,14 @@
try {
UserEntity ue = tokenService.getCurrentUser(req);
if (ue == null) {
- WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�"));
+ writeInfo(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�", res);
+ return;
}
DownloadEntity de = downloadService.selectByGuid(guid);
if (de == null) {
- WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�"));
+ writeInfo(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�", res);
+ return;
}
de.setDcount(de.getDcount() + 1);
@@ -129,11 +131,15 @@
String filePath = exportService.getDownloadFilePath(de);
WebHelper.download(filePath, de.getName(), res);
} catch (Exception ex) {
- try {
- WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage()));
- } catch (Exception e) {
- log.error(e.getMessage(), e);
- }
+ writeInfo(HttpStatus.ERROR, ex.getMessage(), res);
+ }
+ }
+
+ private void writeInfo(HttpStatus status, String info, HttpServletResponse res) {
+ try {
+ WebHelper.write2Page(res, WebHelper.getErrJson(status, info));
+ } catch (Exception e) {
+ log.error(e.getMessage(), e);
}
}
}
--
Gitblit v1.9.3