From 1f0549559468a1501ac572f8b26b3daedd2b44e1 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期三, 15 二月 2023 14:15:18 +0800 Subject: [PATCH] 1 --- src/main/java/com/lf/server/controller/show/ComprehensiveController.java | 121 +++++++++++++++++++++++++++++++++++++++- 1 files changed, 118 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/lf/server/controller/show/ComprehensiveController.java b/src/main/java/com/lf/server/controller/show/ComprehensiveController.java index 8843aa0..33e4624 100644 --- a/src/main/java/com/lf/server/controller/show/ComprehensiveController.java +++ b/src/main/java/com/lf/server/controller/show/ComprehensiveController.java @@ -2,16 +2,33 @@ 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.sys.UserEntity; +import com.lf.server.helper.PathHelper; +import com.lf.server.helper.StringHelper; +import com.lf.server.helper.WebHelper; +import com.lf.server.mapper.show.MarkMapper; import com.lf.server.service.all.BaseQueryService; +import com.lf.server.service.all.BaseUploadService; +import com.lf.server.service.data.DownloadService; +import com.lf.server.service.show.MarkService; +import com.lf.server.service.sys.DownlogService; +import com.lf.server.service.sys.TokenService; import io.swagger.annotations.Api; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; import io.swagger.annotations.ApiOperation; import org.springframework.beans.factory.annotation.Autowired; -import org.springframework.web.bind.annotation.GetMapping; -import org.springframework.web.bind.annotation.RequestMapping; -import org.springframework.web.bind.annotation.RestController; +import org.springframework.web.bind.annotation.*; + +import javax.servlet.http.HttpServletRequest; +import javax.servlet.http.HttpServletResponse; +import java.util.List; /** * 缁煎悎灞曠ず @@ -23,6 +40,21 @@ public class ComprehensiveController extends BaseController { @Autowired BaseQueryService baseQueryService; + + @Autowired + MarkService markService; + + @Autowired + TokenService tokenService; + + @Autowired + BaseUploadService baseUploadService; + + @Autowired + DownlogService downlogService; + + @Autowired + DownloadService downloadService; @SysLog() @ApiOperation(value = "鏌ヨ璺綉") @@ -42,4 +74,87 @@ return fail(ex.getMessage(), null); } } + + @SysLog() + @ApiOperation(value = "涓婁紶ShapeFile鏂囦欢") + @ResponseBody + @PostMapping(value = "/uploadShp", produces = "application/json; charset=UTF-8") + public ResponseMsg<Object> uploadShp(HttpServletRequest req, HttpServletResponse res) { + try { + UserEntity ue = tokenService.getCurrentUser(req); + if (ue == null) { + return fail("鐢ㄦ埛鏈櫥褰�", null); + } + + List<MetaFileEntity> meList = baseUploadService.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) { + return fail(ex.getMessage(), null); + } + } + + @SysLog() + @ApiOperation(value = "涓嬭浇ShapeFile鏂囦欢") + @ApiImplicitParams({ + @ApiImplicitParam(name = "list", value = "鏍囩粯JSON瀹炰綋绫婚泦鍚�", dataType = "MarkJsonEntity", paramType = "body") + }) + @ResponseBody + @PostMapping(value = "/downloadShp", produces = "application/json; charset=UTF-8") + 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); + 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, "鐢ㄦ埛鏈櫥褰�")); + } + + DownloadEntity de = downloadService.selectByGuid(guid); + if (de == null) { + WebHelper.write2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "鏂囦欢涓嶅瓨鍦�")); + } + + downlogService.updateInfos(ue, de, req); + + String filePath = markService.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); + } + } + } } -- Gitblit v1.9.3