package com.lf.server.controller.show; import com.lf.server.annotation.SysLog; 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.DownloadReqEntity; 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.data.PublishEntity; import com.lf.server.entity.show.PipelineEntity; import com.lf.server.entity.sys.AttachEntity; import com.lf.server.entity.sys.UserEntity; import com.lf.server.helper.StringHelper; import com.lf.server.helper.WebHelper; import com.lf.server.service.all.BaseQueryService; import com.lf.server.service.all.BaseUploadService; import com.lf.server.service.all.UploadAttachService; import com.lf.server.service.data.DataQueryService; import com.lf.server.service.data.DownloadService; import com.lf.server.service.data.PublishService; import com.lf.server.service.show.MarkService; import com.lf.server.service.show.PipelineService; import com.lf.server.service.sys.AttachService; 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.*; import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import java.util.List; /** * 综合展示 * @author WWW */ @Api(tags = "综合展示\\首页") @RestController @RequestMapping("/comprehensive") public class ComprehensiveController extends ModelController { @Autowired BaseQueryService baseQueryService; @Autowired MarkService markService; @Autowired TokenService tokenService; @Autowired AttachService attachService; @Autowired DownlogService downlogService; @Autowired PublishService publishService; @Autowired DownloadService downloadService; @Autowired PipelineService pipelineService; @Autowired DataQueryService dataQueryService; @Autowired BaseUploadService baseUploadService; @Autowired UploadAttachService uploadAttachService; @SysLog() @ApiOperation(value = "根据ID查询发布清单") @ApiImplicitParams({ @ApiImplicitParam(name = "id", value = "ID", dataType = "int", paramType = "query", example = "1") }) @GetMapping(value = "/selectPubById") public ResponseMsg selectPubById(int id) { try { PublishEntity entity = publishService.selectById(id); return success(entity); } catch (Exception ex) { return fail(ex.getMessage(), null); } } @SysLog() @ApiOperation(value = "查询管线") @ApiImplicitParams({ @ApiImplicitParam(name = "name", value = "名称", dataType = "String", paramType = "query", example = "西气东输"), }) @GetMapping(value = "/selectPipelines") public ResponseMsg selectPipelines(String name) { try { List rs = pipelineService.selectPipelines(name); return success(rs); } catch (Exception ex) { return fail(ex, null); } } @SysLog() @ApiOperation(value = "查询路网") @ApiImplicitParams({ @ApiImplicitParam(name = "x1", value = "X1", dataType = "Double", paramType = "query", example = "116.78999"), @ApiImplicitParam(name = "y1", value = "Y1", dataType = "Double", paramType = "query", example = "39.9468"), @ApiImplicitParam(name = "x2", value = "X2", dataType = "Double", paramType = "query", example = "116.80458"), @ApiImplicitParam(name = "y2", value = "Y2", dataType = "Double", paramType = "query", example = "39.94758") }) @GetMapping(value = "/selectRoute") public ResponseMsg selectRoute(double x1, double y1, double x2, double y2) { try { String route = baseQueryService.selectRoute(x1, y1, x2, y2); return success(route); } catch (Exception ex) { return fail(ex, null); } } @SysLog() @ApiOperation(value = "上传ShapeFile文件") @ResponseBody @PostMapping(value = "/uploadShp", produces = "application/json; charset=UTF-8") public ResponseMsg uploadShp(HttpServletRequest req, HttpServletResponse res) { try { UserEntity ue = tokenService.getCurrentUser(req); if (ue == null) { return fail("用户未登录", null); } List meList = baseUploadService.uploadData(null, null, false, req, res); if (meList == null || meList.size() < StaticData.FOUR) { return fail("没有找到已上传的数据或不完整", null); } List list = markService.readShpForMarks(meList); return success(list); } catch (Exception ex) { return fail(ex, 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 downloadShp(@RequestBody List 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, null); } } @SysLog() @ApiOperation(value = "上传文件") @ResponseBody @ApiImplicitParams({ @ApiImplicitParam(name = "tabName", value = "表名", dataType = "String", paramType = "query", example = "lf.sys_style"), @ApiImplicitParam(name = "eventid", value = "主键", dataType = "String", paramType = "query", example = "fa25979a5ef8b43ba82a0be35b3fb0d4") }) @PostMapping(value = "/uploadFiles") public ResponseMsg uploadFiles(String tabName, String eventid, @RequestParam("file") MultipartFile[] files, HttpServletRequest req) { try { if (StringHelper.isEmpty(tabName) || StringHelper.isEmpty(eventid)) { return fail("参数不能为空", null); } if (null == files || files.length == 0) { return fail("文件上传为空", 0); } UserEntity ue = tokenService.getCurrentUser(req); Integer count = dataQueryService.uploadFiles(ue, tabName, eventid, files, req); return success(count); } catch (Exception ex) { return fail(ex, 0); } } @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.writeStr2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, "用户未登录")); } DownloadEntity de = downloadService.selectByGuid(guid); if (de == null) { WebHelper.writeStr2Page(res, WebHelper.getErrJson(HttpStatus.NOT_FOUND, "文件不存在")); } downlogService.updateInfos(ue, de, req); String filePath = downloadService.getDownloadFilePath(de); WebHelper.download(filePath, de.getName(), res); } catch (Exception ex) { WebHelper.writeStr2Page(res, WebHelper.getErrJson(HttpStatus.UNAUTHORIZED, ex.getMessage())); log.error(ex.getMessage(), ex); } } @SysLog() @ApiOperation(value = "查看文件") @ApiImplicitParams({ @ApiImplicitParam(name = "guid", value = "附件Guid", dataType = "String", paramType = "body") }) @GetMapping(value = "/downloadForView") public void downloadForView(String guid, HttpServletResponse res) { uploadAttachService.download(guid, true, res); } @SysLog() @ApiOperation(value = "请求下载") @ApiImplicitParams({ @ApiImplicitParam(name = "guids", value = "附件Guid数组", dataType = "String", paramType = "body", allowMultiple = true, example = "e5b6ae0889b88111f13a4b6e048348db,fa4f299e901a0c46e634f8fcc8185c0c") }) @ResponseBody @PostMapping(value = "/downloadReqForGuids") public ResponseMsg downloadReqForGuids(@RequestBody DownloadReqEntity dre, HttpServletRequest req, HttpServletResponse res) { try { if (null == dre || null == dre.getGuids() || dre.getGuids().isEmpty()) { return fail("Guid数组为空"); } UserEntity ue = tokenService.getCurrentUser(req); String guid = uploadAttachService.downloadReqForGuids(ue, dre.getGuids()); return success(guid); } catch (Exception ex) { return fail(ex, null); } } @SysLog() @ApiOperation(value = "请求下载") @ApiImplicitParams({ @ApiImplicitParam(name = "tab", value = "表名", dataType = "String", paramType = "body", example = "bd.b_borehole"), @ApiImplicitParam(name = "guids", value = "附件Guid数组", dataType = "String", paramType = "body", allowMultiple = true, example = "08e5b4ad-93b8-46f6-adaa-46a6274af4ce,6f94f4f8-6e8d-44e7-8803-3d068d34983f") }) @ResponseBody @PostMapping(value = "/downloadReqForTabGuids") public ResponseMsg downloadReqForTabGuids(@RequestBody DownloadReqEntity dre, HttpServletRequest req, HttpServletResponse res) { try { if (null == dre || StringHelper.isEmpty(dre.getTab())) { return fail("表名为空"); } if (null != dre.getGuids() && dre.getGuids().size() == 0) { dre.setIds(null); } UserEntity ue = tokenService.getCurrentUser(req); String guid = uploadAttachService.downloadReqForTabGuids(ue, dre.getTab(), dre.getGuids()); return success(guid); } catch (Exception ex) { return fail(ex, null); } } @SysLog() @ApiOperation(value = "查询文件") @ApiImplicitParams({ @ApiImplicitParam(name = "tabName", value = "表名", dataType = "String", paramType = "query", example = "lf.sys_style"), @ApiImplicitParam(name = "eventid", value = "主键", dataType = "String", paramType = "query", example = "fa25979a5ef8b43ba82a0be35b3fb0d4") }) @GetMapping(value = "/selectFiles") public ResponseMsg> selectFiles(String tabName, String eventid) { try { if (StringHelper.isEmpty(tabName) || StringHelper.isEmpty(eventid)) { return fail("参数不能为空", null); } List list = attachService.selectByTab(tabName, eventid); return success(list); } catch (Exception ex) { return fail(ex, null); } } @SysLog() @ApiOperation(value = "删除文件") @ApiImplicitParams({ @ApiImplicitParam(name = "ids", value = "ID数组", dataType = "Integer", paramType = "query", example = "1,2") }) @GetMapping(value = "/deletes") public ResponseMsg deleteFiles(@RequestParam List ids) { try { if (ids == null || ids.isEmpty()) { return fail("id数组不能为空", -1); } int rows = attachService.deletes(ids); return success(rows); } catch (Exception ex) { return fail(ex, 0); } } }