From ed8c7a5effd0d423ce1118b680ecdca6fe732609 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期三, 02 七月 2025 16:43:13 +0800 Subject: [PATCH] Merge branch 'master' of http://192.168.11.205:9000/r/P2022036_Service --- src/main/java/com/lf/server/service/data/MetaService.java | 672 +++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 643 insertions(+), 29 deletions(-) diff --git a/src/main/java/com/lf/server/service/data/MetaService.java b/src/main/java/com/lf/server/service/data/MetaService.java index 6f052a5..3b81ffc 100644 --- a/src/main/java/com/lf/server/service/data/MetaService.java +++ b/src/main/java/com/lf/server/service/data/MetaService.java @@ -1,12 +1,31 @@ package com.lf.server.service.data; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.lf.server.entity.all.StaticData; +import com.lf.server.entity.ctrl.DownloadReqEntity; +import com.lf.server.entity.data.DownloadEntity; import com.lf.server.entity.data.MetaEntity; -import com.lf.server.helper.StringHelper; +import com.lf.server.entity.sys.AttachEntity; +import com.lf.server.entity.sys.MetaDownEntity; +import com.lf.server.entity.sys.UserEntity; +import com.lf.server.helper.*; +import com.lf.server.mapper.all.BasicMapper; +import com.lf.server.mapper.all.GeomBaseMapper; import com.lf.server.mapper.data.MetaMapper; +import com.lf.server.service.all.BaseQueryService; +import com.lf.server.service.sys.MetaDownService; +import net.lingala.zip4j.ZipFile; +import net.lingala.zip4j.model.FileHeader; +import net.lingala.zip4j.model.ZipParameters; +import org.apache.commons.logging.Log; +import org.apache.commons.logging.LogFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; -import java.util.List; +import javax.servlet.http.HttpServletResponse; +import java.io.File; +import java.io.FileInputStream; +import java.util.*; /** * 鍏冩暟鎹� @@ -15,53 +34,84 @@ @Service public class MetaService implements MetaMapper { @Autowired + PathHelper pathHelper; + + @Autowired MetaMapper metaMapper; - @Override - public Integer selectCount(Integer depid, Integer dirid, String name) { - name = StringHelper.getLikeStr(name); + @Autowired + MetaDownService metaDownService; - return metaMapper.selectCount(depid, dirid, name); + @Autowired + DownloadService downloadService; + + private final static Log log = LogFactory.getLog(MetaService.class); + + @Override + public Integer selectCount(String depcode, String dircode, Integer verid, String name) { + depcode = StringHelper.getRightLike(depcode); + dircode = StringHelper.getRightLike(dircode); + name = StringHelper.getLikeUpperStr(name); + + return metaMapper.selectCount(depcode, dircode, verid, name); } @Override - public List<MetaEntity> selectByPage(Integer depid, Integer dirid, String name, Integer limit, Integer offset) { - name = StringHelper.getLikeStr(name); + public List<MetaEntity> selectByPage(String depcode, String dircode, Integer verid, String name, Integer limit, Integer offset) { + depcode = StringHelper.getRightLike(depcode); + dircode = StringHelper.getRightLike(dircode); + name = StringHelper.getLikeUpperStr(name); - return metaMapper.selectByPage(depid, dirid, name, limit, offset); + return metaMapper.selectByPage(depcode, dircode, verid, name, limit, offset); } @Override - public Integer selectMetasForCount(Integer depid, String dirs, String name) { - name = StringHelper.getLikeStr(name); - - return metaMapper.selectMetasForCount(depid, dirs, name); + public List<MetaEntity> selectGdbByGuid(String guid) { + return metaMapper.selectGdbByGuid(guid); } @Override - public List<MetaEntity> selectMetasForPage(Integer depid, String dirs, String name, Integer limit, Integer offset) { - name = StringHelper.getLikeStr(name); + public Integer selectMetasForCount(String depcode, String dirs, String name) { + depcode = StringHelper.getRightLike(depcode); + name = StringHelper.getLikeUpperStr(name); - return metaMapper.selectMetasForPage(depid, dirs, name, limit, offset); + return metaMapper.selectMetasForCount(depcode, dirs, name); + } + + @Override + public List<MetaEntity> selectMetasForPage(String depcode, String dirs, String name, Integer limit, Integer offset) { + depcode = StringHelper.getRightLike(depcode); + name = StringHelper.getLikeUpperStr(name); + + return metaMapper.selectMetasForPage(depcode, dirs, name, limit, offset); } @Override public Integer selectCountForUpload(String name, Integer createUser, String types) { - name = StringHelper.getLikeStr(name); + name = StringHelper.getLikeUpperStr(name); return metaMapper.selectCountForUpload(name, createUser, types); } @Override public List<MetaEntity> selectByPageForUpload(String name, Integer createUser, String types, Integer limit, Integer offset) { - name = StringHelper.getLikeStr(name); + name = StringHelper.getLikeUpperStr(name); return metaMapper.selectByPageForUpload(name, createUser, types, limit, offset); } @Override - public List<MetaEntity> selectAll() { - return metaMapper.selectAll(); + public Integer selectCountByPid(Integer metaid, String name) { + name = StringHelper.getLikeUpperStr(name); + + return metaMapper.selectCountByPid(metaid, name); + } + + @Override + public List<MetaEntity> selectPageByPid(Integer metaid, String name, Integer limit, Integer offset) { + name = StringHelper.getLikeUpperStr(name); + + return metaMapper.selectPageByPid(metaid, name, limit, offset); } @Override @@ -70,13 +120,47 @@ } @Override - public MetaEntity selectByGuid(String guid, String tab) { - return metaMapper.selectByGuid(guid, tab); + public MetaEntity selectByGuid(String guid, String dircode, String tab) { + return metaMapper.selectByGuid(guid, dircode, tab); + } + + @Override + public List<MetaEntity> selectByIdsForTab(String ids) { + return metaMapper.selectByIdsForTab(ids); + } + + @Override + public List<MetaEntity> selectByIds(String ids) { + return metaMapper.selectByIds(ids); } @Override public List<MetaEntity> selectMetaFiles(List<Integer> ids) { return metaMapper.selectMetaFiles(ids); + } + + @Override + public List<MetaEntity> selectXlsAnnex(Integer[] ids, String tabs) { + return metaMapper.selectXlsAnnex(ids, tabs); + } + + @Override + public List<MetaEntity> selectMetasByDirCode(String dircode) { + dircode = StringHelper.getRightLike(dircode); + + return metaMapper.selectMetasByDirCode(dircode); + } + + @Override + public List<String> selectMetaOverflowDep(String ids, String depcode) { + depcode = StringHelper.getRightLike(depcode); + + return metaMapper.selectMetaOverflowDep(ids, depcode); + } + + @Override + public Integer selectCountByPath(String path) { + return metaMapper.selectCountByPath(path); } @Override @@ -90,13 +174,8 @@ } @Override - public Integer delete(int id) { - return metaMapper.delete(id); - } - - @Override - public Integer deletes(List<Integer> ids) { - return metaMapper.deletes(ids); + public Integer deletes(String sql, String ids) { + return metaMapper.deletes(sql, ids); } @Override @@ -108,4 +187,539 @@ public Integer updates(List<MetaEntity> list) { return metaMapper.updates(list); } + + /** + * 鍒犻櫎鍏冩暟鎹� + */ + public Integer deletes(List<Integer> list) { + String ids = StringHelper.join(list, ","); + List<MetaEntity> metas = selectByIds(ids); + + String sql = null; + List<MetaEntity> tabMetas = selectByIdsForTab(ids); + if (null != tabMetas && tabMetas.size() > 0) { + sql = getDelTabsSql(tabMetas); + } + + int rows = deletes(sql, ids); + deleteMetaFiles(metas); + + return rows; + } + + /** + * 鑾峰彇鍒犻櫎琛ㄨ褰昐QL + */ + public String getDelTabsSql(List<MetaEntity> metas) { + List<String> list = getTabDeletes(metas, metas.get(0).getTab()); + addCascadDeletes(list); + addAttachDeletes(list); + Collections.reverse(list); + + return StringHelper.join(list, ";"); + } + + /** + * 鑾峰彇鍒犻櫎琛ㄨ褰昐QL + */ + private List<String> getTabDeletes(List<MetaEntity> metas, String tab) { + List<String> list = new ArrayList<>(); + + List<String> pids = new ArrayList<>(); + for (MetaEntity me : metas) { + if (tab.equals(me.getTab())) { + pids.add("'" + me.getEventid() + "'"); + } else { + list.add(String.format("delete from %s where parentid in (%s)", tab, StringHelper.join(pids, ","))); + + pids.clear(); + tab = me.getTab(); + pids.add("'" + me.getEventid() + "'"); + } + } + if (pids.size() > 0) { + list.add(String.format("delete from %s where parentid in (%s)", tab, StringHelper.join(pids, ","))); + } + + return list; + } + + /** + * 娣诲姞绾ц仈鍒犻櫎 + */ + private void addCascadDeletes(List<String> list) { + int i = 0, c = list.size(); + while (i < c) { + String str = list.get(i); + if (str.contains("bs.m_pipelinepoint ")) { + list.add(str.replace("bs.m_pipelinepoint ", "bs.m_pipesegment ")); + list.add(str.replace("bs.m_pipelinepoint ", "bs.m_pipeline ")); + } + if (str.contains("bs.s_explorationpoint ")) { + list.add(str.replace("bs.s_explorationpoint ", "bs.s_surveyworksite ")); + } + if (str.contains("bs.m_surface_deformation_data ")) { + list.add(str.replace("bs.m_surface_deformation_data ", "bs.m_surface_deformation_data_date ")); + } + + i++; + } + } + + /** + * 娣诲姞闄勪欢鍒犻櫎 + */ + private void addAttachDeletes(List<String> list) { + int i = 0, c = list.size(); + while (i < c) { + String str = list.get(i); + String tab = str.substring("delete from ".length(), str.indexOf(" where ")); + String select = str.replace("delete ", "select eventid "); + + list.add(String.format("delete from lf.sys_attach where tab='%s' and tab_guid in (%s)", tab, select)); + + i++; + } + } + + /** + * 鍒犻櫎鍏冩暟鎹枃浠� + */ + private void deleteMetaFiles(List<MetaEntity> metas) { + String uploadPath = pathHelper.getConfig().getUploadPath(); + for (MetaEntity meta : metas) { + if (StringHelper.isEmpty(meta.getPath())) { + continue; + } + Integer rows = selectCountByPath(meta.getPath()); + if (rows > 0) { + continue; + } + + try { + String file = uploadPath + File.separator + meta.getPath(); + switch ("." + meta.getType()) { + case StaticData.MPT: + deleteFiles(file, meta.getType(), StaticData.MPT_EXT); + break; + case StaticData.JPG: + deleteFiles(file, meta.getType(), StaticData.JPG_EXT); + break; + case StaticData.IMG: + deleteFiles(file, meta.getType(), StaticData.IMG_EXT); + break; + case StaticData.TIF: + deleteFiles(file, meta.getType(), StaticData.TIF_EXT); + break; + case StaticData.TIFF: + deleteFiles(file, meta.getType(), StaticData.TIFF_EXT); + break; + case StaticData.SHP: + deleteFiles(file, meta.getType(), StaticData.SHP_EXT); + break; + case StaticData.GDB: + case StaticData.OSGB: + FileHelper.deleteDir(file); + break; + default: + deleteFiles(file, meta.getType(), null); + break; + } + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + } + } + + /** + * 鍒犻櫎澶氫釜鏂囦欢 + */ + private void deleteFiles(String file, String type, List<String> extList) { + File f = new File(file); + if (f.exists() && !f.isDirectory()) { + f.delete(); + } + if (null == extList || extList.isEmpty()) { + return; + } + + for (String ext : extList) { + f = new File(file.replace("." + type, ext)); + if (f.exists() && !f.isDirectory()) { + f.delete(); + } + } + } + + /** + * 鏌ョ湅鏂囦欢 + */ + public void downloadForView(String guid, boolean inline, HttpServletResponse res) { + try { + if (StringHelper.isEmpty(guid)) { + WebHelper.writeStr2Page(res, StaticData.NO_FILE); + return; + } + + MetaEntity me = selectByGuid(guid, null, null); + if (me == null) { + WebHelper.writeStr2Page(res, StaticData.NO_FILE); + return; + } + + String filePath = pathHelper.getConfig().getUploadPath() + File.separator + me.getPath(); + File file = new File(filePath); + if (!file.exists() || file.isDirectory()) { + WebHelper.writeJson2Page(res, "鏂囦欢涓嶅瓨鍦�"); + } + + WebHelper.download(filePath, me.getName(), inline, res); + } catch (Exception ex) { + WebHelper.writeJson2Page(res, "鏂囦欢涓嬭浇鍑洪敊"); + log.error(ex.getMessage(), ex); + } + } + + /** + * 鏌ヨ鍏冩暟鎹腑婧㈠嚭鐨勫崟浣岻D + */ + public List<String> selectMetaOverflowDep(UserEntity ue, DownloadReqEntity dr) { + String ids = StringHelper.join(dr.getIds(), ","); + + return selectMetaOverflowDep(ids, ue.getDepcode()); + } + + /** + * 璇锋眰鍏冩暟鎹笅杞� + * + * @param ue 鐢ㄦ埛瀹炰綋 + * @param dr 璇锋眰涓嬭浇瀹炰綋 + * @return 涓嬭浇鏂囦欢GUID + * @throws Exception 寮傚父 + */ + public String downloadMeteReq(UserEntity ue, DownloadReqEntity dr) throws Exception { + List<MetaEntity> list = selectMetaFiles(dr.getIds()); + if (null == list || list.isEmpty()) { + return null; + } + + Map<String, List<String>> tabs = getTabs(list); + rmRepeatMetas(list); + + Map<String, List<?>> dataMap = new HashMap<>(2); + Map<String, List<AttachEntity>> annexMap = new HashMap<>(2); + queryData(tabs, dataMap, annexMap); + + String tempName = StringHelper.YMDHMS2_FORMAT.format(new Date()); + String tempPath = pathHelper.getTempPath(tempName); + String gdbPath = tempPath + File.separator + "tabs.gdb"; + + File gdbFile = new File(gdbPath); + if (gdbFile.exists() && gdbFile.isDirectory()) { + FileHelper.deleteDir(gdbPath); + } + if (dataMap.size() > 0) { + GdbHelper.createGdb(gdbPath, dataMap); + } + + String zipFile = pathHelper.getDownloadFullPath() + File.separator + tempName + ".zip"; + ZipFile zip = Zip4jHelper.createZipFile(zipFile, dr.getPwd()); + ZipParameters params = Zip4jHelper.getZipParams(true); + addMetaFiles(zip, params, list); + if (dataMap.size() > 0) { + zip.addFolder(new File(gdbPath), params); + addAnnex(zip, params, annexMap); + } + + String dbPwd = Md5Helper.reverse(Md5Helper.generate(dr.getPwd())); + DownloadEntity de = getDownloadEntity(ue, zipFile, dbPwd); + int rows = downloadService.insert(de); + if (de.getId() > 0) { + insertMetaDown(ue, list, de); + } + + return rows > 0 ? de.getGuid() : null; + } + + /** + * 鑾峰彇鏁版嵁琛� + */ + private Map<String, List<String>> getTabs(List<MetaEntity> list) { + Map<String, List<String>> tabs = new HashMap<>(2); + for (MetaEntity meta : list) { + if (StringHelper.isEmpty(meta.getTab()) || meta.getRows() == 0 || StringHelper.isEmpty(meta.getEventid())) { + continue; + } + + if (!tabs.containsKey(meta.getTab())) { + tabs.put(meta.getTab(), new ArrayList<>()); + } + + List<String> ids = tabs.get(meta.getTab()); + if (!ids.contains(meta.getEventid())) { + ids.add(meta.getEventid()); + } + } + + return tabs; + } + + /** + * 绉婚櫎閲嶅鐨勫厓鏁版嵁鏂囦欢 + */ + private void rmRepeatMetas(List<MetaEntity> list) { + List<String> guidList = new ArrayList<>(); + + int i = 0; + while (i < list.size()) { + MetaEntity entity = list.get(i); + if (guidList.contains(entity.getGuid())) { + list.remove(i); + continue; + } + + guidList.add(entity.getGuid()); + i++; + } + } + + /** + * 鏌ヨ鏁版嵁 + */ + private void queryData(Map<String, List<String>> tabs, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) { + for (String tab : tabs.keySet()) { + try { + String entity = tab.toLowerCase().replace("_", "").split("\\.")[1]; + BasicMapper baseMapper = ClassHelper.getBasicMapper(entity); + if (null == baseMapper) { + continue; + } + + QueryWrapper wrapper = createQueryWrapper(baseMapper, tabs.get(tab)); + addData(entity, baseMapper, wrapper, dataMap, annexMap); + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + } + } + + /** + * 娣诲姞鏁版嵁 + */ + public void addData(String entity, BasicMapper baseMapper, QueryWrapper wrapper, Map<String, List<?>> dataMap, Map<String, List<AttachEntity>> annexMap) { + List list = baseMapper.selectList(wrapper); + if (null == list || list.size() == 0) { + return; + } + + if (!dataMap.containsKey(entity)) { + dataMap.put(entity, list); + } else { + dataMap.get(entity).addAll(list); + } + if (wrapper.isEmptyOfWhere()) { + wrapper.apply("1 = 1"); + } + if (StaticData.BBOREHOLE.equals(entity)) { + wrapper.last("limit 100"); + } + + String tab = BaseQueryService.getTabName(baseMapper); + List<AttachEntity> annex = baseMapper.selectAnnex(tab, wrapper); + if (null == annex || annex.isEmpty()) { + return; + } + + if (!annexMap.containsKey(entity)) { + annexMap.put(tab.replace(StaticData.POINT, "_"), annex); + } else { + annexMap.get(tab.replace(StaticData.POINT, "_")).addAll(annex); + } + } + + /** + * 鍒涘缓鏌ヨ鍖呰鍣� + */ + private <T> QueryWrapper<T> createQueryWrapper(BasicMapper baseMapper, List<String> ids) { + for (int i = 0, c = ids.size(); i < c; i++) { + ids.set(i, "'" + ids.get(i) + "'"); + } + String filter = String.format("parentid in (%s)", StringHelper.join(ids, ",")); + + QueryWrapper<T> wrapper = new QueryWrapper<T>(); + if (baseMapper instanceof GeomBaseMapper) { + wrapper.select("ST_AsText(geom) as geom, *"); + } + wrapper.apply(filter); + + return wrapper; + } + + /** + * 娣诲姞闄勪欢 + */ + public void addAnnex(ZipFile zip, ZipParameters params, Map<String, List<AttachEntity>> annexMap) { + List<String> files = new ArrayList<>(); + String uploadPath = pathHelper.getConfig().getUploadPath(); + for (String key : annexMap.keySet()) { + for (AttachEntity ae : annexMap.get(key)) { + try { + File srcFile = new File(uploadPath + File.separator + ae.getPath()); + if (!srcFile.exists() || srcFile.isDirectory()) { + continue; + } + if (files.contains(srcFile.getPath())) { + continue; + } + + files.add(srcFile.getPath()); + params.setFileNameInZip("annex" + File.separator + key + File.separator + ae.getName()); + zip.addStream(new FileInputStream(srcFile), params); + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + } + } + } + + /** + * 娣诲姞鍏冩暟鎹枃浠惰嚦Zip鍖� + */ + private void addMetaFiles(ZipFile zip, ZipParameters params, List<MetaEntity> list) { + List<String> names = new ArrayList<>(); + String uploadPath = pathHelper.getConfig().getUploadPath(); + for (MetaEntity mf : list) { + if (names.contains(mf.getName())) { + mf.setName(mf.getId() + "_" + mf.getName()); + } else { + names.add(mf.getName()); + } + + try { + switch ("." + mf.getType()) { + case StaticData.MPT: + addMultiFile(uploadPath, mf, zip, params, StaticData.MPT_EXT); + break; + case StaticData.JPG: + addMultiFile(uploadPath, mf, zip, params, StaticData.JPG_EXT); + break; + case StaticData.IMG: + addMultiFile(uploadPath, mf, zip, params, StaticData.IMG_EXT); + break; + case StaticData.TIF: + addMultiFile(uploadPath, mf, zip, params, StaticData.TIF_EXT); + break; + case StaticData.TIFF: + addMultiFile(uploadPath, mf, zip, params, StaticData.TIFF_EXT); + break; + case StaticData.SHP: + addMultiFile(uploadPath, mf, zip, params, StaticData.SHP_EXT); + break; + case StaticData.GDB: + continue; + case StaticData.OSGB: + addFolderFile(uploadPath, mf, zip, params); + break; + default: + addSingleFile(uploadPath, mf, zip, params); + break; + } + } catch (Exception ex) { + log.error(ex.getMessage(), ex); + } + } + } + + /** + * 娣诲姞鐩綍鏂囦欢 + */ + private void addFolderFile(String uploadPath, MetaEntity mf, ZipFile zip, ZipParameters params) throws Exception { + File file = new File(uploadPath + File.separator + mf.getPath()); + if (!file.exists() || !file.isDirectory()) { + return; + } + zip.addFolder(file, params); + + String fileName = FileHelper.getFileName(file.getPath()); + FileHeader header = zip.getFileHeader(fileName); + if (null != header) { + zip.renameFile(header, mf.getName()); + } + } + + /** + * 娣诲姞澶氭枃浠� + */ + private void addMultiFile(String uploadPath, MetaEntity mf, ZipFile zip, ZipParameters params, List<String> extList) throws Exception { + addSingleFile(uploadPath, mf, zip, params); + + for (String ext : extList) { + File file = new File(uploadPath + File.separator + mf.getPath().replace("." + mf.getType(), ext)); + if (!file.exists() || file.isDirectory()) { + continue; + } + zip.addFile(file, params); + + String fileName = FileHelper.getFileName(file.getPath()); + FileHeader header = zip.getFileHeader(fileName); + if (null != header) { + zip.renameFile(header, mf.getName().replace("." + mf.getType(), ext)); + } + } + } + + /** + * 娣诲姞鍗曟枃浠� + */ + private void addSingleFile(String uploadPath, MetaEntity mf, ZipFile zip, ZipParameters params) throws Exception { + File file = new File(uploadPath + File.separator + mf.getPath()); + if (!file.exists() || file.isDirectory()) { + return; + } + zip.addFile(file, params); + + String fileName = FileHelper.getFileName(file.getPath()); + FileHeader header = zip.getFileHeader(fileName); + if (null != header) { + zip.renameFile(header, mf.getName()); + } + } + + /** + * 鑾峰彇涓嬭浇瀹炰綋绫� + */ + private DownloadEntity getDownloadEntity(UserEntity ue, String file, String pwd) { + DownloadEntity de = new DownloadEntity(); + de.setName(FileHelper.getFileName(file)); + // 1-Shp鏂囦欢锛�2-涓撻鍥撅紝3-鍏冩暟鎹紝4-涓氬姟鏁版嵁锛�5-绠¢亾鍒嗘瀽锛�6-缁熻鎶ュ憡锛�7-闄勪欢鏂囦欢锛�8-鐡︾墖鏂囦欢 + de.setType(3); + de.setSizes(FileHelper.sizeToMb(new File(file).length())); + de.setDepid(ue.getDepid()); + de.setDcount(0); + de.setPwd(pwd); + de.setUrl(FileHelper.getRelativePath(file)); + de.setDescr("鍏冩暟鎹枃浠�"); + de.setGuid(FileHelper.getFileMd5(file)); + de.setCreateUser(ue.getId()); + // de.setGeom(null) + + return de; + } + + /** + * 鎻掑叆鍏冩暟鎹�-涓嬭浇琛� + */ + private void insertMetaDown(UserEntity ue, List<MetaEntity> metas, DownloadEntity de) { + List<MetaDownEntity> list = new ArrayList<>(); + for (MetaEntity me : metas) { + MetaDownEntity md = new MetaDownEntity(); + md.setMetaid(me.getId()); + md.setDownid(de.getId()); + md.setCreateUser(ue.getId()); + + list.add(md); + } + + metaDownService.inserts(list); + } } -- Gitblit v1.9.3