package com.fdkankan.fusion.down; import cn.hutool.core.io.FileUtil; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.deepoove.poi.XWPFTemplate; import com.fdkankan.fusion.common.FilePath; import com.fdkankan.fusion.common.ResultData; import com.fdkankan.fusion.common.util.*; import com.fdkankan.fusion.config.CacheUtil; import com.fdkankan.fusion.entity.*; import com.fdkankan.fusion.httpClient.LaserService; import com.fdkankan.fusion.httpClient.response.FdkkResponse; import com.fdkankan.fusion.request.CaseParam; import com.fdkankan.fusion.response.DownVo; import com.fdkankan.fusion.response.DownloadProcessVo; import com.fdkankan.fusion.response.FusionNumVo; import com.fdkankan.fusion.response.SceneVo; import com.fdkankan.fusion.service.*; import com.fdkankan.fusion.service.impl.DownService; import com.fdkankan.redis.util.RedisUtil; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang3.StringUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.scheduling.annotation.Async; import org.springframework.stereotype.Service; import java.io.File; import java.io.FileOutputStream; import java.nio.charset.StandardCharsets; import java.util.HashMap; import java.util.HashSet; import java.util.List; @Service @Slf4j public class FusionDownService { @Value("${server.servlet.context-path}") String basePath; public static String jsonDataName = "data.json"; public static String caseFusionInfo = "/caseFusion/info?fusionId="; public static String caseFusionList= "/caseFusion/list?fusionId="; public static String mapConfig = "/notAuth/getMapConfig"; public static String caseSettingsInfo = "/caseSettings/info?fusionId="; public static String hostIconTreeList = "/edit/hotIcon/treeList?fusionId="; public static String caseTag = "/caseTag/allList?fusionId="; public static String caseTagPoint = "/caseTagPoint/allList?tagId="; public static String fusionGuide = "/fusionGuide/allList?fusionId="; public static String fusionGuidePath = "/fusionGuidePath/allList?guideId="; public static String fusionMeter = "/fusionMeter/allList?fusionId="; public static String casePathInfoUrl = "/casePath/info?fusionId="; public static String caseAnimationUrl = "/caseAnimation/list?fusionId="; public static String laserData = "/laser/dataset/%s/getDataSet"; public static String laserDataSetAndControlPoint = "/laser/4dage/%s/getDataSetAndControlPoint"; public static String laserDataQuery = "/laser/filter/%s/query?datasetId=%s"; @Autowired ICaseSettingsService caseSettingsService; @Autowired IFusionNumService fusionNumService; @Autowired ICaseViewService caseViewService; @Autowired ICaseVideoFolderService caseVideoFolderService; @Autowired ICaseVideoService caseVideoService; @Autowired ICaseFilesService caseFilesService; @Autowired ICaseFilesTypeService caseFilesTypeService; @Autowired IHotIconService hotIconService; @Autowired ICaseTagService caseTagService; @Autowired ICaseTagPointService caseTagPointService; @Autowired IFusionGuideService fusionGuideService; @Autowired IFusionGuidePathService fusionGuidePathService; @Autowired ICaseInquestService caseInquestService; @Autowired ICaseExtractDetailService caseExtractDetailService; @Autowired IFusionMeterService fusionMeterService; @Autowired IModelService modelService; @Autowired LaserService laserService; @Autowired ICaseOfflineService caseOfflineService; @Autowired ICaseImgService caseImgService; @Autowired ICasePathService casePathService; @Autowired ICaseFusionService caseFusionService; @Autowired ICaseAnimationService caseAnimationService; @Autowired IMapConfigService mapConfigService; @Autowired RedisUtil redisUtil; public static String downProcessKey = "fusion:down:offline:process:fusionId:%s"; @Async public void downOffline(Integer fusionId){ String caseOffPath = null; String zipName = null; try { String redisKey = String.format(downProcessKey, fusionId); if(redisUtil.hasKey(redisKey)){ String res = redisUtil.get(redisKey); DownloadProcessVo downloadProcessVo = JSONObject.parseObject(res, DownloadProcessVo.class); if(downloadProcessVo.getStatus() != 1003 && (downloadProcessVo.getPercent()== null || downloadProcessVo.getPercent() != 100)){ return; } } String timeKey = DateUtils.dateStr(); caseOffPath =FilePath.OFFLINE_PACKAGE_PATH+timeKey+fusionId ; setRedisProcess(fusionId,0); //复制前端资源 cpIndexHtml(fusionId,caseOffPath); setRedisProcess(fusionId,10); //创建data.json并下载资源 createDataJson(fusionId,caseOffPath+"/www"); //打包zip zipName = caseOffPath+".zip"; ShellUtil.zip(zipName,caseOffPath); setRedisProcess(fusionId,70); //上传oss String ossUrl = zipName.replace("/mnt/", ""); ShellUtil.yunUpload(zipName,ossUrl); ossUrl = queryPath + ossUrl; setRedisProcess(fusionId,100,ossUrl); }catch (Exception e){ log.info("down-offline-error:{}",fusionId,e); setRedisProcess(fusionId,0,null,1003); }finally { delFile(caseOffPath); delFile(zipName); } } private void delFile(String path){ if(StringUtils.isNotBlank(path)){ try { FileUtil.del(path); }catch (Exception e){ log.info("删除文件失败:{}",path); } } } public void setRedisProcess(Integer caseId,Integer num){ setRedisProcess(caseId,num,null,1000); } public void setRedisProcess(Integer caseId,Integer num,String url){ setRedisProcess(caseId,num,url,1000); } public void setRedisProcess(Integer caseId,Integer num,String url,Integer status){ String redisKey = String.format(downProcessKey, caseId); log.info("down-offline-process:{},{},{}",caseId,num,url); DownloadProcessVo processVo = new DownloadProcessVo(); processVo.setStatus(status); processVo.setPercent(num); processVo.setUrl( url); redisUtil.set(redisKey,JSONObject.toJSONString(processVo)); } public DownloadProcessVo process(Integer caseId) { DownloadProcessVo downVo = new DownloadProcessVo(); String redisKey = String.format(downProcessKey, caseId); if(redisUtil.hasKey(redisKey)){ return JSONObject.parseObject(redisUtil.get(redisKey),DownloadProcessVo.class); } return downVo; } public void createDataJson(Integer fusionId,String caseOffPath){ log.info("down-offline-createDataJson:{}",fusionId); JSONObject jsonObject = new JSONObject(); CaseFusion caseFusion = caseFusionService.getById(fusionId); jsonObject.put(basePath+caseFusionInfo+fusionId, ResultData.ok(caseFusion)); List sceneListVo = caseFusionService.getSceneListVo(fusionId); List list = mapConfigService.list(); jsonObject.put(basePath+mapConfig, ResultData.ok(list)); List caseSettings = caseSettingsService.getByFusionId(fusionId); jsonObject.put(basePath+caseSettingsInfo+fusionId, ResultData.ok(caseSettings)); for (CaseSettings caseSetting : caseSettings) { downResource(caseSetting.getBack(),caseOffPath); downResource(caseSetting.getCover(),caseOffPath); } HashMap repMap = new HashMap<>(); jsonObject.put(basePath+fusionMeter+fusionId, ResultData.ok(fusionMeterService.getListByFusionId(fusionId,null))); List fusionNumVos = caseFusionService.getSceneListVo(fusionId); for (FusionNumVo fusionNumVo : fusionNumVos) { SceneVo sceneData = fusionNumVo.getSceneData(); //下载模型 if(StringUtils.isNotBlank(sceneData.getModelGlbUrl())){ downModel(sceneData.getModelGlbUrl(),caseOffPath); } if(sceneData.getType() != 3){ //下载场景离线包 repMap.put(sceneData.getCutModelPath(),sceneData.getCutModelPath().replace("/"+ CacheUtil.queryPath+"/","/swkk/"+sceneData.getNum() +"/wwwroot/")); repMap.put(sceneData.getSceneJsonPath(),sceneData.getSceneJsonPath().replace("/"+CacheUtil.queryPath+"/","/swkk/"+sceneData.getNum() +"/wwwroot/")); repMap.put(sceneData.getSurveillancePath(),sceneData.getSurveillancePath().replace("/"+CacheUtil.queryPath+"/","/swkk/"+sceneData.getNum() +"/wwwroot/")); downSwkk(caseOffPath,sceneData.getNum(),sceneData.getType()); } if(NumTypeUtils.isLaser(sceneData.getType())){ FdkkResponse sceneInfo = laserService.getSceneInfo(sceneData.getNum()); HashSet dataSetIds = new HashSet<>(); if(sceneInfo != null){ JSONArray jsonArray = JSONArray.parseArray(JSONArray.toJSONString(sceneInfo.getData())); JSONArray newJsonArray = new JSONArray(); for (Object object : jsonArray) { JSONObject sceneInfoObj = (JSONObject) object; String newPath = String.format(FilePath.OFFLINE_LASER_OSS_PATH, sceneData.getNum(), sceneData.getNum()); String oldPath = sceneInfoObj.getString("webBin"); sceneInfoObj.put("oldWebBin",oldPath); sceneInfoObj.put("webBin",newPath + oldPath); newJsonArray.add(sceneInfoObj); dataSetIds.add( sceneInfoObj.getString("id")); } sceneInfo.setData(newJsonArray); jsonObject.put(String.format(laserData,sceneData.getNum()),sceneInfo); if(!dataSetIds.isEmpty()){ for (String dataSetId : dataSetIds) { FdkkResponse sceneInfo2 = laserService.getSceneInfoQuery(sceneData.getNum(),dataSetId); jsonObject.put(String.format(laserDataQuery,sceneData.getNum(),dataSetId),sceneInfo2); } } } FdkkResponse dataSetAndControlPoint = laserService.getDataSetAndControlPoint(sceneData.getNum()); if(dataSetAndControlPoint !=null){ jsonObject.put(String.format(laserDataSetAndControlPoint,sceneData.getNum()),dataSetAndControlPoint); } } } jsonObject.put(basePath+caseFusionList+fusionId, ResultData.ok(sceneListVo)); List hotIconList = hotIconService.getListByFusionId(fusionId); for (HotIcon hotIcon : hotIconList) { downResource(hotIcon.getIconUrl(),caseOffPath); } // jsonObject.put(basePath+hostIcon+fusionId, ResultData.ok(hotIconList)); List treeList = hotIconService.treeList(hotIconList); jsonObject.put(basePath+hostIconTreeList+fusionId, ResultData.ok(treeList)); List caseTagList = caseTagService.getListByFusionId(fusionId); jsonObject.put(basePath+caseTag+fusionId, ResultData.ok(caseTagList)); for (CaseTag tag : caseTagList) { jsonObject.put(basePath+caseTagPoint+tag.getTagId(), ResultData.ok(caseTagPointService.allList(tag.getTagId()))); downResources(tag.getTagImgUrl(),caseOffPath); downResource(tag.getHotIconUrl(),caseOffPath); downResource(tag.getAudio(),caseOffPath); } jsonObject.put(basePath+casePathInfoUrl+fusionId, ResultData.ok(casePathService.getByFusionId(fusionId))); List fusionGuides = fusionGuideService.getByFusionId(fusionId); jsonObject.put(basePath+fusionGuide+fusionId, ResultData.ok(fusionGuides)); for (FusionGuide guide : fusionGuides) { downResource(guide.getCover(),caseOffPath); List listByGuideId = fusionGuidePathService.getListByGuideId(guide.getFusionGuideId()); for (FusionGuidePath guidePath : listByGuideId) { downResource(guidePath.getCover(),caseOffPath); } jsonObject.put(basePath+fusionGuidePath+guide.getFusionGuideId(), ResultData.ok(listByGuideId)); } List listByFusionId = caseAnimationService.getListByFusionId(fusionId); for (CaseAnimation caseAnimation : listByFusionId) { downModel(caseAnimation.getUrl(),caseOffPath); } jsonObject.put(basePath+caseAnimationUrl+fusionId, ResultData.ok(listByFusionId)); String jsonString = jsonObject.toJSONString(); if(!repMap.isEmpty()){ for (String key : repMap.keySet()) { jsonString = jsonString.replace(key,repMap.get(key)); } } FileUtil.writeString(jsonString, caseOffPath+"/www/package/"+jsonDataName,"UTF-8"); } //http://127.0.0.1:8080/offline.html?caseId=362&app=1&share=1#/show/summary static String batName = "start-browser.bat"; public void cpIndexHtml(Integer caseId,String caseOfflinePath){ log.info("down-offline-cpIndexHtml:{}",caseId); FileUtil.copyContent(new File(FilePath.OFFLINE_TEMPLATE_PATH_Fusion),new File(caseOfflinePath),true); String s = FileUtil.readString(caseOfflinePath + File.separator + batName, StandardCharsets.UTF_8); s = s.replaceAll("@caseId", String.valueOf(caseId)); FileUtil.writeString(s, caseOfflinePath + File.separator + batName,"UTF-8"); } @Autowired DownService downService; @Autowired UploadToOssUtil uploadToOssUtil; public void downSwkk(String path,String num,Integer type){ String swkkPath = path + "/swkk/"+num; String swkkZipPath = swkkPath +".zip"; String swssPath = path + "/swss/"+num; String swssZipPath = swssPath +".zip"; Integer isObj = 0; if(type == 4 || type == 6){ isObj =1; } try { DownVo downVo = downService.checkDownLoad(num, isObj,"offline"); log.info("down:{}",downVo); if(downVo.getDownloadStatus() == 3 && StringUtils.isNotBlank(downVo.getDownloadUrl())){ downZip(type,downVo.getDownloadUrl(),swkkZipPath,swkkPath,swssZipPath,swssPath); }else { DownVo down = downService.down(num, isObj,"offline"); if(down.getDownloadStatus() == 1){ DownloadProcessVo downloadProcessVo = downService.downloadProcess(num, isObj); while (downloadProcessVo.getStatus() != 1002 ){ downloadProcessVo = downService.downloadProcess(num, isObj); Thread.sleep(2000L); } downZip(type,downloadProcessVo.getUrl(),swkkZipPath,swkkPath,swssZipPath,swssPath); } } }catch (Exception e){ log.info("下载场景离线包失败:{}",num,e); } } @Value("${upload.query-path}") private String queryPath; public void downZip(Integer type ,String uri,String kkzipPath,String kknumPath,String sszipPath,String ssNumPath){ try { if(type != 2 && type != 5){ //深时点云 if(uri.contains("?")){ uri = uri.split("[?]")[0]; } ShellUtil.yunDownload(uri.replace(queryPath, ""), kkzipPath); ShellUtil.unZip(kkzipPath,kknumPath); FileUtil.del(kkzipPath); }else { ShellUtil.yunDownloadSs(uri.replace(queryPath, ""), sszipPath); ShellUtil.unZip(sszipPath,ssNumPath); FileUtil.del(sszipPath); } }catch (Exception e){ log.info("下载场景离线包失败:{}",uri,e); } } public void downModel( String modelGlbUrl,String path) { JSONArray jsonArray = JSONArray.parseArray(modelGlbUrl); for (Object object : jsonArray) { String res = (String) object; log.info("下载模型:{}",res); res = res.replace(queryPath, ""); if(res.contains(".json") ){ res = new File(res).getParentFile().getPath(); } ShellUtil.yunDownload(res, path +"/"+ res); } } public void downResources(String urls,String offPath) { JSONArray jsonArray = JSONArray.parseArray(urls); for (Object object : jsonArray) { String res = (String) object; downResource(res,offPath); } } public void downResource(String url,String path) { if(StringUtils.isBlank(url) ){ return; } url = url.replace(queryPath, ""); if( !uploadToOssUtil.existKey(url)){ log.info("downResource文件不存在:{},{}",url,path); return; } ShellUtil.yunDownload(url,path+"/"+url); } }