SceneEditInfoServiceImpl.java 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394
  1. package com.fdkankan.scene.service.impl;
  2. import cn.hutool.core.bean.BeanUtil;
  3. import cn.hutool.core.collection.CollUtil;
  4. import cn.hutool.core.util.StrUtil;
  5. import cn.hutool.json.JSONUtil;
  6. import com.alibaba.fastjson.JSON;
  7. import com.alibaba.fastjson.JSONArray;
  8. import com.alibaba.fastjson.JSONObject;
  9. import com.amazonaws.services.devicefarm.model.UploadType;
  10. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  11. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  12. import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
  13. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  14. import com.fdkankan.common.constant.*;
  15. import com.fdkankan.common.exception.BusinessException;
  16. import com.fdkankan.common.response.ResultData;
  17. import com.fdkankan.common.util.ComputerUtil;
  18. import com.fdkankan.common.util.CreateObjUtil;
  19. import com.fdkankan.common.util.FileMd5Util;
  20. import com.fdkankan.common.util.FileUtil;
  21. import com.fdkankan.common.util.FileUtils;
  22. import com.fdkankan.fyun.constant.StorageType;
  23. import com.fdkankan.fyun.oss.UploadToOssUtil;
  24. import com.fdkankan.platform.api.feign.PlatformGoodsClient;
  25. import com.fdkankan.platform.api.feign.PlatformUserClient;
  26. import com.fdkankan.platform.api.vo.Camera;
  27. import com.fdkankan.platform.api.vo.CameraDetail;
  28. import com.fdkankan.platform.api.vo.Company;
  29. import com.fdkankan.redis.constant.RedisKey;
  30. import com.fdkankan.redis.constant.RedisLockKey;
  31. import com.fdkankan.redis.util.RedisLockUtil;
  32. import com.fdkankan.redis.util.RedisUtil;
  33. import com.fdkankan.scene.bean.SceneJsonBean;
  34. import com.fdkankan.scene.entity.SceneDataDownload;
  35. import com.fdkankan.scene.entity.SceneEditControls;
  36. import com.fdkankan.scene.entity.SceneEditInfo;
  37. import com.fdkankan.scene.entity.SceneEditInfoExt;
  38. import com.fdkankan.scene.entity.ScenePlus;
  39. import com.fdkankan.scene.entity.ScenePlusExt;
  40. import com.fdkankan.scene.entity.ScenePro;
  41. import com.fdkankan.scene.entity.SceneProExt;
  42. import com.fdkankan.scene.factory.FloorLogoHandlerFactory;
  43. import com.fdkankan.scene.factory.MusicHandlerFactory;
  44. import com.fdkankan.scene.factory.ScreenshotHandlerFactory;
  45. import com.fdkankan.scene.mapper.ISceneEditInfoMapper;
  46. import com.fdkankan.scene.service.ISceneDataDownloadService;
  47. import com.fdkankan.scene.service.ISceneEditControlsService;
  48. import com.fdkankan.scene.service.ISceneEditInfoExtService;
  49. import com.fdkankan.scene.service.ISceneEditInfoService;
  50. import com.fdkankan.scene.service.IScenePlusExtService;
  51. import com.fdkankan.scene.service.IScenePlusService;
  52. import com.fdkankan.scene.service.ISceneProExtService;
  53. import com.fdkankan.scene.service.ISceneProService;
  54. import com.fdkankan.scene.vo.*;
  55. import com.google.common.collect.Lists;
  56. import com.google.errorprone.annotations.Var;
  57. import java.io.File;
  58. import java.util.Date;
  59. import java.util.HashMap;
  60. import java.util.List;
  61. import java.util.Map;
  62. import java.util.Map.Entry;
  63. import java.util.UUID;
  64. import java.util.stream.Collectors;
  65. import lombok.extern.slf4j.Slf4j;
  66. import org.springframework.beans.factory.annotation.Autowired;
  67. import org.springframework.beans.factory.annotation.Value;
  68. import org.springframework.stereotype.Service;
  69. import java.io.IOException;
  70. import java.util.Calendar;
  71. import java.util.Objects;
  72. import org.springframework.transaction.annotation.Transactional;
  73. import org.springframework.web.multipart.MultipartFile;
  74. /**
  75. * <p>
  76. * 服务实现类
  77. * </p>
  78. *
  79. * @author
  80. * @since 2022-01-18
  81. */
  82. @Slf4j
  83. @Service
  84. public class SceneEditInfoServiceImpl extends ServiceImpl<ISceneEditInfoMapper, SceneEditInfo> implements ISceneEditInfoService {
  85. @Value("${oss.prefix.ali}")
  86. private String prefixAli;
  87. @Value("${oss.prefix.url}")
  88. private String ossUrlPrefix;
  89. @Value("${upload.type}")
  90. private String type;
  91. @Value("${oss.bucket:4dkankan}")
  92. private String bucket;
  93. @Value("${main.url}")
  94. private String mainUrl;
  95. @Autowired
  96. private ISceneEditControlsService sceneEditControlsService;
  97. @Autowired
  98. private ISceneProService sceneProService;
  99. @Autowired
  100. private UploadToOssUtil uploadToOssUtil;
  101. @Autowired
  102. RedisUtil redisUtil;
  103. @Autowired
  104. private ISceneProExtService sceneProExtService;
  105. @Autowired
  106. private RedisLockUtil redisLockUtil;
  107. @Autowired
  108. ISceneDataDownloadService sceneDataDownloadService;
  109. @Autowired
  110. PlatformGoodsClient platformGoodsClient;
  111. @Autowired
  112. PlatformUserClient platformUserClient;
  113. @Autowired
  114. private ISceneEditInfoExtService sceneEditInfoExtService;
  115. @Autowired
  116. IScenePlusService scenePlusService;
  117. @Autowired
  118. IScenePlusExtService scenePlusExtService;
  119. @Transactional
  120. @Override
  121. public SceneEditInfoVO saveScene(SceneEditInfoParamVO param) {
  122. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum());
  123. // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum());
  124. if(Objects.isNull(scenePlus)){
  125. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  126. }
  127. SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId());
  128. SceneEditControls sceneEditControlsDb = null;
  129. if(Objects.nonNull(sceneEditInfoDb)){
  130. sceneEditControlsDb = sceneEditControlsService.getBySceneEditId(sceneEditInfoDb.getId());
  131. }
  132. SceneEditInfo sceneEditInfo = BeanUtil.copyProperties(param, SceneEditInfo.class);
  133. sceneEditInfo.setScenePlusId(scenePlus.getId());
  134. if(Objects.isNull(sceneEditInfoDb)){
  135. this.save(sceneEditInfo);
  136. }else{
  137. sceneEditInfo.setId(sceneEditInfoDb.getId());
  138. this.updateById(sceneEditInfo);
  139. }
  140. if(Objects.nonNull(param.getControls())){
  141. SceneEditControls sceneEditControls = BeanUtil.copyProperties(param.getControls(), SceneEditControls.class);
  142. sceneEditControls.setEditInfoId(sceneEditInfo.getId());
  143. if(Objects.isNull(sceneEditControlsDb)){
  144. sceneEditControlsService.save(sceneEditControls);
  145. }else{
  146. sceneEditControls.setId(sceneEditControlsDb.getId());
  147. sceneEditControlsService.updateById(sceneEditControls);
  148. }
  149. }
  150. SceneEditInfoVO result = BeanUtil.copyProperties(param, SceneEditInfoVO.class);
  151. result.setCreateTime(sceneEditInfo.getCreateTime());
  152. return result;
  153. }
  154. @Override
  155. public SceneEditInfo getBySceneProId(long sceneProId) {
  156. return this.getOne(new LambdaQueryWrapper<SceneEditInfo>()
  157. .eq(SceneEditInfo::getSceneProId, sceneProId));
  158. }
  159. @Override
  160. public SceneEditInfo getByScenePlusId(long scenePlusId) {
  161. return this.getOne(new LambdaQueryWrapper<SceneEditInfo>()
  162. .eq(SceneEditInfo::getScenePlusId, scenePlusId));
  163. }
  164. @Transactional
  165. @Override
  166. public ResultData publicScene(SceneEditInfoParamVO param) throws Exception{
  167. String num = param.getNum();
  168. //根据场景码查询数据库
  169. ScenePro scenePro = sceneProService.findBySceneNum(sceneNum);
  170. SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  171. SceneEditInfo sceneEditInfo = this.getBySceneProId(scenePro.getId());
  172. SceneEditInfoExt sceneEditInfoExt = null;
  173. SceneEditControls sceneEditControls = null;
  174. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  175. // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  176. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  177. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  178. SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId());
  179. SceneEditControls sceneEditControls = null;
  180. if(sceneEditInfo == null){
  181. sceneEditInfo = new SceneEditInfo();
  182. sceneEditInfo.setSceneProId(scenePlus.getId());
  183. }else{
  184. sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());
  185. sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());
  186. }
  187. if(sceneEditInfoExt == null){
  188. sceneEditInfoExt = new SceneEditInfoExt();
  189. sceneEditInfoExt.setFloorPlanAngle(0f);
  190. sceneEditInfoExt.setFloorPlanCompass(0f);
  191. }
  192. // if(sceneEditControls == null){
  193. // sceneEditControls = new SceneEditControls();
  194. // }
  195. //生成sceneJson
  196. SceneJsonBean sceneJson = new SceneJsonBean();
  197. BeanUtil.copyProperties(sceneEditInfoExt, sceneJson);
  198. BeanUtil.copyProperties(sceneEditInfo, sceneJson);
  199. SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class);
  200. sceneJson.setControls(sceneEditControlsVO);
  201. sceneJson.setNum(num);
  202. // if(StrUtil.isNotEmpty(sceneEditInfo.getFloorPlanPath())){
  203. // sceneJson.setFloorPlanPaths(sceneEditInfo.getFloorPlanPath().split(","));
  204. // }
  205. sceneJson.setCreateTime(scenePlus.getCreateTime());
  206. sceneJson.setSceneResolution(scenePlusExt.getSceneResolution());
  207. sceneJson.setSceneFrom(scenePlusExt.getSceneFrom());
  208. if(StrUtil.isNotEmpty(scenePlusExt.getVideos())){
  209. sceneJson.setVideos(JSON.parseObject(scenePlusExt.getVideos()));
  210. }
  211. //处理热点数据,生成hot.json
  212. this.publicHotData(num, sceneEditInfo);
  213. // TODO: 2022/3/2 这里的逻辑放在上传球幕视频接口中做了,这里先暂时注释掉,以后要删除
  214. //处理球幕视频
  215. // this.buildVideo(sceneEditInfo, sceneProExt.getDataSource(), sceneNum);
  216. //本地写sceneJson文件
  217. String localSceneJsonPath = String.format(ConstantFilePath.SCENE_PATH_FORMAT, num);
  218. FileUtils.writeFile(localSceneJsonPath, JSON.toJSONString(sceneJson));
  219. //上传sceneJson文件
  220. String sceneJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH+"scene.json", num);
  221. uploadToOssUtil.upload(JSON.toJSONBytes(sceneJson), sceneJsonPath);
  222. //sceneJson放入缓存
  223. String key = String.format(RedisKey.SCENE_JSON, num);
  224. redisUtil.set(key, JSON.toJSONString(sceneJson));
  225. //删除发布数据中的user目录
  226. String publicUserPath = String.format(UploadFilePath.USER_VIEW_PATH, num);
  227. uploadToOssUtil.deleteFile(publicUserPath);
  228. // String editUserPath = String.format(UploadFilePath.USER_EDIT_PATH, sceneNum);
  229. //复制编辑目录到发布目录
  230. String editPath = String.format(UploadFilePath.EDIT_PATH, param.getNum());
  231. String viewPath = String.format(UploadFilePath.VIEW_PATH, param.getNum());
  232. uploadToOssUtil.copyFiles(editPath,viewPath);
  233. //入库
  234. if(sceneEditInfo.getId() == null){
  235. this.save(sceneEditInfo);
  236. }else{
  237. this.updateById(sceneEditInfo);
  238. }
  239. sceneEditInfoExt.setSceneProId(scenePro.getId());
  240. sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId());
  241. sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt);
  242. // sceneEditControls.setEditInfoId(sceneEditInfo.getId());
  243. // if(sceneEditControls.getId() == null){
  244. // sceneEditControlsService.save(sceneEditControls);
  245. // }else{
  246. // sceneEditControlsService.updateById(sceneEditControls);
  247. // }
  248. return ResultData.ok();
  249. }
  250. private void buildVideo(SceneEditInfo sceneEditInfo, String path, String num) throws Exception{
  251. if(CommonStatus.NO.equals(sceneEditInfo.getBuildVideoStatus())){
  252. return;
  253. }
  254. if(path != null && !"".equals(path) && path.startsWith("http")){
  255. path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2];
  256. }
  257. String target = path + "_images";
  258. log.info("球幕视频路径:{}", target + File.separator + "extras/video");
  259. File video = new File(target + File.separator + "extras/video");
  260. //如果文件夹或者文件不存在,跳出
  261. if(!video.exists() || video.listFiles() == null || video.listFiles().length == 0){
  262. return;
  263. }
  264. String userEditPath = String.format(UploadFilePath.USER_EDIT_PATH, num);
  265. for(String videoName : video.list()){
  266. log.info("球幕视频名称:{}", videoName);
  267. uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName,userEditPath + videoName);
  268. CreateObjUtil.mp4ToFlv(target + File.separator + "extras/video/" + videoName,
  269. target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"));
  270. uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"),userEditPath + videoName.replace("mp4", "flv"));
  271. //覆盖原始视频资源
  272. FileUtils.copyFile(target + File.separator + "extras/video/" + videoName,
  273. path + File.separator + "caches/videos/" + videoName, true);
  274. }
  275. FileUtils.deleteDirectory(target + File.separator + "extras/video/");
  276. //重置状态
  277. sceneEditInfo.setBuildVideoStatus(CommonStatus.NO.code());
  278. }
  279. private void publicHotData(String sceneNum, SceneEditInfo sceneEditInfo) throws IOException {
  280. String hotDataKey = String.format(RedisKey.SCENE_HOT_DATA, sceneNum);
  281. Map<String, String> hotMap = redisUtil.hmget(hotDataKey);
  282. List<String> hotList = Lists.newArrayList();
  283. hotMap.entrySet().stream().forEach(entry->{
  284. if(StrUtil.isNotEmpty(entry.getValue())){
  285. hotList.add(entry.getValue());
  286. }
  287. });
  288. if(CollUtil.isNotEmpty(hotList)){
  289. JSONArray jsonhots = new JSONArray();
  290. hotList.stream().forEach(hot->{
  291. jsonhots.add(JSONObject.parseObject(hot));
  292. });
  293. String hotJsonPath = String.format(UploadFilePath.USER_EDIT_PATH, sceneNum) + "hot.json";
  294. uploadToOssUtil.upload(jsonhots.toString().getBytes(), hotJsonPath);
  295. //修改tags状态为是,标识有热点数据
  296. this.saveTagsToSceneEditInfo(sceneNum, sceneEditInfo);
  297. }
  298. }
  299. @Override
  300. public SceneInfoVO getSceneInfo(@Var SceneInfoParamVO param) throws Exception{
  301. SceneInfoReqType sceneInfoReqType = SceneInfoReqType.get(param.getReqType());
  302. switch (sceneInfoReqType){
  303. //如果是编辑页面请求,查数据库
  304. case EDIT:
  305. return this.getSceneInfo4Edit(param.getNum());
  306. //如果是查看页面请求,查redis
  307. case VIEW:
  308. return this.getSceneInfo4View(param.getNum());
  309. }
  310. return null;
  311. }
  312. /**
  313. * <p>
  314. 编辑页面获取场景详情信息
  315. * </p>
  316. * @author dengsixing
  317. * @date 2022/1/21
  318. * @param num
  319. * @return com.fdkankan.scene.vo.SceneInfoVO
  320. **/
  321. private SceneInfoVO getSceneInfo4Edit(String num) throws Exception{
  322. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  323. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  324. if(Objects.isNull(scenePlus)){
  325. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  326. }
  327. // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  328. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  329. SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId());
  330. if(Objects.isNull(sceneEditInfo)){
  331. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  332. }
  333. SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());
  334. SceneEditControls sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());
  335. SceneInfoVO sceneInfoVO = new SceneInfoVO();
  336. BeanUtil.copyProperties(sceneEditInfoExt, sceneInfoVO);
  337. BeanUtil.copyProperties(sceneEditInfo, sceneInfoVO);
  338. if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){
  339. sceneInfoVO.setFloorPlanAngle(0f);
  340. }
  341. if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){
  342. sceneInfoVO.setFloorPlanCompass(0f);
  343. }
  344. sceneInfoVO.setControls(BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class));
  345. sceneInfoVO.setNum(num);
  346. //生成sceneJson
  347. // if(StrUtil.isNotEmpty(sceneEditInfo.getFloorPlanPath())){
  348. // sceneInfoVO.setFloorPlanPaths(sceneEditInfo.getFloorPlanPath().split(","));
  349. // }
  350. sceneInfoVO.setCreateTime(scenePlus.getCreateTime());
  351. sceneInfoVO.setSceneResolution(scenePlusExt.getSceneResolution());
  352. sceneInfoVO.setSceneFrom(scenePlusExt.getSceneFrom());
  353. sceneInfoVO.setVideos(scenePlusExt.getVideos());
  354. this.setExtData(sceneInfoVO, scenePlus.getCameraId());
  355. return sceneInfoVO;
  356. }
  357. /**
  358. * <p>
  359. getInfo接口返回字段扩展
  360. * </p>
  361. * @author dengsixing
  362. * @date 2022/3/9
  363. **/
  364. private void setExtData(SceneInfoVO sceneInfoVO, Long cameraId) throws Exception{
  365. ResultData<CameraDetail> cameraResultData = platformGoodsClient.getCameraDetailByCameraId(cameraId);
  366. if(!cameraResultData.getSuccess()){
  367. throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message());
  368. }
  369. CameraDetail camera = cameraResultData.getData();
  370. if(Objects.isNull(camera)){
  371. throw new BusinessException(ErrorCode.FAILURE_CODE_7010);
  372. }
  373. if(Objects.isNull(camera.getCompanyId())){
  374. return;
  375. }
  376. ResultData<Company> companyResultData = platformUserClient.getCompanyById(camera.getCompanyId());
  377. if(!companyResultData.getSuccess()){
  378. throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message());
  379. }
  380. Company company = companyResultData.getData();
  381. if(Objects.isNull(company)){
  382. return;
  383. }
  384. sceneInfoVO.setDataSync(company.getDataSync());
  385. }
  386. /**
  387. * <p>
  388. 查看页面获取场景详情信息
  389. * </p>
  390. * @author dengsixing
  391. * @date 2022/1/21
  392. * @param num
  393. * @return com.fdkankan.scene.vo.SceneInfoVO
  394. **/
  395. private SceneInfoVO getSceneInfo4View(String num) throws Exception{
  396. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  397. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  398. String key = String.format(RedisKey.SCENE_JSON, num);
  399. String sceneJson = redisUtil.get(key);
  400. SceneInfoVO sceneInfoVO = null;
  401. //先查询redis
  402. if(StrUtil.isNotEmpty(sceneJson)) {
  403. sceneInfoVO = JSON.parseObject(sceneJson, SceneInfoVO.class);
  404. this.setExtData(sceneInfoVO, scenePlus.getCameraId());
  405. sceneInfoVO.setScenePassword(null);
  406. if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){
  407. sceneInfoVO.setFloorPlanAngle(0f);
  408. }
  409. if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){
  410. sceneInfoVO.setFloorPlanCompass(0f);
  411. }
  412. return sceneInfoVO;
  413. }
  414. //如果redis没找到,从scene.json中获取
  415. String objectName = String.format(ConstantFilePath.SCENE_VIEW_DATA_DATA_SCENEJSON, num);
  416. String objectContent = uploadToOssUtil.getObjectContent(bucket, objectName);
  417. if(StrUtil.isEmpty(objectContent))
  418. return null;
  419. sceneInfoVO = JSON.parseObject(objectContent, SceneInfoVO.class);
  420. this.setExtData(sceneInfoVO, scenePlus.getCameraId());
  421. sceneInfoVO.setScenePassword(null);
  422. if(Objects.isNull(sceneInfoVO.getFloorPlanAngle())){
  423. sceneInfoVO.setFloorPlanAngle(0f);
  424. }
  425. if(Objects.isNull(sceneInfoVO.getFloorPlanCompass())){
  426. sceneInfoVO.setFloorPlanCompass(0f);
  427. }
  428. redisUtil.set(key, JSON.toJSONString(sceneInfoVO));
  429. return sceneInfoVO;
  430. }
  431. @Override
  432. // @Transactional
  433. public ResultData saveUpload(SaveUploadParamVO param) throws Exception {
  434. FileBizType fileBizType = FileBizType.get(param.getBizType());
  435. String num = param.getNum();
  436. String fileInfo = param.getFileInfo();
  437. if(Objects.isNull(fileBizType)){
  438. throw new BusinessException(ErrorCode.FAILURE_CODE_7003);
  439. }
  440. switch (fileBizType){
  441. case MUSIC:
  442. return new MusicHandlerFactory().getHandler().save(num, fileInfo);
  443. case FLOOR_LOGO:
  444. return new FloorLogoHandlerFactory().getHandler().save(num, fileInfo);
  445. // case FLOOR_PLAN:
  446. // return new FloorPlanHandlerFactory().getHandler().save(num, fileInfo);
  447. case SCREENSHOT:
  448. return new ScreenshotHandlerFactory().getHandler().save(num, fileInfo);
  449. }
  450. return ResultData.ok();
  451. }
  452. @Override
  453. public ResultData saveCad(BaseDataParamVO param) throws Exception {
  454. String num = param.getNum();
  455. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  456. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  457. if(Objects.isNull(scenePlus))
  458. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  459. String editDataPath = String.format(UploadFilePath.DATA_EDIT_PATH, num);
  460. String editUserPath = String.format(UploadFilePath.USER_EDIT_PATH, num);
  461. String localDataPath = String.format(ConstantFilePath.DATABUFFER_FORMAT, num);
  462. String floorCadUrl = ossUrlPrefix + editDataPath + "floorplan_cad.json?t=" + System.currentTimeMillis();
  463. String floorUserUrl = ossUrlPrefix + editUserPath + "floorplan_user.json?t=" + System.currentTimeMillis();
  464. JSONObject fileInfoJson = JSON.parseObject(param.getData());
  465. String filePathStr = null;
  466. String floors = fileInfoJson.getString("floors");
  467. String cadInfo = null;
  468. Float angel = null;
  469. Float compass = null;
  470. String floorJsonData = fileInfoJson.getString("floorJsonData");
  471. if(StrUtil.isNotBlank(floorJsonData)){
  472. JSONObject jsonObject = JSON.parseObject(floorJsonData);
  473. cadInfo = jsonObject.getString("cadInfo");
  474. angel = jsonObject.getFloat("angle");
  475. compass = jsonObject.getFloat("compass");
  476. }
  477. // String filePaths = fileInfoJson.getString("filePaths");
  478. // Byte reSet = fileInfoJson.getByte("reset");
  479. //户型图文件路径
  480. // if(StrUtil.isNotEmpty(filePaths)){
  481. // cn.hutool.json.JSONArray filePathsArrs = JSONUtil.parseArray(filePaths);
  482. // List<String> filePathsList = filePathsArrs.toList(String.class);
  483. // StringBuilder filePathBuilder = new StringBuilder();
  484. // filePathsList.parallelStream().forEach(path -> {
  485. // filePathBuilder.append(",").append(path);
  486. // });
  487. // filePathStr = filePathBuilder.substring(1);
  488. // }
  489. //处理户型图数据
  490. JSONObject floorUserJson = null;
  491. if(StrUtil.isNotEmpty(floors)) {
  492. //如果入参是空,则加载服务器的
  493. if(StrUtil.isEmpty(floorJsonData)){
  494. if (!StorageType.LOCAL.code().equals(this.type)) {// TODO: 2022/2/15 这里有可能有问题,可能还需要考虑本地部署的情况
  495. FileUtils.downLoadFromUrl(floorUserUrl, "floorplan_user.json", localDataPath);
  496. }
  497. floorJsonData = FileUtils.readFile(localDataPath + "floorplan_user.json");
  498. }
  499. //如果floors不为空,需要根据对应楼层id修改楼层名称
  500. if (StrUtil.isNotEmpty(floorJsonData)) {
  501. floorUserJson = this.updateFloorName(floors, floorJsonData);
  502. floorJsonData = floorUserJson.toJSONString();
  503. }
  504. }
  505. //上传floorplan_user.json文件
  506. FileUtils.writeFile(localDataPath + "floorplan_user.json", floorJsonData);
  507. uploadToOssUtil.upload(localDataPath + "floorplan_user.json", editUserPath + "floorplan_user.json");
  508. //如果floors不为空,需要根据对应楼层id修改楼层名称
  509. if(StrUtil.isNotEmpty(floors)) {
  510. if (!StorageType.LOCAL.code().equals(this.type)) {// TODO: 2022/2/15 这里有可能有问题,可能还需要考虑本地部署的情况
  511. FileUtils.downLoadFromUrl(floorCadUrl, "floorplan_cad.json", localDataPath);
  512. }
  513. String floorCadStr = FileUtils.readFile(localDataPath + "floorplan_cad.json");
  514. if (StrUtil.isNotEmpty(floorCadStr)) {
  515. JSONObject floorCadJson = this.updateFloorName(floors, floorCadStr);
  516. FileUtils.writeFile(localDataPath + "floorplan_cad.json", floorCadJson.toString());
  517. uploadToOssUtil.upload(localDataPath + "floorplan_cad.json",editDataPath + "floorplan_cad.json");
  518. }
  519. }
  520. //写入数据库
  521. Byte floorPlanUser = null;
  522. if(StrUtil.isNotEmpty(floorJsonData)){
  523. floorPlanUser = CommonStatus.YES.code();
  524. }
  525. // if(CommonStatus.YES.code().equals(reSet)){
  526. // floorPlanUser = CommonStatus.NO.code();
  527. // }
  528. SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId());
  529. if(Objects.nonNull(sceneEditInfoDb)){
  530. LambdaUpdateWrapper<SceneEditInfo> updateWrapper = new LambdaUpdateWrapper<SceneEditInfo>()
  531. .setSql("version=version+" + 1)
  532. .eq(SceneEditInfo::getId, sceneEditInfoDb.getId());
  533. // if(StrUtil.isNotEmpty(filePathStr)){
  534. // updateWrapper.set(SceneEditInfo::getFloorPlanPath, filePathStr);
  535. // }
  536. if(StrUtil.isNotEmpty(cadInfo)){
  537. updateWrapper.set(SceneEditInfo::getCadInfo, cadInfo);
  538. }
  539. if(floorPlanUser != null){
  540. updateWrapper.set(SceneEditInfo::getFloorPlanUser, floorPlanUser);
  541. }
  542. this.update(updateWrapper);
  543. }else{
  544. sceneEditInfoDb = new SceneEditInfo();
  545. sceneEditInfoDb.setScenePlusId(scenePlus.getId());
  546. // sceneEditInfoDb.setFloorPlanPath(filePathStr);
  547. sceneEditInfoDb.setCadInfo(cadInfo);
  548. sceneEditInfoDb.setFloorPlanUser(floorPlanUser);
  549. this.save(sceneEditInfoDb);
  550. }
  551. SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfoDb.getId());
  552. if(Objects.isNull(sceneEditInfoExt)){
  553. sceneEditInfoExt = new SceneEditInfoExt();
  554. sceneEditInfoExt.setEditInfoId(sceneEditInfoDb.getId());
  555. }
  556. if(Objects.nonNull(angel)){
  557. sceneEditInfoExt.setFloorPlanAngle(angel);
  558. }
  559. if(Objects.nonNull(compass)){
  560. sceneEditInfoExt.setFloorPlanCompass(compass);
  561. }
  562. sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt);
  563. return ResultData.ok();
  564. }
  565. @Override
  566. public ResultData resetCad(String num){
  567. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  568. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  569. if(Objects.isNull(scenePlus))
  570. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  571. SceneEditInfo sceneEditInfoDb = this.getByScenePlusId(scenePlus.getId());
  572. SceneEditInfoExt sceneEditInfoExt = null;
  573. if(Objects.nonNull(sceneEditInfoDb)){
  574. LambdaUpdateWrapper<SceneEditInfo> updateWrapper = new LambdaUpdateWrapper<SceneEditInfo>()
  575. .setSql("version=version+" + 1)
  576. .set(SceneEditInfo::getFloorPlanUser, CommonStatus.NO.code())
  577. .eq(SceneEditInfo::getId, sceneEditInfoDb.getId());
  578. this.update(updateWrapper);
  579. sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfoDb.getId());
  580. }else{
  581. sceneEditInfoDb = new SceneEditInfo();
  582. sceneEditInfoDb.setScenePlusId(scenePlus.getId());
  583. sceneEditInfoDb.setFloorPlanUser(CommonStatus.NO.code());
  584. this.save(sceneEditInfoDb);
  585. }
  586. if(Objects.isNull(sceneEditInfoExt)){
  587. sceneEditInfoExt = new SceneEditInfoExt();
  588. sceneEditInfoExt.setSceneProId(scenePro.getId());
  589. sceneEditInfoExt.setEditInfoId(sceneEditInfoDb.getId());
  590. }
  591. sceneEditInfoExt.setFloorPlanAngle(0f);
  592. sceneEditInfoExt.setFloorPlanCompass(0f);
  593. sceneEditInfoExtService.saveOrUpdate(sceneEditInfoExt);
  594. return ResultData.ok();
  595. }
  596. private JSONObject updateFloorName(String sourceFloors, String targeFloors){
  597. JSONArray sourceFloorsJson = JSON.parseArray(sourceFloors);
  598. JSONObject targeFloorsJson = JSONObject.parseObject(targeFloors);
  599. JSONArray array = targeFloorsJson.getJSONArray("floors");
  600. for (int i = 0; i < array.size(); ++i) {
  601. JSONObject targetFloor = array.getJSONObject(i);
  602. int targetId = targetFloor.getIntValue("id");
  603. for (int j = 0; j < sourceFloorsJson.size(); ++j) {
  604. JSONObject floor = sourceFloorsJson.getJSONObject(j);
  605. int id = floor.getIntValue("id");
  606. String name = floor.getString("name");
  607. if (targetId != id)
  608. continue;
  609. targetFloor.put("name", name);
  610. }
  611. }
  612. return targeFloorsJson;
  613. }
  614. @Override
  615. public void upgradeVersionById(Long id) {
  616. this.update(new LambdaUpdateWrapper<SceneEditInfo>()
  617. .setSql("version=version+" + 1)
  618. .eq(SceneEditInfo::getId, id));
  619. }
  620. @Override
  621. public ResultData uploadPanorama(String num, MultipartFile file) throws Exception {
  622. //校验压缩包格式
  623. log.info("全景图压缩包文件名:" + file.getOriginalFilename());
  624. if(!file.getOriginalFilename().endsWith(".zip")){
  625. throw new BusinessException(ErrorCode.FAILURE_CODE_7015);
  626. }
  627. ScenePro scenePro = sceneProService.findBySceneNum(num);
  628. if(scenePro == null){
  629. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  630. }
  631. SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  632. //原始计算根目录
  633. String path = sceneProExt.getDataSource();
  634. //全景图计算根目录
  635. String target = path + "_images";
  636. //解压缩文件存放目录
  637. String targetImagesPath = target + "/extras/images";
  638. //压缩文件保存目录
  639. String zipTargetFilePath = targetImagesPath + File.separator + file.getOriginalFilename();
  640. //压缩包保存到本地
  641. // String cachePath = String.format(ConstantFilePath.SCENE_CACHE_IMAGES, num);
  642. // String path = "F:\\mnt\\4Dkankan\\scene\\t-ieXdyGl6Md\\caches\\images\\";
  643. //先删除本地文件
  644. FileUtils.deleteDirectory(targetImagesPath);
  645. File targetFile = new File(zipTargetFilePath);
  646. if(!targetFile.getParentFile().exists()){
  647. targetFile.getParentFile().mkdirs();
  648. }
  649. file.transferTo(targetFile);
  650. //解压zip包
  651. FileUtil.unZip(zipTargetFilePath, targetImagesPath);
  652. //删除压缩包
  653. FileUtil.delFile(zipTargetFilePath);
  654. //获取解压后的文件列表
  655. List<String> uploadFileList = FileUtil.getFileList(targetImagesPath);
  656. if(CollUtil.isEmpty(uploadFileList)){
  657. throw new BusinessException(ErrorCode.FAILURE_CODE_7014);
  658. }
  659. //列出caches/images中的文件列表
  660. String imgCachePath = String.format(UploadFilePath.IMG_CACHES_PATH, num);
  661. List<String> keyList = uploadToOssUtil.listKeys(imgCachePath);
  662. //比对图片列表,不存在的要返回名称集合
  663. List<String> notExistFileList = uploadFileList.stream().filter(filePath -> {
  664. filePath = filePath.substring(filePath.lastIndexOf(File.separator) + 1);
  665. for (String key : keyList) {
  666. key = key.substring(key.lastIndexOf("/") + 1);
  667. if (filePath.equals(key)) {
  668. return false;
  669. }
  670. }
  671. return true;
  672. }).map(filePath -> {
  673. return filePath.substring(filePath.lastIndexOf(File.separator) + 1);
  674. }).collect(Collectors.toList());
  675. //有文件对不上号,就退出
  676. if(CollUtil.isNotEmpty(notExistFileList)){
  677. return ResultData.error(ErrorCode.FAILURE_CODE_7012, notExistFileList);
  678. }
  679. //上传
  680. Map<String, String> map = new HashMap<>();
  681. uploadFileList.stream().forEach(filePath->{
  682. map.put(filePath, filePath.replace(targetImagesPath, imgCachePath));
  683. });
  684. String imgViewPath = String.format(UploadFilePath.IMG_VIEW_PATH, num);
  685. if(StorageType.AWS.code().equals(this.type)){
  686. FileUtils.downLoadFromUrl(ConstantUrl.PREFIX_AWS + imgViewPath + "vision.modeldata" + "?m="+new Date().getTime(),
  687. "vision.modeldata", target + File.separator + "extras" + File.separator);
  688. }
  689. if(StorageType.OSS.code().equals(this.type)){
  690. FileUtils.downLoadFromUrl(prefixAli + imgViewPath + "vision.modeldata" + "?m="+new Date().getTime(),
  691. "vision.modeldata", target + File.separator + "extras" + File.separator);
  692. }
  693. CreateObjUtil.convertVisionmodeldataToTxt(target + File.separator + "extras" + File.separator + "vision.modeldata",
  694. target + File.separator + "extras" + File.separator + "vision.txt");
  695. FileUtils.copyFile(path + File.separator + "data.json", target + File.separator+"data.json", true);
  696. FileUtils.copyFile(path + File.separator + "project.json", target + File.separator+"project.json", true);
  697. //data.json增加extras为执行重建算法
  698. String data = FileUtils.readFile(target + File.separator+"data.json");
  699. if(data != null){
  700. JSONObject floorplanJson = new JSONObject();
  701. floorplanJson.put("has_source_images", true);
  702. floorplanJson.put("has_vision_txt", true);
  703. JSONObject dataJson = JSONObject.parseObject(data);
  704. dataJson.put("extras", floorplanJson);
  705. //V5表示不需要生成high,low文件
  706. dataJson.put("skybox_type", "SKYBOX_V6");
  707. if(scenePro.getSceneScheme() == 11){
  708. dataJson.put("skybox_type", "SKYBOX_V7");
  709. }
  710. dataJson.put("split_type", "SPLIT_V8");
  711. //sceneScheme为3切成瓦片图
  712. if(scenePro.getSceneScheme() == 3){
  713. dataJson.put("skybox_type", "SKYBOX_V4");
  714. }
  715. FileUtils.writeFile(target + File.separator+"data.json", new String(dataJson.toString().getBytes(), "UTF-8"));
  716. }
  717. if(new File(target + File.separator + "capture").exists()){
  718. new File(target + File.separator + "capture").delete();
  719. }
  720. if(new File(target + File.separator + "results").exists()){
  721. FileUtils.delAllFile(target + File.separator + "results");
  722. }
  723. if(StorageType.AWS.code().equals(this.type)){
  724. //亚马逊保持旧方式,超链接capture
  725. CreateObjUtil.createSoftConnection(path + File.separator + "capture", target + File.separator + "capture");
  726. }
  727. CreateObjUtil.ossUtilCp(ConstantFilePath.OSS_PREFIX + path.replace(ConstantFilePath.BUILD_MODEL_PATH, "") + "/data.fdage", target + File.separator + "capture/");
  728. CreateObjUtil.build3dModel(target , "1");
  729. String uploadJsonPath= target + File.separator + "results" +File.separator+"upload.json";
  730. Thread.sleep(2000);
  731. boolean exist = ComputerUtil.checkComputeCompleted(uploadJsonPath, 3, 5000);
  732. if(!exist){
  733. throw new BusinessException(ErrorCode.FAILURE_CODE_5042);
  734. }
  735. String uploadData = FileUtils.readFile(uploadJsonPath);
  736. JSONObject uploadJson = null;
  737. JSONArray array = null;
  738. if(uploadData!=null) {
  739. uploadJson = JSONObject.parseObject(uploadData);
  740. array = uploadJson.getJSONArray("upload");
  741. }
  742. if(array == null){
  743. throw new Exception("upload.json数据出错");
  744. }
  745. JSONObject fileJson = null;
  746. String fileName = "";
  747. for(int i = 0, len = array.size(); i < len; i++) {
  748. fileJson = array.getJSONObject(i);
  749. fileName = fileJson.getString("file");
  750. //文件不存在抛出异常
  751. if (!new File(target + File.separator + "results" + File.separator + fileName)
  752. .exists()) {
  753. throw new Exception(
  754. target + File.separator + "results" + File.separator + fileName + "文件不存在");
  755. }
  756. //high文件夹
  757. if (fileJson.getIntValue("clazz") == 3) {
  758. map.put(target + File.separator + "results" + File.separator + fileName,
  759. imgViewPath + "pan/high/" + fileName.replace("high/", ""));
  760. continue;
  761. }
  762. //low文件夹
  763. if (fileJson.getIntValue("clazz") == 4) {
  764. map.put(target + File.separator + "results" + File.separator + fileName,
  765. imgViewPath + "pan/low/" + fileName.replace("low/", ""));
  766. continue;
  767. }
  768. //tiles文件夹,亚马逊没有裁剪图片api,不需要上传4k图
  769. if (fileJson.getIntValue("clazz") == 5) {
  770. map.put(target + File.separator + "results" + File.separator + fileName,
  771. imgViewPath + fileName);
  772. continue;
  773. }
  774. //tiles文件夹,亚马逊瓦片图
  775. if (fileJson.getIntValue("clazz") == 7) {
  776. map.put(target + File.separator + "results" + File.separator + fileName,
  777. imgViewPath + fileName);
  778. continue;
  779. }
  780. }
  781. if(map.size()>0) {
  782. uploadToOssUtil.uploadMulFiles(map);
  783. }
  784. return null;
  785. }
  786. @Override
  787. public ResultData downloadPanorama(FileParamVO param) throws Exception {
  788. String num = param.getNum();
  789. String fileName = param.getFileName();
  790. String cachePath = String.format(ConstantFilePath.SCENE_CACHE, num);
  791. String imgCachePath = String.format(UploadFilePath.IMG_CACHES_PATH, num);
  792. String localImagesPath = String.format(ConstantFilePath.SCENE_CACHE_IMAGES, num);
  793. String cacheFormat = "downloads/scene/%s/caches/";
  794. String cacheImageFormat = "downloads/scene/%s/caches/images/";
  795. //删除本地文件
  796. FileUtils.deleteDirectory(localImagesPath);
  797. //如果入参文件名不为空,则是单个文件下载,不需要打包
  798. if(StrUtil.isNotEmpty(fileName)){
  799. //先下载到本地
  800. if (!StorageType.LOCAL.code().equals(this.type)) {// TODO: 2022/2/15 这里有可能有问题,可能还需要考虑本地部署的情况
  801. String filePath = imgCachePath + fileName;
  802. String imageUrl = ossUrlPrefix + filePath + "?t=" + System.currentTimeMillis();
  803. FileUtils.downLoadFromUrl(imageUrl, fileName, localImagesPath);
  804. }
  805. }else{
  806. //先下载到本地
  807. List<String> keyList = uploadToOssUtil.listKeys(imgCachePath);
  808. if (!StorageType.LOCAL.code().equals(this.type)) {// TODO: 2022/2/15 这里有可能有问题,可能还需要考虑本地部署的情况
  809. keyList.stream().forEach(key->{
  810. String file = key.substring(key.lastIndexOf("/") + 1);
  811. String imageUrl = ossUrlPrefix + imgCachePath + file + "?t=" + System.currentTimeMillis();
  812. FileUtils.downLoadFromUrl(imageUrl, file, localImagesPath);
  813. });
  814. }
  815. }
  816. //打包
  817. String zipName = num + "_images.zip";
  818. String zipPath = cachePath + zipName;
  819. FileUtils.zipFile(zipPath, localImagesPath);
  820. //上传压缩包
  821. uploadToOssUtil.upload(zipPath, String.format(cacheFormat, num) + zipName);
  822. String url = ossUrlPrefix + String.format(cacheFormat, num) + zipName + "?t=" + Calendar.getInstance().getTimeInMillis();
  823. //删除本地压缩包
  824. FileUtils.deleteFile(zipPath);
  825. //删除本地目录
  826. FileUtils.deleteDirectory(localImagesPath);
  827. Map<String, Object> map = new HashMap<>();
  828. map.put("fileUrl", url + "?t=" + System.currentTimeMillis());
  829. map.put("fileName", zipName);
  830. return ResultData.ok(map);
  831. }
  832. @Override
  833. public void saveTagsToSceneEditInfo(String num, SceneEditInfo sceneEditInfo){
  834. //查询缓存是否包含热点数据
  835. String key = String.format(RedisKey.SCENE_HOT_DATA, num);
  836. Map<String, String> allTagsMap = redisUtil.hmget(key);
  837. boolean hashTags = false;
  838. for (Entry<String, String> tagMap : allTagsMap.entrySet()) {
  839. if(StrUtil.isEmpty(tagMap.getValue())){
  840. continue;
  841. }
  842. hashTags = true;
  843. break;
  844. }
  845. //更改热点状态
  846. sceneEditInfo.setTags(hashTags ? CommonStatus.YES.code() : CommonStatus.NO.code());
  847. //version 是空的代表 sceneEditInfo记录集不存在
  848. if(sceneEditInfo.getVersion() != null){
  849. sceneEditInfo.setVersion(sceneEditInfo.getVersion() + 1);
  850. }
  851. }
  852. @Override
  853. public ResultData saveVideoBox(FileNameAndDataParamVO param) throws Exception {
  854. JSONObject boxVideo = JSONObject.parseObject(param.getData());
  855. String sid = boxVideo.getString("sid");
  856. if(StrUtil.isEmpty(sid)){
  857. throw new BusinessException(ErrorCode.PARAM_REQUIRED);
  858. }
  859. // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum());
  860. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum());
  861. if(Objects.isNull(scenePlus))
  862. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  863. SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId());
  864. //转换视频格式
  865. this.transferToFlv(param.getNum(), param.getFileName());
  866. //生成boxVideos数据
  867. String boxVideos = this.createBoxVideos(sid, boxVideo, sceneEditInfo, OperationType.ADDORUPDATE.code());
  868. //更新数据库
  869. this.updateBoxVideos(sceneEditInfo, scenePlus.getId(), boxVideos);
  870. return ResultData.ok();
  871. }
  872. @Override
  873. public ResultData deleteVideoBox(DeleteVidoeBoxParamVO param) throws Exception {
  874. // ScenePro scenePro = sceneProService.findBySceneNum(param.getNum());
  875. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(param.getNum());
  876. if(Objects.isNull(scenePlus))
  877. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  878. SceneEditInfo sceneEditInfo = this.getByScenePlusId(scenePlus.getId());
  879. //根据sid移除json
  880. String boxVideos = this.createBoxVideos(param.getSid(), null, sceneEditInfo, OperationType.DELETE.code());
  881. //写数据库
  882. this.updateBoxVideos(sceneEditInfo,scenePlus.getId(),boxVideos);
  883. return ResultData.ok();
  884. }
  885. @Override
  886. public DownloadVO downloadBallScreenVideo(BallScreenVideoParamVO param) {
  887. String videoPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, param.getNum()) + param.getFileName();
  888. String url = ossUrlPrefix + videoPath + "?t=" + System.currentTimeMillis();
  889. return DownloadVO.builder()
  890. .fileName(param.getFileName())
  891. .url(url)
  892. .build();
  893. }
  894. @Override
  895. public ResultData uploadBallScreenVideo(String num, String fileName, MultipartFile file) throws Exception {
  896. //校验格式
  897. if(!fileName.endsWith(".mp4")){
  898. throw new BusinessException(ErrorCode.FAILURE_CODE_7016);
  899. }
  900. String videosViewPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, num);
  901. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  902. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  903. if(scenePlus == null){
  904. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  905. }
  906. // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  907. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  908. String path = scenePlusExt.getDataSource();
  909. //校验文件名
  910. String videos = scenePro.getVideos();
  911. if(StrUtil.isEmpty(videos)){
  912. throw new BusinessException(ErrorCode.FAILURE_CODE_7012);
  913. }
  914. JSONObject videosJson = JSON.parseObject(videos);
  915. JSONArray dataArr = videosJson.getJSONArray("data");
  916. boolean exists = false;
  917. for (Object o : dataArr) {
  918. JSONObject jsonObject = (JSONObject)o;
  919. if(jsonObject.getString("id").equals(fileName.replace(".mp4", ""))){
  920. exists = true;
  921. break;
  922. }
  923. }
  924. if(!exists){
  925. throw new BusinessException(ErrorCode.FAILURE_CODE_7012);
  926. }
  927. if(path != null && !"".equals(path) && path.startsWith("http")){
  928. path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2];
  929. }
  930. String target = path + "_images";
  931. String filePath = target + File.separator + "extras/video" + File.separator + fileName;
  932. File targetFile = new File(filePath);
  933. if(!targetFile.getParentFile().exists()){
  934. targetFile.getParentFile().mkdirs();
  935. }
  936. //保存视频到本地
  937. file.transferTo(targetFile);
  938. File video = new File(target + File.separator + "extras/video");
  939. //如果文件夹或者文件不存在,跳出
  940. if(!video.exists() || video.listFiles() == null || video.listFiles().length == 0){
  941. throw new BusinessException(ErrorCode.FAILURE_CODE_7009);
  942. }
  943. for(String videoName : video.list()){
  944. log.info("球幕视频名称:{}", videoName);
  945. uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName,videosViewPath + videoName);
  946. CreateObjUtil.mp4ToFlv(target + File.separator + "extras/video/" + videoName,
  947. target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"));
  948. uploadToOssUtil.upload(target + File.separator + "extras/video/" + videoName.replace("mp4", "flv"),videosViewPath + videoName.replace("mp4", "flv"));
  949. //覆盖原始视频资源
  950. FileUtils.copyFile(target + File.separator + "extras/video/" + videoName,
  951. path + File.separator + "caches/videos/" + videoName, true);
  952. }
  953. FileUtils.deleteDirectory(target + File.separator + "extras/video/");
  954. // SceneEditInfo sceneEditInfo = this.getBySceneProId(scenePro.getId());
  955. // if(sceneEditInfo == null){
  956. // sceneEditInfo = new SceneEditInfo();
  957. // sceneEditInfo.setSceneProId(scenePro.getId());
  958. // sceneEditInfo.setBuildVideoStatus(CommonStatus.YES.code());
  959. // this.save(sceneEditInfo);
  960. // }else{
  961. // sceneEditInfo.setBuildVideoStatus(CommonStatus.YES.code());
  962. // this.updateById(sceneEditInfo);
  963. // }
  964. return ResultData.ok();
  965. }
  966. @Override
  967. public ResultData sceneSync(String num, String type, String floorPlanJson, String ajkJson, String cameraJson,
  968. MultipartFile[] files) throws Exception {
  969. String lockKey = String.format(RedisLockKey.LOCK_SCENE_SYNC, num);
  970. Boolean lock = redisLockUtil.lock(lockKey, RedisKey.EXPIRE_TIME_2_HOUR);
  971. if(!lock){
  972. throw new BusinessException(ErrorCode.FAILURE_CODE_5036);
  973. }
  974. try {
  975. // ScenePro scenePro = sceneProService.findBySceneNum(num);
  976. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  977. if(scenePlus == null){
  978. throw new BusinessException(ErrorCode.FAILURE_CODE_5005);
  979. }
  980. // SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  981. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  982. //更新scene.json文件
  983. String strsceneInfos = FileUtils.readFile(ConstantFilePath.SCENE_PATH + "data" + File.separator + "data" + num + File.separator + "scene.json");
  984. if(strsceneInfos == null)
  985. new File(ConstantFilePath.SCENE_PATH + "data" + File.separator + "data" + num + File.separator + "scene.json").createNewFile();
  986. String path = scenePlusExt.getDataSource();// /mnt/data/0662c5389/831989883441512448/4898cab04f8c_202104141602356060/
  987. if(path != null && !"".equals(path) && path.startsWith("http")){
  988. path = ConstantFilePath.BUILD_MODEL_PATH + File.separator + path.split("/")[path.split("/").length - 2];
  989. }
  990. String target = path + "_ajk"; // /mnt/data/0662c5389/831989883441512448/4898cab04f8c_202104141602356060_ajk
  991. File editPath = new File(target);
  992. if(!editPath.exists()){
  993. editPath.mkdirs();
  994. }
  995. //创建文件夹软连接并且复制data.json和project.json
  996. if(new File(target + File.separator + "capture").exists()){
  997. new File(target + File.separator + "capture").delete();
  998. }
  999. if(new File(target + File.separator + "caches").exists()){
  1000. //删除link
  1001. new File(target + File.separator + "caches" + File.separator + "images").delete();
  1002. //删除所有文件
  1003. FileUtils.delAllFile(target + File.separator + "caches");
  1004. }
  1005. if(new File(target + File.separator + "results").exists()){
  1006. FileUtils.delAllFile(target + File.separator + "results");
  1007. }
  1008. //创建文件夹,并link文件夹
  1009. new File(target + File.separator + "caches").mkdirs();
  1010. CreateObjUtil.createSoftConnection(path + File.separator + "capture", target + File.separator + "capture");
  1011. if(new File(path + File.separator + "caches" + File.separator + "images").exists()){
  1012. CreateObjUtil.createSoftConnection(path + File.separator + "caches" + File.separator + "images", target + File.separator + "caches" + File.separator + "images");
  1013. }
  1014. FileUtils.copyFile(path + File.separator + "data.json", target + File.separator+"data.json", true);
  1015. FileUtils.copyFile(path + File.separator + "project.json", target + File.separator+"project.json", true);
  1016. //data.json增加extras为执行重建算法
  1017. String project = FileUtils.readFile(target + File.separator+"project.json");
  1018. if(project != null){
  1019. JSONObject projectJson = JSONObject.parseObject(project);
  1020. projectJson.put("parent", projectJson.get("uuid"));
  1021. projectJson.put("uuid", UUID.randomUUID().toString());
  1022. projectJson.put("time", System.currentTimeMillis());
  1023. FileUtils.writeFile(path + File.separator + "project.json", projectJson.toString());
  1024. }
  1025. String data = FileUtils.readFile(target + File.separator+"data.json");
  1026. if(data != null){
  1027. JSONObject floorplanJson = new JSONObject();
  1028. floorplanJson.put("has_floor_ajk_json", true);
  1029. floorplanJson.put("has_vision_txt", true);
  1030. floorplanJson.put("has_floorplan_json", true);
  1031. JSONObject dataJson = JSONObject.parseObject(data);
  1032. dataJson.put("extras", floorplanJson);
  1033. //V5表示不需要生成high,low文件
  1034. dataJson.put("skybox_type", "SKYBOX_V8");
  1035. dataJson.put("split_type", "SPLIT_V10");
  1036. FileUtils.writeFile(target + File.separator+"data.json", new String(dataJson.toString().getBytes(), "UTF-8"));
  1037. }
  1038. //文件上传的位置可以自定义
  1039. log.info("画墙重建模型开始");
  1040. File targetFile = new File(target + File.separator + "extras" + File.separator + "floor_ajk.json");
  1041. if(!targetFile.getParentFile().exists()){
  1042. targetFile.getParentFile().mkdirs();
  1043. }
  1044. if(targetFile.exists()){
  1045. FileUtils.deleteFile(target + File.separator + "extras" + File.separator + "floor_ajk.json");
  1046. }
  1047. // 保存
  1048. FileUtils.writeFile(target + File.separator + "extras" + File.separator + "floorplan.json", new String(floorPlanJson.getBytes(), "UTF-8"));
  1049. FileUtils.writeFile(target + File.separator + "extras" + File.separator + "floor_ajk.json", new String(ajkJson.getBytes(), "UTF-8"));
  1050. FileUtils.writeFile(ConstantFilePath.SCENE_PATH+"data"+File.separator+"data"+num + File.separator + "floor_ajk.json", new String(ajkJson.getBytes(), "UTF-8"));
  1051. FileUtils.writeFile(target + File.separator + "extras" + File.separator + "vision.txt", new String(cameraJson.getBytes(), "UTF-8"));
  1052. FileUtils.writeFile(ConstantFilePath.SCENE_PATH+"data"+File.separator+"data"+num + File.separator + "camera.json", new String(cameraJson.getBytes(), "UTF-8"));
  1053. for(int i = 0; i < files.length; i ++){
  1054. File cadImg = new File(target + File.separator + "extras" + File.separator + "Floorplans/" + files[i].getOriginalFilename());
  1055. if(!cadImg.getParentFile().exists()){
  1056. cadImg.getParentFile().mkdirs();
  1057. }
  1058. if(cadImg.exists())
  1059. {
  1060. cadImg.delete();
  1061. }
  1062. files[i].transferTo(cadImg);
  1063. }
  1064. //下载封面图
  1065. FileUtils.downLoadFromUrl(scenePlusExt.getThumb() + "?t=" + System.currentTimeMillis(),
  1066. "Cover.png", target + File.separator + "extras" + File.separator + "CoverImage");
  1067. //转换成jpg
  1068. FileUtils.pngToJpg(target + File.separator + "extras" + File.separator + "CoverImage/Cover.png",
  1069. target + File.separator + "extras" + File.separator + "CoverImage/Cover.jpg");
  1070. //安居客算法运行
  1071. log.info("安居客算法:开始建模——"+num);
  1072. CreateObjUtil.build3dModel(target , "1");
  1073. String uploadJsonPath = target + File.separator + "results" + File.separator + "upload.json";
  1074. log.info("uploadJsonPath=" + uploadJsonPath);
  1075. //因为共享目录有延迟,这里循环检测算法是否计算完毕3次,每次隔五秒
  1076. boolean exist = ComputerUtil.checkComputeCompleted(uploadJsonPath, 3, 5000);
  1077. if(!exist){
  1078. throw new BusinessException(ErrorCode.FAILURE_CODE_5042);
  1079. }
  1080. String zipPath = target + File.separator + "results/" + num + ".zip";
  1081. new File(zipPath).delete();
  1082. FileUtils.zipFile(zipPath, target + File.separator + "results/ajk/");
  1083. // String command = "bash /opt/ossutil/gzip.sh " + zipPath.replace(".zip", "") + " " + target + File.separator + "results/ajk/";
  1084. // log.info("压缩文件:" + command);
  1085. // CreateObjUtil.callshell(command);
  1086. if(!new File(zipPath).exists()){
  1087. return ResultData.error(ErrorCode.FAILURE_CODE_5043);
  1088. }
  1089. String fileMD5 = FileMd5Util.getFileMD5(new File(zipPath));
  1090. uploadToOssUtil.upload(zipPath, "data_download/" + num + ".zip");
  1091. SceneDataDownload sceneDataDownload = sceneDataDownloadService.findBySceneNum(num);
  1092. if(sceneDataDownload == null){
  1093. sceneDataDownload = new SceneDataDownload();
  1094. sceneDataDownload.setNum(num);
  1095. sceneDataDownload.setDownloadPath(prefixAli + "data_download/" + num + ".zip");
  1096. sceneDataDownload.setFileMd5(fileMD5);
  1097. sceneDataDownloadService.save(sceneDataDownload);
  1098. return ResultData.ok();
  1099. }
  1100. sceneDataDownload.setFileMd5(fileMD5);
  1101. sceneDataDownload.setUpdateTime(Calendar.getInstance().getTime());
  1102. sceneDataDownloadService.updateById(sceneDataDownload);
  1103. }catch (Exception e){
  1104. log.error("画墙重建模型失败...", e);
  1105. throw new BusinessException(ErrorCode.FAILURE_CODE_5039);
  1106. }finally {
  1107. redisLockUtil.unlockLua(lockKey);
  1108. }
  1109. return ResultData.ok();
  1110. }
  1111. @Override
  1112. public ResultData checkKey(SceneCheckKeyParamVO param) throws Exception {
  1113. String sceneJson = redisUtil.get(String.format(RedisKey.SCENE_JSON, param.getNum()));
  1114. SceneJsonBean sceneJsonBean = JSON.parseObject(sceneJson, SceneJsonBean.class);
  1115. if(!param.getPassword().equals(sceneJsonBean.getScenePassword())){
  1116. throw new BusinessException(ErrorCode.FAILURE_CODE_5021);
  1117. }
  1118. return ResultData.ok();
  1119. }
  1120. private void updateBoxVideos(SceneEditInfo sceneEditInfo, Long scenePlusId, String boxVideos){
  1121. if(Objects.isNull(sceneEditInfo)){
  1122. sceneEditInfo = new SceneEditInfo();
  1123. sceneEditInfo.setScenePlusId(scenePlusId);
  1124. sceneEditInfo.setBoxVideos(boxVideos);
  1125. this.save(sceneEditInfo);
  1126. }else{
  1127. this.update(new UpdateWrapper<SceneEditInfo>()
  1128. .setSql("version = version + 1")
  1129. .set("box_videos", boxVideos)
  1130. .eq("id", sceneEditInfo.getId()));
  1131. }
  1132. }
  1133. private String createBoxVideos(String sid, JSONObject boxVideo, SceneEditInfo sceneEditInfo, int type){
  1134. String boxVideos = null;
  1135. if(sceneEditInfo != null){
  1136. boxVideos = sceneEditInfo.getBoxVideos();
  1137. }
  1138. JSONArray boxVideosJson = null;
  1139. if (StrUtil.isNotEmpty(boxVideos)) {
  1140. boxVideosJson = JSONArray.parseArray(boxVideos);
  1141. }else {
  1142. boxVideosJson = new JSONArray();
  1143. }
  1144. String result = null;
  1145. //删除
  1146. if(type == OperationType.DELETE.code()){
  1147. if(boxVideosJson.size() == 0)
  1148. return null;
  1149. for(int i=0;i<boxVideosJson.size();++i){
  1150. JSONObject ele = boxVideosJson.getJSONObject(i);
  1151. if(ele.getString("sid").equals(sid)){
  1152. boxVideosJson.remove(i);
  1153. }
  1154. }
  1155. }else{
  1156. // TODO: 2022/3/1 目前需求只保留一个,所以这里先注释掉,sid不同时,不追加
  1157. //更新
  1158. // boolean exist = false;
  1159. // for(int i=0;i<boxVideosJson.size();++i){
  1160. // JSONObject ele = boxVideosJson.getJSONObject(i);
  1161. // if(ele.getString("sid").equals(sid)){
  1162. // boxVideosJson.set(i, boxVideo);
  1163. // exist = true;
  1164. // }
  1165. // }
  1166. // //新增
  1167. // if(!exist){
  1168. // boxVideosJson.add(boxVideo);
  1169. // }
  1170. boxVideosJson.clear();
  1171. boxVideosJson.add(boxVideo);
  1172. }
  1173. if(boxVideosJson.size() != 0){
  1174. result = boxVideosJson.toJSONString();
  1175. }
  1176. return result;
  1177. }
  1178. private void transferToFlv(String num, String fileName) throws Exception {
  1179. String userEditPath = String.format(UploadFilePath.USER_EDIT_PATH, num);
  1180. String localImagesPath = String.format(ConstantFilePath.IMAGESBUFFER_FORMAT, num);
  1181. String localFilePath = localImagesPath + fileName;
  1182. File targetFile = new File(localImagesPath);
  1183. if (!targetFile.exists()){
  1184. targetFile.mkdirs();
  1185. }
  1186. targetFile = new File(localFilePath);
  1187. if (targetFile.exists()){
  1188. FileUtils.deleteFile(localFilePath);
  1189. }
  1190. //从用户编辑目录中下载视频到本地
  1191. if (!StorageType.LOCAL.code().equals(this.type)) {
  1192. String filePath = userEditPath + fileName;
  1193. String imageUrl = ossUrlPrefix + filePath + "?t=" + System.currentTimeMillis();
  1194. FileUtils.downLoadFromUrl(imageUrl, fileName, localImagesPath);
  1195. }
  1196. //视频格式转换
  1197. CreateObjUtil.mp4ToFlv(localFilePath, localFilePath.replace("mp4", "flv"));
  1198. //上传
  1199. String flvFileName = fileName.replace("mp4", "flv");
  1200. uploadToOssUtil.upload(localFilePath, userEditPath + fileName);
  1201. uploadToOssUtil.upload(localFilePath.replace("mp4", "flv"), userEditPath+flvFileName);
  1202. }
  1203. }