From e26af85e049516e6ce2b082bc2bc90bf71643e95 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期五, 24 三月 2023 09:06:30 +0800 Subject: [PATCH] 1 --- src/main/java/com/lf/server/controller/all/BaseQueryController.java | 309 ++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 230 insertions(+), 79 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 91e41f5..e57f134 100644 --- a/src/main/java/com/lf/server/controller/all/BaseQueryController.java +++ b/src/main/java/com/lf/server/controller/all/BaseQueryController.java @@ -6,13 +6,12 @@ import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.lf.server.annotation.SysLog; 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; import com.lf.server.entity.ctrl.TabEntity; -import com.lf.server.entity.data.DictEntity; -import com.lf.server.entity.data.DomainEntity; -import com.lf.server.entity.data.DownloadEntity; +import com.lf.server.entity.data.*; import com.lf.server.entity.sys.AttachEntity; import com.lf.server.entity.sys.UserEntity; import com.lf.server.helper.AesHelper; @@ -21,7 +20,10 @@ 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; @@ -51,6 +53,12 @@ DepService depService; @Autowired + DirService dirService; + + @Autowired + MetaService metaService; + + @Autowired DataLibService dataLibService; @Autowired @@ -58,6 +66,9 @@ @Autowired BaseQueryService baseQueryService; + + @Autowired + DictService dictService; @SysLog() @ApiOperation(value = "鏌ヨ璁板綍鏁�") @@ -83,7 +94,7 @@ return success(count); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -91,6 +102,8 @@ @ApiOperation(value = "鍒嗛〉鏌ヨ") @ApiImplicitParams({ @ApiImplicitParam(name = "name", value = "鏄犲皠鍚嶇О", dataType = "String", paramType = "query", example = "dlgagnp"), + @ApiImplicitParam(name = "depcode", value = "鍗曚綅缂栫爜", dataType = "String", paramType = "query", example = "00"), + @ApiImplicitParam(name = "dirs", value = "鐩綍缂栫爜", dataType = "String", paramType = "query", example = "00,01"), @ApiImplicitParam(name = "filter", value = "杩囨护鏉′欢", dataType = "String", paramType = "query", example = "gid >= 50 and name like '鍘�' and objectid < 5000.0"), @ApiImplicitParam(name = "wkt", value = "WKT锛堣憲鍚嶆枃鏈級", dataType = "String", paramType = "query", example = ""), //@ApiImplicitParam(name = "srid", value = "绌洪棿寮曠敤鏍囪瘑绗�", dataType = "Integer", paramType = "query", example = "4326"), @@ -99,7 +112,7 @@ }) @GetMapping(value = "/selectByPage") @PostMapping(value = "/selectByPage") - public ResponseMsg<List<Object>> selectByPage(String name, String filter, String wkt, Integer pageIndex, Integer pageSize) { + public ResponseMsg<List<Object>> selectByPage(String name, String depcode, String dirs, String filter, String wkt, Integer pageIndex, Integer pageSize) { try { BasicMapper baseMapper = ClassHelper.getBasicMapper(name); if (baseMapper == null) { @@ -109,14 +122,92 @@ QueryWrapper wrapper = new QueryWrapper(); baseQueryService.addFilterWrapper(wrapper, filter); baseQueryService.addGeomWrapper(baseMapper, wrapper, wkt); + if (!StringHelper.isEmpty(depcode)) { + wrapper.likeRight("depid", depcode); + } + + dirs = DataLibService.copeCodes(dirs, "dirid"); + if (!StringHelper.isEmpty(dirs)) { + wrapper.apply(dirs); + } Page<Object> page = new Page<>(pageIndex, pageSize); - page.addOrder(OrderItem.asc("gid")); + page.addOrder(OrderItem.desc("gid")); IPage<Object> paged = baseMapper.selectPage(page, wrapper); return success(paged.getTotal(), paged.getRecords()); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); + } + } + + @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.ONE_HUNDRED) { + limit = 20; + } + if (null == buffer || buffer < 0 || buffer > StaticData.ONE_HUNDRED_THOUSAND) { + 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); + } + } + + @SysLog() + @ApiOperation(value = "鏍规嵁鐖禝D鍒嗛〉鏌ヨ骞惰繑鍥炶褰曟暟") + @ApiImplicitParams({ + @ApiImplicitParam(name = "metaid", value = "鐖禝D", dataType = "String", paramType = "query", example = "0"), + @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = ""), + @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"), + @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1") + }) + @GetMapping(value = "/selectPageAndCountByPid") + public ResponseMsg<List<MetaEntity>> selectPageAndCountByPid(Integer metaid, String name, Integer pageSize, Integer pageIndex) { + try { + if (pageSize < 1 || pageIndex < 1) { + return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null); + } + if (null == metaid || metaid < 1) { + return fail("鐖禝D涓嶈兘涓虹┖涓斿ぇ浜�1", null); + } + + int count = metaService.selectCountByPid(metaid, name); + if (count == 0) { + return success(0, null); + } + + List<MetaEntity> rs = metaService.selectPageByPid(metaid, name, pageSize, pageSize * (pageIndex - 1)); + + return success(count, rs); + } catch (Exception ex) { + return fail(ex, null); } } @@ -146,7 +237,54 @@ return success(wkt); } catch (Exception ex) { - return fail(ex.getMessage(), null); + 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 { + GeomBaseMapper baseMapper = ClassHelper.getGeoBaseMapper(name); + if (baseMapper == null) { + return fail("鏌ヨ瀵硅薄涓嶅瓨鍦�", null); + } + + QueryWrapper wrapper = new QueryWrapper(); + wrapper.select("ST_AsText(geom) as geom, *"); + wrapper.eq("gid", gid); + + Object obj = baseMapper.selectOne(wrapper); + + 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); } } @@ -172,16 +310,13 @@ 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); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -197,7 +332,7 @@ return success(rs); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -213,20 +348,24 @@ return success(rs); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @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) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -234,26 +373,42 @@ @ApiOperation(value = "鏍规嵁琛ㄥ悕鍒嗛〉鏌ヨ琛�") @GetMapping(value = "/selectTabsByPage") @ApiImplicitParams({ + @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 = "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 tab, Integer pageSize, Integer pageIndex) { + public ResponseMsg<Object> selectTabsByPage(String depcode, String dirs, String tab, Boolean hasGeom, Integer pageSize, Integer pageIndex) { try { if (pageSize < 1 || pageIndex < 1) { return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null); } + if (StringHelper.isSqlInjection(depcode)) { + return fail("鍗曚綅浠g爜鍚湁闈炴硶瀛楃", null); + } - int count = baseQueryService.selectTabsForCount(tab); + String filters = "1=1"; + if (!StringHelper.isEmpty(depcode)) { + filters += String.format(" and depid like '%s%%'", depcode); + } + dirs = DataLibService.copeCodes(dirs, "dirid"); + if (dirs != null) { + filters += String.format(" and (%s)", dirs); + } + + String field = null == hasGeom || !hasGeom ? "gid" : "geom"; + int count = baseQueryService.selectTabsForCount(tab, field); if (count == 0) { return success(0, null); } - List<TabEntity> rs = baseQueryService.selectTabsByPage(tab, pageSize, pageSize * (pageIndex - 1)); + List<TabEntity> rs = baseQueryService.selectTabsByPage(tab, field, filters, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -280,7 +435,7 @@ return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -301,7 +456,7 @@ return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -328,7 +483,7 @@ return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -349,7 +504,7 @@ return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -369,7 +524,7 @@ return success(rows); } catch (Exception ex) { - return fail(ex.getMessage(), 0); + return fail(ex, 0); } } @@ -395,27 +550,27 @@ return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @SysLog() - @ApiOperation(value = "鏍规嵁鍗曚綅ID闆嗗悎鏌ヨ鍗曚綅") + @ApiOperation(value = "鏍规嵁鍗曚綅缂栫爜鏁扮粍鏌ヨ鍗曚綅") @ApiImplicitParams({ - @ApiImplicitParam(name = "ids", value = "鍗曚綅ID闆嗗悎", dataType = "Integer", paramType = "query", allowMultiple = true, example = "1,10") + @ApiImplicitParam(name = "codes", value = "鍗曚綅缂栫爜鏁扮粍", dataType = "String", paramType = "query", allowMultiple = true, example = "00,0001") }) - @GetMapping(value = "/selectDepsByIds") - public ResponseMsg<Object> selectDepsByIds(Integer[] ids) { + @GetMapping(value = "/selectDepsByCodes") + public ResponseMsg<Object> selectDepsByCodes(String[] codes) { try { - if (null == ids || ids.length == 0) { + if (null == codes || codes.length == 0) { return fail("鍗曚綅ID闆嗗悎涓虹┖"); } - List<IdNameEntity> list = depService.selectDepsByIds(ids); + List<IdNameEntity> list = depService.selectDepsByCodes(codes); return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -446,11 +601,12 @@ } UserEntity ue = tokenService.getCurrentUser(req); - String guid = dataLibService.createZipFile(ue, dre.getEntities(), dre.getWkt(), dre.getPwd()); + 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()); return success(guid); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @@ -477,12 +633,23 @@ } UserEntity ue = tokenService.getCurrentUser(req); - String guid = dataLibService.zipDbData(ue, dre.getEntities().get(0), dre.getFilter(), dre.getPwd()); + 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.getMessage(), null); + return fail(ex, null); } + } + + @SysLog() + @ApiOperation(value = "鏌ョ湅鏂囦欢") + @ApiImplicitParams({ + @ApiImplicitParam(name = "guid", value = "闄勪欢Guid", dataType = "String", paramType = "body") + }) + @GetMapping(value = "/downloadForView") + public void downloadForView(String guid, HttpServletResponse res) { + metaService.downloadForView(guid, true, res); } @SysLog() @@ -504,79 +671,63 @@ return fail("鐢ㄦ埛鏈櫥褰�", null); } - int count = downloadService.selectCountForUser(ue.getCreateUser(), 4, name); + int count = downloadService.selectCountForUser(ue.getId(), 4, name); if (count == 0) { return success(0, null); } - List<DownloadEntity> rs = downloadService.selectByPageForUser(ue.getCreateUser(), 4, name, pageSize, pageSize * (pageIndex - 1)); + List<DownloadEntity> rs = downloadService.selectByPageForUser(ue.getId(), 4, name, pageSize, pageSize * (pageIndex - 1)); return success(count, rs); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @SysLog() - @ApiOperation(value = "鏌ヨ鍩虹鍦扮悊绫诲埆") - @GetMapping(value = "/selectBaseType") - public ResponseMsg<Object> selectBaseType() { + @ApiOperation(value = "鏌ヨ鐩綍绫诲埆") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = "D") + }) + @GetMapping(value = "/selectDirTypes") + public ResponseMsg<Object> selectDirTypes(String name) { try { - List<KeyValueEntity> list = baseQueryService.selectBaseType(); + List<KeyValueEntity> list = baseQueryService.selectDirTypes(name); return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); - } - } - - @SysLog() - @ApiOperation(value = "鏌ヨ涓氬姟绫诲埆") - @GetMapping(value = "/selectBusinessType") - public ResponseMsg<Object> selectBusinessType() { - try { - List<KeyValueEntity> list = baseQueryService.selectBusinessType(); - - return success(list); - } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @SysLog() @ApiOperation(value = "鏌ヨ椤圭洰鍚嶇О") + @ApiImplicitParams({ + @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = "瑗�") + }) @GetMapping(value = "/selectProject") - public ResponseMsg<Object> selectProject() { + public ResponseMsg<Object> selectProject(String name) { try { - List<KeyValueEntity> list = baseQueryService.selectProject(); + List<DirEntity> list = dirService.selectProject(name); return success(list); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } @SysLog() - @ApiOperation(value = "鏌ヨ瀛楀吀琛ㄥ苟缁熻璁板綍") + @ApiOperation(value = "鏍规嵁鍏冩暟鎹甀D鏌ヨ") @ApiImplicitParams({ - @ApiImplicitParam(name = "depcode", value = "鍗曚綅缂栫爜", dataType = "String", paramType = "query", example = "00"), - @ApiImplicitParam(name = "dirs", value = "鐩綍缂栫爜", dataType = "String", paramType = "query", example = "00,01"), - @ApiImplicitParam(name = "name", value = "鍚嶇О", dataType = "String", paramType = "query", example = ""), - @ApiImplicitParam(name = "pageSize", value = "姣忛〉鏉℃暟", dataType = "Integer", paramType = "query", example = "10"), - @ApiImplicitParam(name = "pageIndex", value = "鍒嗛〉鏁帮紙浠�1寮�濮嬶級", dataType = "Integer", paramType = "query", example = "1") + @ApiImplicitParam(name = "id", value = "ID", dataType = "int", paramType = "query", example = "1") }) - @GetMapping(value = "/selectTabCount") - public ResponseMsg<Object> selectTabCount(String depcode, String dirs, String name, Integer pageSize, Integer pageIndex) { + @GetMapping(value = "/selectMetaById") + public ResponseMsg<MetaEntity> selectMetaById(int id) { try { - if (pageSize < 1 || pageIndex < 1) { - return fail("姣忛〉椤垫暟鎴栧垎椤垫暟灏忎簬1", null); - } - dirs = DataLibService.copeCodes(dirs, "dirid"); - - List<DictEntity> list = baseQueryService.selectTabCount(depcode, dirs, name, pageSize, pageSize * (pageIndex - 1)); + MetaEntity entity = metaService.selectById(id); - return success(list); + return success(entity); } catch (Exception ex) { - return fail(ex.getMessage(), null); + return fail(ex, null); } } } \ No newline at end of file -- Gitblit v1.9.3