From ce663f476e7134bb4f9e34d5491fc37be53cc09f Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期四, 21 三月 2024 11:30:39 +0800 Subject: [PATCH] 添加项目名称 --- src/main/java/com/lf/server/controller/all/BaseQueryController.java | 329 +++++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 260 insertions(+), 69 deletions(-) diff --git a/src/main/java/com/lf/server/controller/all/BaseQueryController.java b/src/main/java/com/lf/server/controller/all/BaseQueryController.java index e9ad8fa..8e2b9b9 100644 --- a/src/main/java/com/lf/server/controller/all/BaseQueryController.java +++ b/src/main/java/com/lf/server/controller/all/BaseQueryController.java @@ -5,7 +5,9 @@ import com.baomidou.mybatisplus.core.metadata.OrderItem; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.lf.server.annotation.SysLog; +import com.lf.server.entity.all.BaseGeoEntity; 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.IdNameEntity; import com.lf.server.entity.ctrl.KeyValueEntity; @@ -19,12 +21,14 @@ import com.lf.server.mapper.all.BasicMapper; import com.lf.server.mapper.all.GeomBaseMapper; import com.lf.server.service.all.BaseQueryService; +import com.lf.server.service.data.DictService; import com.lf.server.service.data.DirService; import com.lf.server.service.data.DownloadService; import com.lf.server.service.data.MetaService; import com.lf.server.service.show.DataLibService; import com.lf.server.service.sys.AttachService; import com.lf.server.service.sys.DepService; +import com.lf.server.service.sys.DownlogService; import com.lf.server.service.sys.TokenService; import io.swagger.annotations.ApiImplicitParam; import io.swagger.annotations.ApiImplicitParams; @@ -34,6 +38,7 @@ import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; +import java.util.ArrayList; import java.util.List; /** @@ -60,10 +65,16 @@ DataLibService dataLibService; @Autowired - DownloadService downloadService; + public DownlogService downlogService; + + @Autowired + public DownloadService downloadService; @Autowired BaseQueryService baseQueryService; + + @Autowired + DictService dictService; @SysLog() @ApiOperation(value = "鏌ヨ璁板綍鏁�") @@ -103,14 +114,15 @@ @ApiImplicitParam(name = "wkt", value = "WKT锛堣憲鍚嶆枃鏈級", dataType = "String", paramType = "query", example = ""), //@ApiImplicitParam(name = "srid", value = "绌洪棿寮曠敤鏍囪瘑绗�", dataType = "Integer", paramType = "query", example = "4326"), @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1"), - @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10") + @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"), + @ApiImplicitParam(name = "hasGeom", value = "鏄�/鍚﹀寘鍚┖闂翠俊鎭�", dataType = "Integer", paramType = "query", example = "1") }) @GetMapping(value = "/selectByPage") @PostMapping(value = "/selectByPage") - public ResponseMsg<List<Object>> selectByPage(String name, String depcode, String dirs, String filter, String wkt, Integer pageIndex, Integer pageSize) { + public ResponseMsg<List<?>> selectByPage(String name, String depcode, String dirs, String filter, String wkt, Integer pageIndex, Integer pageSize, Integer hasGeom) { try { BasicMapper baseMapper = ClassHelper.getBasicMapper(name); - if (baseMapper == null) { + if (null == baseMapper) { return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); } @@ -126,11 +138,88 @@ wrapper.apply(dirs); } + boolean hasWkt = null != hasGeom && hasGeom > 0 && baseMapper instanceof GeomBaseMapper; + if (hasWkt) { + wrapper.select("ST_AsText(geom) as geom, *"); + } + Page<Object> page = new Page<>(pageIndex, pageSize); - page.addOrder(OrderItem.asc("gid")); - IPage<Object> paged = baseMapper.selectPage(page, wrapper); + page.addOrder(OrderItem.desc("gid")); + IPage<?> paged = baseMapper.selectPage(page, wrapper); + if (hasWkt && null != paged.getRecords()) { + copeWkt(paged.getRecords()); + } return success(paged.getTotal(), paged.getRecords()); + } catch (Exception ex) { + return fail(ex, null); + } + } + + /** + * 澶勭悊WKT + */ + private void copeWkt(List<?> list) { + for (int i = 0, c = list.size(); i < c; i++) { + Object obj = list.get(i); + setWktEncrypt(obj); + } + } + + /** + * 璁剧疆WKT鍔犲瘑 + */ + private void setWktEncrypt(Object obj) { + if (obj instanceof BaseGeoEntity) { + BaseGeoEntity entity = (BaseGeoEntity) obj; + if (StringHelper.isEmpty(entity.getGeom())) { + return; + } + + try { + String wkt = AesHelper.encrypt(entity.getGeom()); + entity.setGeom(wkt); + } catch (Exception ex) { + entity.setGeom(null); + log.error(ex.getMessage(), ex); + } + } + } + + @SysLog() + @ApiOperation(value = "缂撳啿鍖烘煡璇�") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鏄犲皠鍚嶇О", dataType = "String", paramType = "query", example = "dlgagnp"), + @ApiImplicitParam(name = "wkt", value = "WKT锛堣憲鍚嶆枃鏈級", dataType = "String", paramType = "query", example = ""), + @ApiImplicitParam(name = "buffer", value = "缂撳啿鍖哄ぇ灏�", dataType = "Double", paramType = "query", example = "10"), + @ApiImplicitParam(name = "limit", value = "闄愬埗鏉℃暟", dataType = "Integer", paramType = "query", example = "20") + }) + @GetMapping(value = "/selectByBuffer") + public ResponseMsg<List<?>> selectByBuffer(String name, String wkt, Double buffer, Integer limit) { + try { + if (StringHelper.isEmpty(wkt)) { + return fail("WKT涓嶈兘涓虹┖", null); + } + if (null == limit || limit < 1 || limit > StaticData.I100) { + limit = 20; + } + if (null == buffer || buffer < 0 || buffer > StaticData.I100000) { + buffer = 10.0; + } + + GeomBaseMapper baseMapper = ClassHelper.getGeoBaseMapper(name); + if (baseMapper == null) { + return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + } + + QueryWrapper wrapper = new QueryWrapper(); + wrapper.select("ST_AsText(geom) as geom, *"); + baseQueryService.addBufferWrapper(baseMapper, wrapper, wkt, buffer); + wrapper.last("limit " + limit); + + List<?> list = baseMapper.selectList(wrapper); + + return success(null == list ? 0 : list.size(), list); } catch (Exception ex) { return fail(ex, null); } @@ -177,13 +266,13 @@ public ResponseMsg<String> selectWktById(String name, Integer gid) { try { GeomBaseMapper baseMapper = ClassHelper.getGeoBaseMapper(name); - if (baseMapper == null) { - return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + if (null == baseMapper) { + return fail("鏌ヨ瀵硅薄涓嶅瓨鍦ㄦ垨涓嶆槸绌洪棿琛�", null); } String tab = BaseQueryService.getTabName(baseMapper); if (StringHelper.isNull(tab)) { - return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + return fail("鏌ヨ瀵硅薄鐨勮〃鍚嶄笉瀛樺湪", null); } String wkt = baseMapper.selectWktById(tab, gid); @@ -192,6 +281,56 @@ } return success(wkt); + } catch (Exception ex) { + return fail(ex, null); + } + } + + @SysLog() + @ApiOperation(value = "鏍规嵁GID鏌ヨ") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鏄犲皠鍚嶇О", dataType = "String", paramType = "query", example = "dlgagnp"), + @ApiImplicitParam(name = "gid", value = "GID", dataType = "int", paramType = "query", example = "1") + }) + @GetMapping(value = "/selectByGid") + public ResponseMsg<Object> selectByGid(String name, int gid) { + try { + BasicMapper baseMapper = ClassHelper.getBasicMapper(name); + if (null == baseMapper) { + return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + } + + QueryWrapper wrapper = new QueryWrapper(); + if (baseMapper instanceof GeomBaseMapper) { + wrapper.select("ST_AsText(geom) as geom, *"); + } + wrapper.eq("gid", gid); + + Object obj = baseMapper.selectOne(wrapper); + setWktEncrypt(obj); + + return success(null == obj ? 0 : 1, obj); + } catch (Exception ex) { + return fail(ex, null); + } + } + + @SysLog() + @ApiOperation(value = "鏍规嵁瀹炰綋鍚嶆煡璇㈣〃鍚�") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鏄犲皠鍚嶇О", dataType = "String", paramType = "query", example = "dlgagnp") + }) + @GetMapping(value = "/selectTabByEntity") + public ResponseMsg<Object> selectTabByEntity(String name) { + try { + BasicMapper baseMapper = ClassHelper.getBasicMapper(name); + if (baseMapper == null) { + return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + } + + String tab = BaseQueryService.getTabName(baseMapper); + + return success(tab); } catch (Exception ex) { return fail(ex, null); } @@ -219,11 +358,8 @@ if (StringHelper.isEmpty(field) || StringHelper.isSqlInjection(field)) { return fail("鏌ヨ瀛楁涓嶆纭�", null); } - if (!StringHelper.isEmpty(value)) { - value = value.trim().replace("'", ""); - } - value = StringHelper.getLikeStr2(value); + value = StringHelper.isEmpty(value) ? "%" : StringHelper.getLikeUpperStr(value); List<String> rs = baseMapper.selectFieldFuzzy(tab, field, value); return success(rs); @@ -266,10 +402,14 @@ @SysLog() @ApiOperation(value = "鏌ヨ鎵�鏈夎〃") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = "鐐�"), + @ApiImplicitParam(name = "hasGeom", value = "鍚湁Geom瀛楁", dataType = "Boolean", paramType = "query", example = "false") + }) @GetMapping(value = "/selectTabs") - public ResponseMsg<List<TabEntity>> selectTabs() { + public ResponseMsg<List<TabEntity>> selectTabs(String name, Boolean hasGeom) { try { - List<TabEntity> list = baseQueryService.selectTabs(); + List<TabEntity> list = dictService.selectDictTab(name, null == hasGeom || !hasGeom ? "gid" : "geom"); return success(list); } catch (Exception ex) { @@ -284,10 +424,12 @@ @ApiImplicitParam(name = "depcode", value = "鍗曚綅缂栫爜", dataType = "String", paramType = "query", example = "00"), @ApiImplicitParam(name = "dirs", value = "鐩綍缂栫爜", dataType = "String", paramType = "query", example = "00,01"), @ApiImplicitParam(name = "tab", value = "琛ㄥ悕", dataType = "String", paramType = "query", example = "dlg_"), + @ApiImplicitParam(name = "types", value = "绫诲埆", dataType = "String", paramType = "query", example = "娴嬬粯锛圗SV锛�"), + @ApiImplicitParam(name = "hasGeom", value = "鍚湁Geom瀛楁", dataType = "Boolean", paramType = "query", example = "false"), @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"), @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1") }) - public ResponseMsg<List<TabEntity>> selectTabsByPage(String depcode, String dirs, String tab, Integer pageSize, Integer pageIndex) { + public ResponseMsg<Object> selectTabsByPage(String depcode, String dirs, String tab, String types, Boolean hasGeom, Integer pageSize, Integer pageIndex) { try { if (pageSize < 1 || pageIndex < 1) { return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null); @@ -302,20 +444,64 @@ } dirs = DataLibService.copeCodes(dirs, "dirid"); if (dirs != null) { - filters += String.format(" and (%s)", dirs); + filters += String.format(" and %s", dirs); } + String typesFilter = getTypesFilter(types); - int count = baseQueryService.selectTabsForCount(tab); + String field = null == hasGeom || !hasGeom ? "gid" : "geom"; + int count = baseQueryService.selectTabsForCount(tab, typesFilter, field); if (count == 0) { return success(0, null); } - List<TabEntity> rs = baseQueryService.selectTabsByPage(tab, filters, pageSize, pageSize * (pageIndex - 1)); + List<TabEntity> rs = baseQueryService.selectTabsByPage(tab, typesFilter, field, filters, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { return fail(ex, null); } + } + + /** + * 鑾峰彇绫诲埆杩囨护鏉′欢 + */ + private String getTypesFilter(String types) { + if (StringHelper.isEmpty(types)) { + return null; + } + + List<String> list = new ArrayList<>(); + for (String type : types.split(StaticData.COMMA)) { + switch (type) { + case "娴嬬粯锛圗SV锛�": + list.add("bak like '宸ョ▼鏁版嵁/娴嬬粯%'"); + break; + case "鍕樺療锛圗GE锛�": + list.add("bak like '宸ョ▼鏁版嵁/鍕樺療%'"); + break; + case "鍦扮伨锛圗GD锛�": + list.add("bak like '宸ョ▼鏁版嵁/鍦扮伨%'"); + break; + case "娲炲簱锛圗GD锛�": + list.add("bak like '宸ョ▼鏁版嵁/娲炲簱%'"); + break; + case "鍩虹娴嬬粯": + list.add("bak like '鍩虹鏁版嵁/娴嬬粯%'"); + break; + case "鍩虹鍦扮伨": + list.add("bak like '鍩虹鏁版嵁/鍦扮伨%'"); + break; + case "鍩虹鍕樺療": + list.add("bak like '鍩虹鏁版嵁/鍕樺療%'"); + break; + case "鍚堣鏁版嵁": + case "绠$悊鏁版嵁": + default: + break; + } + } + + return list.isEmpty() ? null : StringHelper.join(list, " or "); } @SysLog() @@ -435,24 +621,53 @@ } @SysLog() + @ApiOperation(value = "鏌ヨ鍏冩暟鎹腑婧㈠嚭鐨勫崟浣岻D") + @ApiImplicitParams({ + @ApiImplicitParam(name = "reqEntity", value = "璇锋眰涓嬭浇瀹炰綋", dataType = "DownloadReqEntity", paramType = "body") + }) + @ResponseBody + @PostMapping(value = "/selectMetaOverflowDep") + public ResponseMsg<Object> selectMetaOverflowDep(@RequestBody DownloadReqEntity dr, HttpServletRequest req, HttpServletResponse res) { + try { + if (null == dr || null == dr.getIds() || dr.getIds().isEmpty()) { + return fail("璇烽�夋嫨瑕佷笅杞界殑鏂囦欢ID"); + } + + UserEntity ue = tokenService.getCurrentUser(req); + if (StaticData.ADMIN.equals(ue.getUid())) { + return success(new ArrayList<String>()); + } + + List<String> list = metaService.selectMetaOverflowDep(ue, dr); + + return success(list); + } catch (Exception ex) { + return fail(ex, null); + } + } + + @SysLog() @ApiOperation(value = "鏌ヨDB涓孩鍑虹殑鍗曚綅ID") @ApiImplicitParams({ @ApiImplicitParam(name = "reqEntity", value = "璇锋眰涓嬭浇瀹炰綋", dataType = "DownloadReqEntity", paramType = "body") }) @ResponseBody @PostMapping(value = "/selectDbOverflowDep") - public ResponseMsg<Object> selectDbOverflowDep(@RequestBody DownloadReqEntity reqEntity, HttpServletRequest req, HttpServletResponse res) { + public ResponseMsg<Object> selectDbOverflowDep(@RequestBody DownloadReqEntity dr, HttpServletRequest req, HttpServletResponse res) { try { - if (null == reqEntity || null == reqEntity.getEntities() || reqEntity.getEntities().isEmpty()) { + if (null == dr || null == dr.getEntities() || dr.getEntities().isEmpty()) { return fail("璇烽�夋嫨瑕佷笅杞界殑瀹炰綋鍚�"); } - reqEntity.setWkt(AesHelper.decrypt(reqEntity.getWkt())); - if (StringHelper.isEmpty(reqEntity.getWkt())) { - return fail("璇烽�夋嫨瑕佷笅杞界殑WKT鑼冨洿"); + if (!StringHelper.isEmpty(dr.getWkt())) { + dr.setWkt(AesHelper.decrypt(dr.getWkt())); } UserEntity ue = tokenService.getCurrentUser(req); - List<String> list = dataLibService.selectDbOverflowDep(ue, reqEntity.getEntities(), reqEntity.getWkt()); + if (StaticData.ADMIN.equals(ue.getUid())) { + return success(new ArrayList<String>()); + } + + List<String> list = dataLibService.selectDbOverflowDep(ue, dr); return success(list); } catch (Exception ex) { @@ -481,34 +696,32 @@ } @SysLog() - @ApiOperation(value = "璇锋眰DB鏁版嵁涓嬭浇") + @ApiOperation(value = "涓嬭浇DB鏁版嵁") @ApiImplicitParams({ - @ApiImplicitParam(name = "dre", value = "璇锋眰涓嬭浇瀹炰綋", dataType = "DownloadReqEntity", paramType = "body") + @ApiImplicitParam(name = "dr", value = "璇锋眰涓嬭浇瀹炰綋", dataType = "DownloadReqEntity", paramType = "body") }) @ResponseBody - @PostMapping(value = "/downloadDbReq") - public ResponseMsg<Object> downloadDbReq(@RequestBody DownloadReqEntity dre, HttpServletRequest req, HttpServletResponse res) { + @PostMapping(value = "/downloadDbData") + public ResponseMsg<Object> downloadDbData(@RequestBody DownloadReqEntity dr, HttpServletRequest req) { try { - if (null == dre || StringHelper.isEmpty(dre.getPwd())) { - return fail("瀵嗙爜涓嶈兘涓虹┖"); - } - if (null == dre.getEntities() || dre.getEntities().isEmpty()) { + if (null == dr || null == dr.getEntities() || dr.getEntities().isEmpty()) { return fail("璇烽�夋嫨瑕佷笅杞界殑瀹炰綋鍚�"); } - dre.setWkt(AesHelper.decrypt(dre.getWkt())); - if (StringHelper.isEmpty(dre.getWkt())) { - return fail("璇烽�夋嫨瑕佷笅杞界殑WKT鑼冨洿"); + if (StringHelper.isEmpty(dr.getPwd())) { + return fail("瀵嗙爜涓嶈兘涓虹┖"); } - if (!DownloadService.decryptPwd(dre)) { + if (!DownloadService.decryptPwd(dr)) { return fail("瀵嗙爜瑙e瘑澶辫触", null); } - if (StringHelper.isPwdInvalid(dre.getPwd())) { + if (StringHelper.isPwdInvalid(dr.getPwd())) { return fail("瀵嗙爜涓嶇鍚堣姹�"); + } + if (!StringHelper.isEmpty(dr.getWkt())) { + dr.setWkt(AesHelper.decrypt(dr.getWkt())); } UserEntity ue = tokenService.getCurrentUser(req); - String depcode = null == dre.getDepcodes() || dre.getDepcodes().isEmpty() ? null : dre.getDepcodes().get(0); - String guid = dataLibService.createZipFile(ue, dre.getEntities(), depcode, dre.getDirs(), dre.getWkt(), dre.getPwd()); + String guid = dataLibService.downloadDbReq(ue, dr); return success(guid); } catch (Exception ex) { @@ -517,35 +730,13 @@ } @SysLog() - @ApiOperation(value = "璇锋眰瀹炰綋鏁版嵁涓嬭浇") + @ApiOperation(value = "鏌ョ湅鏂囦欢") @ApiImplicitParams({ - @ApiImplicitParam(name = "dre", value = "璇锋眰涓嬭浇瀹炰綋", dataType = "DownloadReqEntity", paramType = "body") + @ApiImplicitParam(name = "guid", value = "闄勪欢Guid", dataType = "String", paramType = "body") }) - @ResponseBody - @PostMapping(value = "/downloadEntityReq") - public ResponseMsg<Object> downloadEntityReq(@RequestBody DownloadReqEntity dre, HttpServletRequest req, HttpServletResponse res) { - try { - if (null == dre || StringHelper.isEmpty(dre.getPwd())) { - return fail("瀵嗙爜涓嶈兘涓虹┖"); - } - if (null == dre.getEntities() || dre.getEntities().isEmpty()) { - return fail("璇烽�夋嫨瑕佷笅杞界殑瀹炰綋鍚�"); - } - if (!DownloadService.decryptPwd(dre)) { - return fail("瀵嗙爜瑙e瘑澶辫触", null); - } - if (StringHelper.isPwdInvalid(dre.getPwd())) { - return fail("瀵嗙爜涓嶇鍚堣姹�"); - } - - UserEntity ue = tokenService.getCurrentUser(req); - String depcode = null == dre.getDepcodes() || dre.getDepcodes().isEmpty() ? null : dre.getDepcodes().get(0); - String guid = dataLibService.zipDbData(ue, dre.getEntities().get(0), depcode, dre.getDirs(), dre.getFilter(), dre.getPwd()); - - return success(guid); - } catch (Exception ex) { - return fail(ex, null); - } + @GetMapping(value = "/downloadForView") + public void downloadForView(String guid, HttpServletResponse res) { + metaService.downloadForView(guid, true, res); } @SysLog() @@ -567,11 +758,11 @@ return fail("鐢ㄦ埛鏈櫥褰�", null); } - int count = downloadService.selectCountForUser(ue.getId(), 4, name); + int count = downloadService.selectCountForUser(ue.getId(), "3,4", name); if (count == 0) { return success(0, null); } - List<DownloadEntity> rs = downloadService.selectByPageForUser(ue.getId(), 4, name, pageSize, pageSize * (pageIndex - 1)); + List<DownloadEntity> rs = downloadService.selectByPageForUser(ue.getId(), "3,4", name, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { -- Gitblit v1.9.3