package com.fdkankan.scene.service.impl; import cn.hutool.core.bean.BeanUtil; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.net.multipart.UploadFile; import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.ZipUtil; import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.dtflys.forest.exceptions.ForestRuntimeException; import com.dtflys.forest.http.ForestRequest; import com.dtflys.forest.http.ForestResponse; import com.fdkankan.common.constant.*; import com.fdkankan.common.exception.BusinessException; import com.fdkankan.common.response.Result; import com.fdkankan.common.response.ResultData; import com.fdkankan.common.util.ComputerUtil; import com.fdkankan.common.util.CreateObjUtil; import com.fdkankan.common.util.FileMd5Util; import com.fdkankan.common.util.FileUtil; import com.fdkankan.common.util.FileUtils; import com.fdkankan.common.util.SceneUtil; import com.fdkankan.fyun.constant.StorageType; import com.fdkankan.fyun.oss.UploadToOssUtil; import com.fdkankan.platform.api.feign.PlatformGoodsClient; import com.fdkankan.platform.api.feign.PlatformUserClient; import com.fdkankan.platform.api.dto.CameraDetail; import com.fdkankan.platform.api.dto.Company; import com.fdkankan.redis.constant.RedisKey; import com.fdkankan.redis.constant.RedisLockKey; import com.fdkankan.redis.util.RedisLockUtil; import com.fdkankan.redis.util.RedisUtil; import com.fdkankan.scene.api.dto.SceneInfoDTO; import com.fdkankan.scene.bean.BoxPhotoBean; import com.fdkankan.scene.bean.SceneJsonBean; import com.fdkankan.scene.bean.TagBean; import com.fdkankan.scene.callback.FdkkMiniReqErrorCallback; import com.fdkankan.scene.callback.FdkkMiniReqSuccessCallback; import com.fdkankan.scene.entity.SceneDataDownload; import com.fdkankan.scene.entity.SceneEditControls; import com.fdkankan.scene.entity.SceneEditInfo; import com.fdkankan.scene.entity.SceneEditInfoExt; import com.fdkankan.scene.entity.ScenePlus; import com.fdkankan.scene.entity.ScenePlusExt; import com.fdkankan.scene.entity.ScenePro; import com.fdkankan.scene.factory.FloorLogoHandlerFactory; import com.fdkankan.scene.factory.MusicHandlerFactory; import com.fdkankan.scene.factory.ScreenshotHandlerFactory; import com.fdkankan.scene.httpclient.FdkankanMiniClient; import com.fdkankan.scene.mapper.ISceneEditInfoMapper; import com.fdkankan.scene.service.ISceneDataDownloadService; import com.fdkankan.scene.service.ISceneEditControlsService; import com.fdkankan.scene.service.ISceneEditInfoExtService; import com.fdkankan.scene.service.ISceneEditInfoService; import com.fdkankan.scene.service.IScenePlusExtService; import com.fdkankan.scene.service.IScenePlusService; import com.fdkankan.scene.service.ISceneProExtService; import com.fdkankan.scene.service.ISceneProService; import com.fdkankan.scene.service.ISceneUploadService; import com.fdkankan.scene.vo.*; import com.google.common.collect.Lists; import com.google.errorprone.annotations.Var; import java.io.File; import java.nio.charset.Charset; import java.util.ArrayList; import java.util.Arrays; import java.util.Comparator; import java.util.Date; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Map.Entry; import java.util.Set; import java.util.UUID; import java.util.stream.Collectors; import lombok.extern.slf4j.Slf4j; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Value; import org.springframework.cloud.context.config.annotation.RefreshScope; import org.springframework.stereotype.Service; import java.io.IOException; import java.util.Calendar; import java.util.Objects; import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; /** *

* 服务实现类 *

* * @author * @since 2022-01-18 */ @Slf4j @RefreshScope @Service public class SceneEditInfoServiceImpl extends ServiceImpl implements ISceneEditInfoService { @Value("${oss.prefix.ali}") private String prefixAli; @Value("${oss.prefix.url}") private String ossUrlPrefix; @Value("${upload.type}") private String type; @Value("${oss.bucket:4dkankan}") private String bucket; @Value("${main.url}") private String mainUrl; @Value("${ecs.checkFile.maxTimes:5}") private int maxCheckTimes; @Value("${ecs.checkFile.waitTime:5000}") private int waitTime; @Value("${http.host.4dkankanMini}") private String fkankanMiniHost; @Autowired private ISceneEditControlsService sceneEditControlsService; @Autowired private ISceneProService sceneProService; @Autowired private UploadToOssUtil uploadToOssUtil; @Autowired RedisUtil redisUtil; @Autowired private ISceneProExtService sceneProExtService; @Autowired private RedisLockUtil redisLockUtil; @Autowired ISceneDataDownloadService sceneDataDownloadService; @Autowired PlatformGoodsClient platformGoodsClient; @Autowired PlatformUserClient platformUserClient; @Autowired private ISceneEditInfoExtService sceneEditInfoExtService; @Autowired IScenePlusService scenePlusService; @Autowired IScenePlusExtService scenePlusExtService; @Autowired private FdkankanMiniClient fdkankanMiniClient; @Autowired private ISceneUploadService sceneUploadService; @Transactional @Override public SceneEditInfoVO saveScene(SceneEditInfoParamVO param) { ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum()); if(Objects.isNull(scenePlus)){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId()); SceneEditControls sceneEditControlsDb = null; if(Objects.nonNull(sceneEditInfoDb)){ sceneEditControlsDb = sceneEditControlsService.getBySceneEditId(sceneEditInfoDb.getId()); } SceneEditInfo sceneEditInfo = BeanUtil.copyProperties(param, SceneEditInfo.class); sceneEditInfo.setScenePlusId(scenePlus.getId()); if(Objects.isNull(sceneEditInfoDb)){ this.save(sceneEditInfo); }else{ sceneEditInfo.setId(sceneEditInfoDb.getId()); sceneEditInfo.setVersion(sceneEditInfoDb.getVersion() + 1); this.updateById(sceneEditInfo); } if(Objects.nonNull(param.getControls())){ SceneEditControls sceneEditControls = BeanUtil.copyProperties(param.getControls(), SceneEditControls.class); sceneEditControls.setEditInfoId(sceneEditInfo.getId()); if(Objects.isNull(sceneEditControlsDb)){ sceneEditControlsService.save(sceneEditControls); }else{ sceneEditControls.setId(sceneEditControlsDb.getId()); sceneEditControlsService.updateById(sceneEditControls); } } SceneEditInfoVO result = BeanUtil.copyProperties(param, SceneEditInfoVO.class); result.setCreateTime(sceneEditInfo.getCreateTime()); return result; } @Override public SceneEditInfo getBySceneProId(long sceneProId) { return this.getOne(new LambdaQueryWrapper() .eq(SceneEditInfo::getSceneProId, sceneProId)); } @Override public SceneEditInfo getByScenePlusId(long scenePlusId) { return this.getOne(new LambdaQueryWrapper() .eq(SceneEditInfo::getScenePlusId, scenePlusId)); } @Transactional @Override public ResultData publicScene(SceneEditInfoParamVO param) throws Exception{ String num = param.getNum(); //根据场景码查询数据库 ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId()); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); SceneEditInfoExt sceneEditInfoExt = null; SceneEditControls sceneEditControls = null; if(sceneEditInfo == null){ sceneEditInfo = new SceneEditInfo(); sceneEditInfo.setSceneProId(scenePlus.getId()); }else{ sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId()); sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId()); } if(sceneEditInfoExt == null){ sceneEditInfoExt = new SceneEditInfoExt(); sceneEditInfoExt.setFloorPlanAngle(0f); sceneEditInfoExt.setFloorPlanCompass(0f); } // if(sceneEditControls == null){ // sceneEditControls = new SceneEditControls(); // } //生成sceneJson SceneJsonBean sceneJson = new SceneJsonBean(); BeanUtil.copyProperties(sceneEditInfoExt, sceneJson); BeanUtil.copyProperties(sceneEditInfo, sceneJson); SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class); sceneJson.setControls(sceneEditControlsVO); sceneJson.setNum(num); // if(StrUtil.isNotEmpty(sceneEditInfo.getFloorPlanPath())){ // sceneJson.setFloorPlanPaths(sceneEditInfo.getFloorPlanPath().split(",")); // } sceneJson.setCreateTime(scenePlus.getCreateTime()); sceneJson.setSceneResolution(scenePlusExt.getSceneResolution()); sceneJson.setSceneFrom(scenePlusExt.getSceneFrom()); if(StrUtil.isNotEmpty(scenePlusExt.getVideos())){ sceneJson.setVideos(scenePlusExt.getVideos()); } //处理热点数据,生成hot.json this.publicHotData(num, sceneEditInfo); // TODO: 2022/3/2 这里的逻辑放在上传球幕视频接口中做了,这里先暂时注释掉,以后要删除 //处理球幕视频 // this.buildVideo(sceneEditInfo, sceneProExt.getDataSource(), sceneNum); //本地写sceneJson文件 String localSceneJsonPath = String.format(ConstantFilePath.SCENE_DATA_PATH_V4, num) + "scene.json"; FileUtils.writeFile(localSceneJsonPath, JSON.toJSONString(sceneJson)); //上传sceneJson文件 String sceneJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH+"scene.json", num); uploadToOssUtil.upload(JSON.toJSONBytes(sceneJson), sceneJsonPath); //sceneJson放入缓存 String key = String.format(RedisKey.SCENE_JSON, num); redisUtil.set(key, JSON.toJSONString(sceneJson)); //删除发布数据中的user目录 String publicUserPath = String.format(UploadFilePath.USER_VIEW_PATH, num); uploadToOssUtil.deleteFile(publicUserPath); // String editUserPath = String.format(UploadFilePath.USER_EDIT_PATH, sceneNum); //复制编辑目录到发布目录 String editPath = String.format(UploadFilePath.USER_EDIT_PATH, param.getNum()); String viewPath = String.format(UploadFilePath.USER_VIEW_PATH, param.getNum()); uploadToOssUtil.copyFiles(editPath,viewPath); //入库 if(sceneEditInfo.getId() == null){ this.save(sceneEditInfo); }else{ this.updateById(sceneEditInfo); } sceneEditInfoExt.setSceneProId(scenePro.getId()); sceneEditInfoExt.setScenePlusId(scenePlus.getId()); sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId()); sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt); // sceneEditControls.setEditInfoId(sceneEditInfo.getId()); // if(sceneEditControls.getId() == null){ // sceneEditControlsService.save(sceneEditControls); // }else{ // sceneEditControlsService.updateById(sceneEditControls); // } return ResultData.ok(); } private void buildVideo(SceneEditInfo sceneEditInfo, String path, String num) throws Exception{ if(CommonStatus.NO.equals(sceneEditInfo.getBuildVideoStatus())){ return; } if(path != null && !"".equals(path) && path.startsWith("http")){ path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2]; } String target = path + "_images"; log.info("球幕视频路径:{}", target + File.separator + "extras/video"); File video = new File(target + File.separator + "extras/video"); //如果文件夹或者文件不存在,跳出 if(!video.exists() || video.listFiles() == null || video.listFiles().length == 0){ return; } String userEditPath = String.format(UploadFilePath.USER_EDIT_PATH, num); for(String videoName : video.list()){ log.info("球幕视频名称:{}", videoName); uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName,userEditPath + videoName); CreateObjUtil.mp4ToFlv(target + File.separator + "extras/video/" + videoName, target + File.separator + "extras/video/" + videoName.replace("mp4", "flv")); uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"),userEditPath + videoName.replace("mp4", "flv")); //覆盖原始视频资源 FileUtils.copyFile(target + File.separator + "extras/video/" + videoName, path + File.separator + "caches/videos/" + videoName, true); } FileUtils.deleteDirectory(target + File.separator + "extras/video/"); //重置状态 sceneEditInfo.setBuildVideoStatus(CommonStatus.NO.code()); } private void publicHotData(String sceneNum, SceneEditInfo sceneEditInfo) throws IOException { String hotDataKey = String.format(RedisKey.SCENE_HOT_DATA, sceneNum); Map hotMap = redisUtil.hmget(hotDataKey); List hotList = Lists.newArrayList(); hotMap.entrySet().stream().forEach(entry->{ if(StrUtil.isNotEmpty(entry.getValue())){ hotList.add(entry.getValue()); } }); if(CollUtil.isNotEmpty(hotList)){ JSONArray jsonhots = new JSONArray(); hotList.stream().forEach(hot->{ jsonhots.add(JSONObject.parseObject(hot)); }); String hotJsonPath = String.format(UploadFilePath.USER_EDIT_PATH, sceneNum) + "hot.json"; uploadToOssUtil.upload(jsonhots.toString().getBytes(), hotJsonPath); //修改tags状态为是,标识有热点数据 this.saveTagsToSceneEditInfo(sceneNum, sceneEditInfo); } } @Override public SceneInfoVO getSceneInfo(@Var SceneInfoParamVO param) throws Exception{ SceneInfoReqType sceneInfoReqType = SceneInfoReqType.get(param.getReqType()); switch (sceneInfoReqType){ //如果是编辑页面请求,查数据库 case EDIT: return this.getSceneInfo4Edit(param.getNum()); //如果是查看页面请求,查redis case VIEW: return this.getSceneInfo4View(param.getNum()); } return null; } @Override public SceneInfoDTO getSceneInfo4Download(SceneInfoParamVO param) throws Exception{ SceneInfoVO sceneInfo4View = this.getSceneInfo4View(param.getNum()); if(Objects.isNull(sceneInfo4View)){ return null; } SceneInfoDTO sceneInfoDTO = BeanUtil.copyProperties(sceneInfo4View, SceneInfoDTO.class); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); sceneInfoDTO.setSceneSource(scenePlus.getSceneSource()); return sceneInfoDTO; } /** *

编辑页面获取场景详情信息 *

* @author dengsixing * @date 2022/1/21 * @param num * @return com.fdkankan.scene.vo.SceneInfoVO **/ private SceneInfoVO getSceneInfo4Edit(String num) throws Exception{ // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(Objects.isNull(scenePlus)){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId()); ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId()); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); if(Objects.isNull(sceneEditInfo)){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId()); SceneEditControls sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId()); SceneInfoVO sceneInfoVO = new SceneInfoVO(); BeanUtil.copyProperties(sceneEditInfoExt, sceneInfoVO); BeanUtil.copyProperties(sceneEditInfo, sceneInfoVO); if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){ sceneInfoVO.setFloorPlanAngle(0f); } if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){ sceneInfoVO.setFloorPlanCompass(0f); } sceneInfoVO.setControls(BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class)); sceneInfoVO.setNum(num); //生成sceneJson // if(StrUtil.isNotEmpty(sceneEditInfo.getFloorPlanPath())){ // sceneInfoVO.setFloorPlanPaths(sceneEditInfo.getFloorPlanPath().split(",")); // } sceneInfoVO.setCreateTime(scenePlus.getCreateTime()); sceneInfoVO.setSceneResolution(scenePlusExt.getSceneResolution()); sceneInfoVO.setSceneFrom(scenePlusExt.getSceneFrom()); sceneInfoVO.setVideos(scenePlusExt.getVideos()); // TODO: 2022/4/24 v3版本停机要切换---------------------------start // this.setExtData(sceneInfoVO, scenePlus.getCameraId()); this.setExtDataFromV3(sceneInfoVO, scenePlus.getCameraId()); // TODO: 2022/4/24 v3版本停机要切换---------------------------end return sceneInfoVO; } /** *

getInfo接口返回字段扩展 *

* @author dengsixing * @date 2022/3/9 **/ private void setExtDataFromV3(SceneInfoVO sceneInfoVO, Long cameraId) throws Exception{ if(Objects.isNull(cameraId)){ return; } String url = fkankanMiniHost + "/api/user/camera/getDataSyncByCameraId?cameraId=" + cameraId; Result dataSyncResult = fdkankanMiniClient .getDataSyncType(url, new FdkkMiniReqSuccessCallback(), new FdkkMiniReqErrorCallback()); if(!"[]".equals(dataSyncResult.getData())){ sceneInfoVO.setDataSync(dataSyncResult.getData()); } } /** *

getInfo接口返回字段扩展 *

* @author dengsixing * @date 2022/3/9 **/ private void setExtData(SceneInfoVO sceneInfoVO, Long cameraId) throws Exception{ ResultData cameraResultData = platformGoodsClient.getCameraDetailByCameraId(cameraId); if(!cameraResultData.getSuccess()){ throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message()); } CameraDetail camera = cameraResultData.getData(); if(Objects.isNull(camera)){ throw new BusinessException(ErrorCode.FAILURE_CODE_7010); } if(Objects.isNull(camera.getCompanyId())){ return; } ResultData companyResultData = platformUserClient.getCompanyById(camera.getCompanyId()); if(!companyResultData.getSuccess()){ throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message()); } Company company = companyResultData.getData(); if(Objects.isNull(company)){ return; } sceneInfoVO.setDataSync(company.getDataSync()); } /** *

查看页面获取场景详情信息 *

* @author dengsixing * @date 2022/1/21 * @param num * @return com.fdkankan.scene.vo.SceneInfoVO **/ private SceneInfoVO getSceneInfo4View(String num) throws Exception{ // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); String key = String.format(RedisKey.SCENE_JSON, num); String sceneJson = redisUtil.get(key); SceneInfoVO sceneInfoVO = null; //先查询redis if(StrUtil.isNotEmpty(sceneJson)) { sceneInfoVO = JSON.parseObject(sceneJson, SceneInfoVO.class); // TODO: 2022/4/24 v3版本停机要切换---------------------------start // this.setExtData(sceneInfoVO, scenePlus.getCameraId()); this.setExtDataFromV3(sceneInfoVO, scenePlus.getCameraId()); // TODO: 2022/4/24 v3版本停机要切换---------------------------end sceneInfoVO.setScenePassword(null); if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){ sceneInfoVO.setFloorPlanAngle(0f); } if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){ sceneInfoVO.setFloorPlanCompass(0f); } return sceneInfoVO; } //如果redis没找到,从scene.json中获取 String objectName = String.format(ConstantFilePath.SCENE_VIEW_DATA_DATA_SCENEJSON, num); String objectContent = uploadToOssUtil.getObjectContent(bucket, objectName); if(StrUtil.isEmpty(objectContent)) return null; sceneInfoVO = JSON.parseObject(objectContent, SceneInfoVO.class); // TODO: 2022/4/24 v3版本停机要切换---------------------------start // this.setExtData(sceneInfoVO, scenePlus.getCameraId()); this.setExtDataFromV3(sceneInfoVO, scenePlus.getCameraId()); // TODO: 2022/4/24 v3版本停机要切换---------------------------end sceneInfoVO.setScenePassword(null); if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){ sceneInfoVO.setFloorPlanAngle(0f); } if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){ sceneInfoVO.setFloorPlanCompass(0f); } redisUtil.set(key, JSON.toJSONString(sceneInfoVO)); return sceneInfoVO; } @Override // @Transactional public ResultData saveUpload(SaveUploadParamVO param) throws Exception { FileBizType fileBizType = FileBizType.get(param.getBizType()); String num = param.getNum(); String fileInfo = param.getFileInfo(); if(Objects.isNull(fileBizType)){ throw new BusinessException(ErrorCode.FAILURE_CODE_7003); } switch (fileBizType){ case MUSIC: return new MusicHandlerFactory().getHandler().save(num, fileInfo); case FLOOR_LOGO: return new FloorLogoHandlerFactory().getHandler().save(num, fileInfo); // case FLOOR_PLAN: // return new FloorPlanHandlerFactory().getHandler().save(num, fileInfo); case SCREENSHOT: return new ScreenshotHandlerFactory().getHandler().save(num, fileInfo); } return ResultData.ok(); } @Override public ResultData saveCad(BaseDataParamVO param) throws Exception { String num = param.getNum(); // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); String editUserPath = String.format(UploadFilePath.USER_EDIT_PATH, num); String localDataPath = String.format(ConstantFilePath.SCENE_USER_PATH_V4, num); JSONObject fileInfoJson = JSON.parseObject(param.getData()); Float angel = null; Float compass = null; String floorJsonData = fileInfoJson.getString("floorJsonData"); if(StrUtil.isNotBlank(floorJsonData)){ JSONObject jsonObject = JSON.parseObject(floorJsonData); angel = jsonObject.getFloat("angle"); compass = jsonObject.getFloat("compass"); } //处理户型图数据 //上传floorplan_user.json文件 FileUtils.writeFile(localDataPath + "floorplan_user.json", floorJsonData); uploadToOssUtil.upload(localDataPath + "floorplan_user.json", editUserPath + "floorplan_user.json"); //写入数据库 Byte floorPlanUser = null; if(StrUtil.isNotEmpty(floorJsonData)){ floorPlanUser = CommonStatus.YES.code(); } SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId()); if(Objects.nonNull(sceneEditInfoDb)){ LambdaUpdateWrapper updateWrapper = new LambdaUpdateWrapper() .setSql("version=version+" + 1) .eq(SceneEditInfo::getId, sceneEditInfoDb.getId()); if(floorPlanUser != null){ updateWrapper.set(SceneEditInfo::getFloorPlanUser, floorPlanUser); } this.update(updateWrapper); }else{ sceneEditInfoDb = new SceneEditInfo(); sceneEditInfoDb.setScenePlusId(scenePlus.getId()); sceneEditInfoDb.setFloorPlanUser(floorPlanUser); this.save(sceneEditInfoDb); } SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfoDb.getId()); if(Objects.isNull(sceneEditInfoExt)){ sceneEditInfoExt = new SceneEditInfoExt(); sceneEditInfoExt.setEditInfoId(sceneEditInfoDb.getId()); } if(Objects.nonNull(angel)){ sceneEditInfoExt.setFloorPlanAngle(angel); } if(Objects.nonNull(compass)){ sceneEditInfoExt.setFloorPlanCompass(compass); } sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt); return ResultData.ok(); } @Override public ResultData resetCad(String num){ ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId()); SceneEditInfoExt sceneEditInfoExt = null; if(Objects.nonNull(sceneEditInfoDb)){ LambdaUpdateWrapper updateWrapper = new LambdaUpdateWrapper() .setSql("version=version+" + 1) .set(SceneEditInfo::getFloorPlanUser, CommonStatus.NO.code()) .eq(SceneEditInfo::getId, sceneEditInfoDb.getId()); this.update(updateWrapper); sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfoDb.getId()); }else{ sceneEditInfoDb = new SceneEditInfo(); sceneEditInfoDb.setSceneProId(scenePro.getId()); sceneEditInfoDb.setScenePlusId(scenePlus.getId()); sceneEditInfoDb.setFloorPlanUser(CommonStatus.NO.code()); this.save(sceneEditInfoDb); } if(Objects.isNull(sceneEditInfoExt)){ sceneEditInfoExt = new SceneEditInfoExt(); sceneEditInfoExt.setSceneProId(scenePro.getId()); sceneEditInfoDb.setScenePlusId(scenePlus.getId()); sceneEditInfoExt.setEditInfoId(sceneEditInfoDb.getId()); } sceneEditInfoExt.setFloorPlanAngle(0f); sceneEditInfoExt.setFloorPlanCompass(0f); sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt); return ResultData.ok(); } @Override public ResultData renameCad(RenameCadParamVO param) throws IOException { String key = String.format(UploadFilePath.DATA_VIEW_PATH, param.getNum()) + "floorplan_cad.json"; String floorplanCadJson = uploadToOssUtil.getObjectContent(this.bucket, key); if(StrUtil.isEmpty(floorplanCadJson)){ return ResultData.ok(); } Map renameMap = new HashMap<>(); param.getFloors().stream().forEach(floor->{ renameMap.put(floor.getSubgroup(), floor.getName()); }); JSONObject jsonObject = JSON.parseObject(floorplanCadJson); if(Objects.isNull(jsonObject)){ return ResultData.ok(); } JSONArray floorArr = jsonObject.getJSONArray("floors"); if(CollUtil.isEmpty(floorArr)){ return ResultData.ok(); } for (Object o : floorArr) { JSONObject item = (JSONObject)o; int subgroup = item.getIntValue("subgroup"); String name = renameMap.get(subgroup); if(StrUtil.isEmpty(name)){ continue; } item.put("name", name); } uploadToOssUtil.upload(jsonObject.toJSONString().getBytes(), key); return ResultData.ok(); } private JSONObject updateFloorName(String sourceFloors, String targeFloors){ JSONArray sourceFloorsJson = JSON.parseArray(sourceFloors); JSONObject targeFloorsJson = JSONObject.parseObject(targeFloors); JSONArray array = targeFloorsJson.getJSONArray("floors"); for (int i = 0; i < array.size(); ++i) { JSONObject targetFloor = array.getJSONObject(i); int targetId = targetFloor.getIntValue("id"); for (int j = 0; j < sourceFloorsJson.size(); ++j) { JSONObject floor = sourceFloorsJson.getJSONObject(j); int id = floor.getIntValue("id"); String name = floor.getString("name"); if (targetId != id) continue; targetFloor.put("name", name); } } return targeFloorsJson; } @Override public void upgradeVersionById(Long id) { this.update(new LambdaUpdateWrapper() .setSql("version=version+" + 1) .eq(SceneEditInfo::getId, id)); } @Override public ResultData uploadPanorama(String num, MultipartFile file) throws Exception { //校验压缩包格式 // log.info("全景图压缩包文件名:" + file.getOriginalFilename()); // if(!file.getOriginalFilename().endsWith(".zip")){ // throw new BusinessException(ErrorCode.FAILURE_CODE_7015); // } if(!file.getOriginalFilename().endsWith(".zip") && !file.getOriginalFilename().endsWith(".jpg")){ throw new BusinessException(ErrorCode.FAILURE_CODE_7007, "jpg或者zip"); } // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(scenePlus == null){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId()); ScenePlusExt scenePlusExt = scenePlusExtService .getScenePlusExtByPlusId(scenePlus.getId()); //原始计算根目录 // String path = "F:\\test"; String path = scenePlusExt.getDataSource(); //全景图计算根目录 String target = path + "_images"; //解压缩文件存放目录 String targetImagesPath = target + "/extras/images/"; //压缩文件保存目录 String zipTargetFilePath = targetImagesPath + file.getOriginalFilename(); //压缩包保存到本地 // String cachePath = String.format(ConstantFilePath.SCENE_CACHE_IMAGES, num); //先删除本地文件 FileUtils.deleteDirectory(targetImagesPath); File targetFile = new File(zipTargetFilePath); if(!targetFile.getParentFile().exists()){ targetFile.getParentFile().mkdirs(); } file.transferTo(targetFile); //如果是压缩包上传,需要解压缩 if(file.getOriginalFilename().endsWith(".zip")){ //解压zip包 FileUtil.unZip(zipTargetFilePath, targetImagesPath); // FileUtils.decompress(zipTargetFilePath, targetImagesPath); //删除压缩包 FileUtil.delFile(zipTargetFilePath); } //判断文件夹目录结构,图片必须放在压缩包根目录下,不支持空文件夹或其他格式文件上传 File[] files = new File(targetImagesPath).listFiles(); Arrays.stream(files).forEach(item->{ if(item.isDirectory()){ throw new BusinessException(ErrorCode.FAILURE_CODE_7018); } }); //获取解压后的文件列表 List uploadFileList = FileUtil.getFileList(targetImagesPath); if(CollUtil.isEmpty(uploadFileList)){ throw new BusinessException(ErrorCode.FAILURE_CODE_5062); } //判断是否有可用的jpg文件 boolean existJpg = false; if(CollUtil.isNotEmpty(uploadFileList)){ existJpg = uploadFileList.stream().anyMatch(str -> { if(str.endsWith(".jpg")){ return true; } return false; }); } if(!existJpg){ throw new BusinessException(ErrorCode.FAILURE_CODE_5062); } //列出caches/images中的文件列表 String imgCachePath = String.format(UploadFilePath.IMG_CACHES_PATH, num); //比对图片列表,不存在的要返回名称集合 String visionPath = path + "/results/vision.txt"; List panoramaImageList = SceneUtil.getPanoramaImageList(visionPath); List notExistFileList = uploadFileList.stream().filter(filePath -> { filePath = filePath.substring(filePath.lastIndexOf(File.separator) + 1); if(!panoramaImageList.contains(filePath)){ return false; } return true; }).collect(Collectors.toList()); if(CollUtil.isNotEmpty(notExistFileList)){ //删除错误文件 notExistFileList.parallelStream().forEach(filePath->{ FileUtils.deleteFile(filePath); }); } //判断成功的图片,如果成功图片为0,就直接返回,不需要执行算法 uploadFileList = FileUtil.getFileList(targetImagesPath); if(CollUtil.isEmpty(uploadFileList)){ if(CollUtil.isNotEmpty(notExistFileList)){ notExistFileList = notExistFileList.stream().map(filePath -> { return filePath.substring(filePath.lastIndexOf(File.separator) + 1); }).collect(Collectors.toList()); } return ResultData.ok(new UploadPanoramaVO(0, notExistFileList)); } //上传 Map map = new HashMap<>(); // uploadFileList.stream().forEach(filePath->{ // map.put(filePath, filePath.replace(targetImagesPath, imgCachePath)); // }); String imgViewPath = String.format(UploadFilePath.IMG_VIEW_PATH, num); FileUtils.downLoadFromUrl(ossUrlPrefix + imgViewPath + "vision.modeldata" + "?m="+new Date().getTime(), "vision.modeldata", target + File.separator + "extras" + File.separator); CreateObjUtil.convertVisionmodeldataToTxt(target + File.separator + "extras" + File.separator + "vision.modeldata", target + File.separator + "extras" + File.separator + "vision.txt"); FileUtils.copyFile(path + File.separator + "data.json", target + File.separator+"data.json", true); FileUtils.copyFile(path + File.separator + "project.json", target + File.separator+"project.json", true); //data.json增加extras为执行重建算法 String data = FileUtils.readFile(target + File.separator+"data.json"); if(data != null){ JSONObject floorplanJson = new JSONObject(); floorplanJson.put("has_source_images", true); floorplanJson.put("has_vision_txt", true); JSONObject dataJson = JSONObject.parseObject(data); dataJson.put("extras", floorplanJson); //V5表示不需要生成high,low文件 dataJson.put("skybox_type", "SKYBOX_V6"); if(scenePlusExt.getSceneScheme() == 11){ dataJson.put("skybox_type", "SKYBOX_V7"); } dataJson.put("split_type", "SPLIT_V8"); //sceneScheme为3切成瓦片图 if(scenePlusExt.getSceneScheme() == 3){ dataJson.put("skybox_type", "SKYBOX_V4"); } FileUtils.writeFile(target + File.separator+"data.json", new String(dataJson.toString().getBytes(), "UTF-8")); } if(new File(target + File.separator + "capture").exists()){ new File(target + File.separator + "capture").delete(); } if(new File(target + File.separator + "results").exists()){ FileUtils.delAllFile(target + File.separator + "results"); } if(StorageType.AWS.code().equals(this.type)){ //亚马逊保持旧方式,超链接capture CreateObjUtil.createSoftConnection(path + File.separator + "capture", target + File.separator + "capture"); } CreateObjUtil.ossUtilCp(ConstantFilePath.OSS_PREFIX + path.replace(ConstantFilePath.BUILD_MODEL_PATH, "") + "/data.fdage", target + File.separator + "capture/"); CreateObjUtil.build3dModel(target , "1"); String uploadJsonPath= target + File.separator + "results" +File.separator+"upload.json"; Thread.sleep(2000); boolean exist = ComputerUtil.checkComputeCompleted(uploadJsonPath, maxCheckTimes, waitTime); if(!exist){ throw new BusinessException(ErrorCode.FAILURE_CODE_7013); } String uploadData = FileUtils.readFile(uploadJsonPath); JSONObject uploadJson = null; JSONArray array = null; if(uploadData!=null) { uploadJson = JSONObject.parseObject(uploadData); array = uploadJson.getJSONArray("upload"); } if(array == null){ throw new BusinessException(ErrorCode.FAILURE_CODE_7013); } JSONObject fileJson = null; String fileName = ""; for(int i = 0, len = array.size(); i < len; i++) { fileJson = array.getJSONObject(i); fileName = fileJson.getString("file"); //文件不存在抛出异常 if (!new File(target + File.separator + "results" + File.separator + fileName) .exists()) { throw new Exception( target + File.separator + "results" + File.separator + fileName + "文件不存在"); } //high文件夹 if (fileJson.getIntValue("clazz") == 3) { map.put(target + File.separator + "results" + File.separator + fileName, imgViewPath + "pan/high/" + fileName.replace("high/", "")); continue; } //low文件夹 if (fileJson.getIntValue("clazz") == 4) { map.put(target + File.separator + "results" + File.separator + fileName, imgViewPath + "pan/low/" + fileName.replace("low/", "")); continue; } //tiles文件夹,亚马逊没有裁剪图片api,不需要上传4k图 if (fileJson.getIntValue("clazz") == 5) { map.put(target + File.separator + "results" + File.separator + fileName, imgViewPath + fileName); continue; } //tiles文件夹,亚马逊瓦片图 if (fileJson.getIntValue("clazz") == 7) { map.put(target + File.separator + "results" + File.separator + fileName, imgViewPath + fileName); continue; } } if(map.size()>0) { uploadToOssUtil.uploadMulFiles(map); } //拷贝修改后的全景图到缓存目录 String cachedImagesPath = String.format(ConstantFilePath.SCENE_CACHE_IMAGES, num); uploadFileList.stream().forEach(srcPath->{ cn.hutool.core.io.FileUtil.copy(srcPath, srcPath.replace(targetImagesPath, cachedImagesPath), true); }); //更新版本号 SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); this.upgradeVersionById(sceneEditInfo.getId()); //如果部分成功,则需要返回成功数量和失败列表 if(CollUtil.isNotEmpty(notExistFileList)){ notExistFileList = notExistFileList.stream().map(filePath -> { return filePath.substring(filePath.lastIndexOf(File.separator) + 1); }).collect(Collectors.toList()); return ResultData.ok(new UploadPanoramaVO(uploadFileList.size(), notExistFileList)); } return ResultData.ok(); } @Override public ResultData downloadPanorama(FileParamVO param) throws Exception { String num = param.getNum(); String fileName = param.getFileName(); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(Objects.isNull(scenePlus)){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId()); String path = scenePlusExt.getDataSource(); String cachePath = String.format(ConstantFilePath.SCENE_CACHE, num); String imgCachePath = String.format(UploadFilePath.IMG_CACHES_PATH, num); String localImagesPath = String.format(ConstantFilePath.SCENE_CACHE_IMAGES, num); String cacheFormat = "downloads/scene/%s/caches/"; String cacheImageFormat = "downloads/scene/%s/caches/images/"; String url = null; String downloadName = null; //删除本地文件 FileUtils.deleteDirectory(localImagesPath); //如果入参文件名不为空,则是单个文件下载,不需要打包 if(StrUtil.isNotEmpty(fileName)){ //如果是单张图片,直接提供oss url String localFilePath = localImagesPath + fileName; String ossFilePath = imgCachePath + fileName; uploadToOssUtil.upload(localFilePath, ossFilePath); url = ossUrlPrefix + ossFilePath; // FileUtils.downLoadFromUrl(imageUrl, fileName, localImagesPath); downloadName = fileName; }else{ //从vision.txt中读取有效全景图名称列表 // String visionPath = path + "/results/vision.txt"; // List imageList = SceneUtil.getPanoramaImageList(visionPath); // // //先下载到本地 // List keyList = cn.hutool.core.io.FileUtil.listFileNames(imgCachePath); // if (!StorageType.LOCAL.code().equals(this.type)) {// TODO: 2022/2/15 这里有可能有问题,可能还需要考虑本地部署的情况 //// System.setProperty("java.util.concurrent.ForkJoinPool.common.parallelism", "8"); // keyList.stream().forEach(key->{ // String file = key.substring(key.lastIndexOf("/") + 1); // if(imageList.contains(file)){ // String imageUrl = ossUrlPrefix + imgCachePath + file + "?t=" + System.currentTimeMillis(); // FileUtils.downLoadFromUrl(imageUrl, file, localImagesPath); // } // }); // } downloadName = num + "_images.zip"; //打包 String zipPath = cachePath + downloadName; ZipUtil.zip(localImagesPath, zipPath); //上传压缩包 uploadToOssUtil.upload(zipPath, String.format(cacheFormat, num) + downloadName); url = ossUrlPrefix + String.format(cacheFormat, num) + downloadName; //删除本地压缩包 FileUtils.deleteFile(zipPath); // //删除本地目录 // FileUtils.deleteDirectory(localImagesPath); } Map map = new HashMap<>(); map.put("fileUrl", url + "?t=" + System.currentTimeMillis()); map.put("fileName", downloadName); return ResultData.ok(map); } @Override public void saveTagsToSceneEditInfo(String num, SceneEditInfo sceneEditInfo){ //查询缓存是否包含热点数据 String key = String.format(RedisKey.SCENE_HOT_DATA, num); Map allTagsMap = redisUtil.hmget(key); boolean hashTags = false; for (Entry tagMap : allTagsMap.entrySet()) { if(StrUtil.isEmpty(tagMap.getValue())){ continue; } hashTags = true; break; } //更改热点状态 sceneEditInfo.setTags(hashTags ? CommonStatus.YES.code() : CommonStatus.NO.code()); //version 是空的代表 sceneEditInfo记录集不存在 if(sceneEditInfo.getVersion() != null){ sceneEditInfo.setVersion(sceneEditInfo.getVersion() + 1); } } @Override public ResultData saveVideoBox(FileNameAndDataParamVO param) throws Exception { JSONObject boxVideo = JSONObject.parseObject(param.getData()); String sid = boxVideo.getString("sid"); if(StrUtil.isEmpty(sid)){ throw new BusinessException(ErrorCode.PARAM_REQUIRED); } // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum()); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); //转换视频格式 this.transferToFlv(param.getNum(), param.getFileName()); //生成boxVideos数据 String boxVideos = this.createBoxVideos(param.getNum(), sid, boxVideo, sceneEditInfo, OperationType.ADDORUPDATE.code()); //更新数据库 this.updateBoxVideos(sceneEditInfo, scenePlus.getId(), boxVideos); return ResultData.ok(); } @Override public ResultData deleteVideoBox(DeleteSidParamVO param) throws Exception { // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum()); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); //根据sid移除json String boxVideos = this.createBoxVideos(param.getNum(), param.getSid(), null, sceneEditInfo, OperationType.DELETE.code()); //写数据库 this.updateBoxVideos(sceneEditInfo,scenePlus.getId(),boxVideos); return ResultData.ok(); } @Override public ResultData saveBoxPhoto(BaseDataParamVO param) throws Exception { JSONObject boxPhoto = JSONObject.parseObject(param.getData()); String sid = boxPhoto.getString("sid"); if(StrUtil.isEmpty(sid)){ throw new BusinessException(ServerCode.PARAM_REQUIRED, sid); } ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); //生成boxVideos数据 String boxPhotos = this.createBoxPhotos(param.getNum(), sid, boxPhoto, sceneEditInfo, OperationType.ADDORUPDATE.code()); //更新数据库 this.updateBoxPhotos(sceneEditInfo, boxPhotos); return ResultData.ok(); } @Override public ResultData deleteBoxPhoto(DeleteSidParamVO param) throws Exception { ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum()); if(Objects.isNull(scenePlus)) throw new BusinessException(ErrorCode.FAILURE_CODE_5005); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); //根据sid移除json String boxPhotos = this.createBoxPhotos(param.getNum(), param.getSid(), null, sceneEditInfo, OperationType.DELETE.code()); //写数据库 this.updateBoxPhotos(sceneEditInfo, boxPhotos); return ResultData.ok(); } @Override public DownloadVO downloadBallScreenVideo(BallScreenVideoParamVO param) { String videoPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, param.getNum()) + param.getFileName(); String url = ossUrlPrefix + videoPath + "?t=" + System.currentTimeMillis(); return DownloadVO.builder() .fileName(param.getFileName()) .url(url) .build(); } @Override public ResultData uploadBallScreenVideo(String num, String fileName, MultipartFile file) throws Exception { //校验格式 if(!fileName.endsWith(".mp4")){ throw new BusinessException(ErrorCode.FAILURE_CODE_7016); } String videosViewPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, num); // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(scenePlus == null){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId()); ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId()); String path = scenePlusExt.getDataSource(); //校验文件名 String videos = scenePlusExt.getVideos(); if(StrUtil.isEmpty(videos)){ throw new BusinessException(ErrorCode.FAILURE_CODE_7012); } JSONObject videosJson = JSON.parseObject(videos); JSONArray dataArr = videosJson.getJSONArray("data"); boolean exists = false; for (Object o : dataArr) { JSONObject jsonObject = (JSONObject)o; if(jsonObject.getString("id").equals(fileName.replace(".mp4", ""))){ exists = true; break; } } if(!exists){ throw new BusinessException(ErrorCode.FAILURE_CODE_7012); } if(path != null && !"".equals(path) && path.startsWith("http")){ path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2]; } String target = path + "_images"; String filePath = target + File.separator + "extras/video" + File.separator + fileName; File targetFile = new File(filePath); if(!targetFile.getParentFile().exists()){ targetFile.getParentFile().mkdirs(); } //保存视频到本地 file.transferTo(targetFile); File video = new File(target + File.separator + "extras/video"); //如果文件夹或者文件不存在,跳出 if(!video.exists() || video.listFiles() == null || video.listFiles().length == 0){ throw new BusinessException(ErrorCode.FAILURE_CODE_7009); } for(String videoName : video.list()){ log.info("球幕视频名称:{}", videoName); uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName,videosViewPath + videoName); CreateObjUtil.mp4ToFlv(target + File.separator + "extras/video/" + videoName, target + File.separator + "extras/video/" + videoName.replace("mp4", "flv")); uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"),videosViewPath + videoName.replace("mp4", "flv")); //覆盖原始视频资源 FileUtils.copyFile(target + File.separator + "extras/video/" + videoName, path + File.separator + "caches/videos/" + videoName, true); } FileUtils.deleteDirectory(target + File.separator + "extras/video/"); SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId()); this.upgradeVersionById(sceneEditInfo.getId()); //更新scene.json版本号 this.upgradeVersionToSceneJson(num); return ResultData.ok(); } @Override public void upgradeVersionToSceneJson(String num) throws Exception{ String key = String.format(RedisKey.SCENE_JSON, num); String objectName = String.format(ConstantFilePath.SCENE_VIEW_DATA_DATA_SCENEJSON, num); String objectContent = uploadToOssUtil.getObjectContent(bucket, objectName); SceneJsonBean sceneJsonBean = JSON.parseObject(objectContent, SceneJsonBean.class); sceneJsonBean.setVersion(sceneJsonBean.getVersion() +1); uploadToOssUtil.upload(JSON.toJSONString(sceneJsonBean).getBytes(), objectName); redisUtil.set(key, JSON.toJSONString(sceneJsonBean)); } @Override public ResultData sceneSync(String num, String type, String floorPlanJson, String ajkJson, String cameraJson, MultipartFile[] files) throws Exception { String lockKey = String.format(RedisLockKey.LOCK_SCENE_SYNC, num); Boolean lock = redisLockUtil.lock(lockKey, RedisKey.EXPIRE_TIME_2_HOUR); if(!lock){ throw new BusinessException(ErrorCode.SYSTEM_BUSY); } try { // ScenePro scenePro = sceneProService.findBySceneNum(num); ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num); if(scenePlus == null){ throw new BusinessException(ErrorCode.FAILURE_CODE_5005); } // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId()); ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId()); //更新scene.json文件 String strsceneInfos = FileUtils.readFile(ConstantFilePath.SCENE_PATH + "data" + File.separator + "data" + num + File.separator + "scene.json"); if(strsceneInfos == null) new File(ConstantFilePath.SCENE_PATH + "data" + File.separator + "data" + num + File.separator + "scene.json").createNewFile(); String path = scenePlusExt.getDataSource();// /mnt/data/0662c5389/831989883441512448/4898cab04f8c_202104141602356060/ if(path != null && !"".equals(path) && path.startsWith("http")){ path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2]; } String target = path + "_ajk"; // /mnt/data/0662c5389/831989883441512448/4898cab04f8c_202104141602356060_ajk File editPath = new File(target); if(!editPath.exists()){ editPath.mkdirs(); } //创建文件夹软连接并且复制data.json和project.json if(new File(target + File.separator + "capture").exists()){ new File(target + File.separator + "capture").delete(); } if(new File(target + File.separator + "caches").exists()){ //删除link new File(target + File.separator + "caches" + File.separator + "images").delete(); //删除所有文件 FileUtils.delAllFile(target + File.separator + "caches"); } if(new File(target + File.separator + "results").exists()){ FileUtils.delAllFile(target + File.separator + "results"); } //创建文件夹,并link文件夹 new File(target + File.separator + "caches").mkdirs(); CreateObjUtil.createSoftConnection(path + File.separator + "capture", target + File.separator + "capture"); if(new File(path + File.separator + "caches" + File.separator + "images").exists()){ CreateObjUtil.createSoftConnection(path + File.separator + "caches" + File.separator + "images", target + File.separator + "caches" + File.separator + "images"); } FileUtils.copyFile(path + File.separator + "data.json", target + File.separator+"data.json", true); FileUtils.copyFile(path + File.separator + "project.json", target + File.separator+"project.json", true); //data.json增加extras为执行重建算法 String project = FileUtils.readFile(target + File.separator+"project.json"); if(project != null){ JSONObject projectJson = JSONObject.parseObject(project); projectJson.put("parent", projectJson.get("uuid")); projectJson.put("uuid", UUID.randomUUID().toString()); projectJson.put("time", System.currentTimeMillis()); FileUtils.writeFile(path + File.separator + "project.json", projectJson.toString()); } String data = FileUtils.readFile(target + File.separator+"data.json"); if(data != null){ JSONObject floorplanJson = new JSONObject(); floorplanJson.put("has_floor_ajk_json", true); floorplanJson.put("has_vision_txt", true); floorplanJson.put("has_floorplan_json", true); JSONObject dataJson = JSONObject.parseObject(data); dataJson.put("extras", floorplanJson); //V5表示不需要生成high,low文件 dataJson.put("skybox_type", "SKYBOX_V8"); dataJson.put("split_type", "SPLIT_V10"); FileUtils.writeFile(target + File.separator+"data.json", new String(dataJson.toString().getBytes(), "UTF-8")); } //文件上传的位置可以自定义 log.info("画墙重建模型开始"); File targetFile = new File(target + File.separator + "extras" + File.separator + "floor_ajk.json"); if(!targetFile.getParentFile().exists()){ targetFile.getParentFile().mkdirs(); } if(targetFile.exists()){ FileUtils.deleteFile(target + File.separator + "extras" + File.separator + "floor_ajk.json"); } // 保存 FileUtils.writeFile(target + File.separator + "extras" + File.separator + "floorplan.json", new String(floorPlanJson.getBytes(), "UTF-8")); FileUtils.writeFile(target + File.separator + "extras" + File.separator + "floor_ajk.json", new String(ajkJson.getBytes(), "UTF-8")); FileUtils.writeFile(ConstantFilePath.SCENE_PATH+"data"+File.separator+"data"+num + File.separator + "floor_ajk.json", new String(ajkJson.getBytes(), "UTF-8")); FileUtils.writeFile(target + File.separator + "extras" + File.separator + "vision.txt", new String(cameraJson.getBytes(), "UTF-8")); FileUtils.writeFile(ConstantFilePath.SCENE_PATH+"data"+File.separator+"data"+num + File.separator + "camera.json", new String(cameraJson.getBytes(), "UTF-8")); for(int i = 0; i < files.length; i ++){ File cadImg = new File(target + File.separator + "extras" + File.separator + "Floorplans/" + files[i].getOriginalFilename()); if(!cadImg.getParentFile().exists()){ cadImg.getParentFile().mkdirs(); } if(cadImg.exists()) { cadImg.delete(); } files[i].transferTo(cadImg); } //下载封面图 FileUtils.downLoadFromUrl(scenePlusExt.getThumb() + "?t=" + System.currentTimeMillis(), "Cover.png", target + File.separator + "extras" + File.separator + "CoverImage"); //转换成jpg FileUtils.pngToJpg(target + File.separator + "extras" + File.separator + "CoverImage/Cover.png", target + File.separator + "extras" + File.separator + "CoverImage/Cover.jpg"); //安居客算法运行 log.info("安居客算法:开始建模——"+num); CreateObjUtil.build3dModel(target , "1"); String uploadJsonPath = target + File.separator + "results" + File.separator + "upload.json"; log.info("uploadJsonPath=" + uploadJsonPath); //因为共享目录有延迟,这里循环检测算法是否计算完毕3次,每次隔五秒 boolean exist = ComputerUtil.checkComputeCompleted(uploadJsonPath, maxCheckTimes, waitTime); if(!exist){ throw new BusinessException(ErrorCode.FAILURE_CODE_7017); } String zipPath = target + File.separator + "results/" + num + ".zip"; new File(zipPath).delete(); ZipUtil.zip(target + File.separator + "results/ajk/", zipPath); // String command = "bash /opt/ossutil/gzip.sh " + zipPath.replace(".zip", "") + " " + target + File.separator + "results/ajk/"; // log.info("压缩文件:" + command); // CreateObjUtil.callshell(command); if(!new File(zipPath).exists()){ return ResultData.error(ErrorCode.FAILURE_CODE_5043); } String fileMD5 = FileMd5Util.getFileMD5(new File(zipPath)); uploadToOssUtil.upload(zipPath, "data_download/" + num + ".zip"); SceneDataDownload sceneDataDownload = sceneDataDownloadService.findBySceneNum(num); if(sceneDataDownload == null){ sceneDataDownload = new SceneDataDownload(); sceneDataDownload.setNum(num); sceneDataDownload.setDownloadPath(ossUrlPrefix + "data_download/" + num + ".zip"); sceneDataDownload.setFileMd5(fileMD5); sceneDataDownloadService.save(sceneDataDownload); return ResultData.ok(); } sceneDataDownload.setFileMd5(fileMD5); sceneDataDownload.setUpdateTime(Calendar.getInstance().getTime()); sceneDataDownloadService.updateById(sceneDataDownload); }catch (Exception e){ log.error("画墙重建模型失败...", e); throw new BusinessException(ErrorCode.FAILURE_CODE_5039); }finally { redisLockUtil.unlockLua(lockKey); } return ResultData.ok(); } @Override public ResultData checkKey(SceneCheckKeyParamVO param) throws Exception { SceneJsonBean sceneJsonBean = null; String sceneJson = redisUtil.get(String.format(RedisKey.SCENE_JSON, param.getNum())); if(StrUtil.isEmpty(sceneJson)){ sceneJson = uploadToOssUtil.getObjectContent(bucket, String.format(UploadFilePath.DATA_VIEW_PATH + "scene.json")); } sceneJsonBean = JSON.parseObject(sceneJson, SceneJsonBean.class); if(!param.getPassword().equals(sceneJsonBean.getScenePassword())){ throw new BusinessException(ErrorCode.FAILURE_CODE_5021); } return ResultData.ok(); } private void updateBoxVideos(SceneEditInfo sceneEditInfo, Long scenePlusId, String boxVideos){ if(Objects.isNull(sceneEditInfo)){ sceneEditInfo = new SceneEditInfo(); sceneEditInfo.setScenePlusId(scenePlusId); sceneEditInfo.setBoxVideos(boxVideos); this.save(sceneEditInfo); }else{ this.update(new UpdateWrapper() .setSql("version = version + 1") .set("box_videos", boxVideos) .eq("id", sceneEditInfo.getId())); } } private void updateBoxPhotos(SceneEditInfo sceneEditInfo, String boxPhotos){ this.update(new LambdaUpdateWrapper() .set(SceneEditInfo::getBoxPhotos, boxPhotos) .setSql("version = version + 1") .eq(SceneEditInfo::getId, sceneEditInfo.getId())); } private String createBoxVideos( String num, String sid, JSONObject boxVideo, SceneEditInfo sceneEditInfo, int type) throws Exception{ String boxVideos = null; if(sceneEditInfo != null){ boxVideos = sceneEditInfo.getBoxVideos(); } JSONArray boxVideosJson = null; if (StrUtil.isNotEmpty(boxVideos)) { boxVideosJson = JSONArray.parseArray(boxVideos); }else { boxVideosJson = new JSONArray(); } String result = null; //删除 if(type == OperationType.DELETE.code()){ Set deleteFile = new HashSet<>(); if(boxVideosJson.size() == 0) return null; for(int i=0;i(deleteFile)).build()); }else{ // TODO: 2022/3/1 目前需求只保留一个,所以这里先注释掉,sid不同时,不追加 //更新 // boolean exist = false; // for(int i=0;i deleteFile = new HashSet<>(); if(boxPhotosJson.size() == 0) return null; for(int i=0;i(deleteFile)).build()); }else{ //更新 boolean exist = false; for(int i=0;i list = Lists.newArrayList(); for (Object o : boxPhotosJson) { JSONObject jsonObject = (JSONObject)o; list.add(BoxPhotoBean.builder().createTime(jsonObject.getLong("createTime")).boxPhoto(jsonObject).build()); } //按创建时间倒叙排序 list.sort(Comparator.comparingLong(BoxPhotoBean::getCreateTime).reversed()); // list转JSONArray JSONArray array = new JSONArray(); list.stream().forEach(bean->{ array.add(bean.getBoxPhoto()); }); result = array.toJSONString(); } return result; } private void transferToFlv(String num, String fileName) throws Exception { String userEditPath = String.format(UploadFilePath.USER_EDIT_PATH, num); String localImagesPath = String.format(ConstantFilePath.SCENE_USER_PATH_V4, num); String localFilePath = localImagesPath + fileName; File targetFile = new File(localImagesPath); if (!targetFile.exists()){ targetFile.mkdirs(); } targetFile = new File(localFilePath); if (targetFile.exists()){ FileUtils.deleteFile(localFilePath); } //从用户编辑目录中下载视频到本地 if (!StorageType.LOCAL.code().equals(this.type)) { String filePath = userEditPath + fileName; String imageUrl = ossUrlPrefix + filePath + "?t=" + System.currentTimeMillis(); FileUtils.downLoadFromUrl(imageUrl, fileName, localImagesPath); } //视频格式转换 CreateObjUtil.mp4ToFlv(localFilePath, localFilePath.replace("mp4", "flv")); //上传 String flvFileName = fileName.replace("mp4", "flv"); // uploadToOssUtil.upload(localFilePath, userEditPath + fileName); uploadToOssUtil.upload(localFilePath.replace("mp4", "flv"), userEditPath+flvFileName); } }