package com.lf.server.controller.show; 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.*; 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 BaseController { @Autowired BaseQueryService baseQueryService; @Autowired MarkService markService; @Autowired TokenService tokenService; @Autowired BaseUploadService baseUploadService; @Autowired DownlogService downlogService; @Autowired DownloadService downloadService; @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 = "下载文件") @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); } } } }