From 07a68b8d93aa059968a9709f2c81e6625338032c Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期一, 21 十一月 2022 20:19:49 +0800
Subject: [PATCH] 12
---
src/main/java/com/lf/server/controller/show/MarkController.java | 74 +++++++++++++++++++++++++++++++++---
1 files changed, 67 insertions(+), 7 deletions(-)
diff --git a/src/main/java/com/lf/server/controller/show/MarkController.java b/src/main/java/com/lf/server/controller/show/MarkController.java
index 2ff0d1f..15ff3d4 100644
--- a/src/main/java/com/lf/server/controller/show/MarkController.java
+++ b/src/main/java/com/lf/server/controller/show/MarkController.java
@@ -1,13 +1,19 @@
package com.lf.server.controller.show;
-import com.lf.server.aspect.SysLog;
+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.all.StaticData;
import com.lf.server.entity.ctrl.MarkJsonEntity;
+import com.lf.server.entity.data.DownloadEntity;
+import com.lf.server.entity.data.MetaFileEntity;
import com.lf.server.entity.show.MarkEntity;
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.UploaderService;
import com.lf.server.service.show.MarkService;
import com.lf.server.service.sys.TokenService;
import io.swagger.annotations.Api;
@@ -34,6 +40,12 @@
@Autowired
TokenService tokenService;
+
+ @Autowired
+ UploaderService uploaderService;
+
+ @Autowired
+ DownloadService downloadService;
@SysLog()
@ApiOperation(value = "鍒嗛〉鏌ヨ骞惰繑鍥炶褰曟暟")
@@ -213,7 +225,17 @@
@PostMapping(value = "/uploadShp", produces = "application/json; charset=UTF-8")
public ResponseMsg<List<MarkJsonEntity>> uploadShp(HttpServletRequest req, HttpServletResponse res) {
try {
- List<MarkJsonEntity> list = null;
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue == null) {
+ return fail("鐢ㄦ埛鏈櫥褰�", null);
+ }
+
+ List<MetaFileEntity> meList = uploaderService.uploadData(null, null, req, res);
+ if (meList == null || meList.size() < StaticData.FOUR) {
+ return fail("娌℃湁鎵惧埌宸蹭笂浼犵殑鏁版嵁鎴栦笉瀹屾暣", null);
+ }
+
+ List<MarkJsonEntity> list = markService.readShpForMarks(meList);
return success(list);
} catch (Exception ex) {
@@ -228,19 +250,57 @@
})
@ResponseBody
@PostMapping(value = "/downloadShp", produces = "application/json; charset=UTF-8")
- public void downloadShp(@RequestBody List<MarkJsonEntity> list, HttpServletRequest req, HttpServletResponse res) {
+ public ResponseMsg<String> downloadShp(@RequestBody List<MarkJsonEntity> list, HttpServletRequest req, HttpServletResponse res) {
+ try {
+ UserEntity ue = tokenService.getCurrentUser(req);
+ if (ue == null) {
+ return fail("鐢ㄦ埛鏈櫥褰�", null);
+ }
+ if (list == null || list.isEmpty()) {
+ return fail("鎵句笉鍒版爣缁樻暟鎹�", null);
+ }
+
+ String guid = markService.downloadShp(ue, list, req, res);
+ if (StringHelper.isNull(guid)) {
+ return fail("鐢熸垚ShapeFile鏂囦欢澶辫触", null);
+ }
+
+ return success(guid);
+ } 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) {
WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "鐢ㄦ埛鏈櫥褰�"));
}
- if (list == null || list.isEmpty()) {
- WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.BAD_REQUEST, "鎵句笉鍒版爣缁樺疄浣�"));
+
+ DownloadEntity de = downloadService.selectByGuid(guid);
+ if (de == null) {
+ WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�"));
}
- markService.downloadShp(ue, list, req, res);
+ de.setDcount(de.getDcount() + 1);
+ de.setDownloadUser(ue.getId());
+ int rows = downloadService.update(de);
+
+ String filePath = markService.getDownloadFilePath(de);
+ WebHelper.download(filePath, de.getName(), res);
} catch (Exception ex) {
- log.error(ex.getMessage(), ex);
+ try {
+ WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage()));
+ } catch (Exception e) {
+ log.error(e.getMessage(), e);
+ }
}
}
}
--
Gitblit v1.9.3