BuildScenePostServiceImpl.java 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. package com.fdkankan.scene.service.impl;
  2. import static com.fdkankan.push.PushMessageConfig.ANDROID_KEY;
  3. import static com.fdkankan.push.PushMessageConfig.ANDROID_KEY_Z;
  4. import static com.fdkankan.push.PushMessageConfig.ANDROID_SECRET;
  5. import static com.fdkankan.push.PushMessageConfig.ANDROID_SECRET_Z;
  6. import static com.fdkankan.push.PushMessageConfig.IOS_KEY;
  7. import static com.fdkankan.push.PushMessageConfig.IOS_KEY_Z;
  8. import static com.fdkankan.push.PushMessageConfig.IOS_SECRET;
  9. import static com.fdkankan.push.PushMessageConfig.IOS_SECRET_Z;
  10. import cn.hutool.core.bean.BeanUtil;
  11. import cn.hutool.core.collection.CollUtil;
  12. import cn.hutool.core.util.StrUtil;
  13. import com.alibaba.fastjson.JSON;
  14. import com.alibaba.fastjson.JSONArray;
  15. import com.alibaba.fastjson.JSONObject;
  16. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  17. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  18. import com.fdkankan.common.constant.BuildStatus;
  19. import com.fdkankan.common.constant.ConstantFilePath;
  20. import com.fdkankan.common.constant.ConstantUrl;
  21. import com.fdkankan.common.constant.ExpiredStatus;
  22. import com.fdkankan.common.constant.PayStatus;
  23. import com.fdkankan.common.constant.SceneFrom;
  24. import com.fdkankan.common.constant.SceneResolution;
  25. import com.fdkankan.common.constant.SceneSource;
  26. import com.fdkankan.common.constant.SceneStatus;
  27. import com.fdkankan.common.constant.ServerCode;
  28. import com.fdkankan.common.constant.UploadFilePath;
  29. import com.fdkankan.common.response.ResultData;
  30. import com.fdkankan.common.util.FileUtil;
  31. import com.fdkankan.common.util.FileUtils;
  32. import com.fdkankan.common.util.MatrixToImageWriterUtil;
  33. import com.fdkankan.fyun.constant.StorageType;
  34. import com.fdkankan.fyun.oss.UploadToOssUtil;
  35. import com.fdkankan.platform.api.dto.CameraDetail;
  36. import com.fdkankan.platform.api.dto.UserIncrement;
  37. import com.fdkankan.platform.api.feign.PlatformGoodsClient;
  38. import com.fdkankan.platform.api.feign.PlatformUserClient;
  39. import com.fdkankan.push.PushMessageConfig;
  40. import com.fdkankan.push.PushMsgUtil;
  41. import com.fdkankan.rabbitmq.bean.BuildSceneResultMqMessage;
  42. import com.fdkankan.redis.constant.RedisKey;
  43. import com.fdkankan.redis.util.RedisUtil;
  44. import com.fdkankan.scene.bean.SceneJsonBean;
  45. import com.fdkankan.scene.entity.SceneEditControls;
  46. import com.fdkankan.scene.entity.SceneEditInfo;
  47. import com.fdkankan.scene.entity.SceneEditInfoExt;
  48. import com.fdkankan.scene.entity.SceneFileBuild;
  49. import com.fdkankan.scene.entity.ScenePlus;
  50. import com.fdkankan.scene.entity.ScenePlusExt;
  51. import com.fdkankan.scene.entity.ScenePro;
  52. import com.fdkankan.scene.entity.SceneProExt;
  53. import com.fdkankan.scene.service.IBuildSceneDTService;
  54. import com.fdkankan.scene.service.IBuildScenePostService;
  55. import com.fdkankan.scene.service.IBuildScenePreService;
  56. import com.fdkankan.scene.service.ISceneEditControlsService;
  57. import com.fdkankan.scene.service.ISceneEditInfoExtService;
  58. import com.fdkankan.scene.service.ISceneEditInfoService;
  59. import com.fdkankan.scene.service.ISceneFileBuildService;
  60. import com.fdkankan.scene.service.IScenePlusExtService;
  61. import com.fdkankan.scene.service.IScenePlusService;
  62. import com.fdkankan.scene.service.ISceneProExtService;
  63. import com.fdkankan.scene.service.ISceneProService;
  64. import com.fdkankan.scene.service.ISceneService;
  65. import com.fdkankan.scene.vo.SceneEditControlsVO;
  66. import com.sun.org.apache.bcel.internal.generic.RETURN;
  67. import java.io.File;
  68. import java.io.IOException;
  69. import java.util.Calendar;
  70. import java.util.HashMap;
  71. import java.util.List;
  72. import java.util.Map;
  73. import java.util.Map.Entry;
  74. import java.util.Objects;
  75. import java.util.stream.Collectors;
  76. import lombok.extern.slf4j.Slf4j;
  77. import org.apache.commons.lang3.StringUtils;
  78. import org.springframework.beans.factory.annotation.Autowired;
  79. import org.springframework.beans.factory.annotation.Value;
  80. import org.springframework.stereotype.Service;
  81. /**
  82. * <p>
  83. * TODO
  84. * </p>
  85. *
  86. * @author dengsixing
  87. * @since 2022/4/20
  88. **/
  89. @Slf4j
  90. @Service
  91. public class BuildScenePostServiceImpl implements IBuildScenePostService {
  92. @Value("${main.url}")
  93. private String mainUrl;
  94. @Value("${scene.pro.url}")
  95. private String sceneProUrl;
  96. @Value("${scene.pro.new.url}")
  97. private String sceneProNewUrl;
  98. @Value("${upload.type}")
  99. private String ossType;
  100. @Value("${oss.prefix.ali}")
  101. private String prefixAli;
  102. @Autowired
  103. ISceneService sceneService;
  104. @Autowired
  105. ISceneFileBuildService sceneFileBuildService;
  106. @Autowired
  107. ISceneProService sceneProService;
  108. @Autowired
  109. ISceneProExtService sceneProExtService;
  110. @Autowired
  111. PlatformGoodsClient platformGoodsClient;
  112. @Autowired
  113. PlatformUserClient platformUserClient;
  114. @Autowired
  115. ISceneEditInfoService sceneEditInfoService;
  116. @Autowired
  117. ISceneEditControlsService sceneEditControlsService;
  118. @Autowired
  119. UploadToOssUtil uploadToOssUtil;
  120. // @Autowired
  121. // IBuildScenePreService buildScenePreService;
  122. @Autowired
  123. RedisUtil redisUtil;
  124. @Autowired
  125. IScenePlusService scenePlusService;
  126. @Autowired
  127. IScenePlusExtService scenePlusExtService;
  128. @Autowired
  129. IBuildSceneDTService buildSceneDTService;
  130. @Autowired
  131. ISceneEditInfoExtService sceneEditInfoExtService;
  132. @Override
  133. public void buildScenePost(BuildSceneResultMqMessage message) {
  134. Integer cameraType = Integer.parseInt(message.getCameraType());
  135. String sceneCode = message.getSceneCode();
  136. String fileId = message.getFileId();
  137. Integer payStatus = message.getPayStatus();
  138. Long computeTime = message.getComputeTime();
  139. Boolean buildSuccess = message.getBuildSuccess();
  140. Integer videoVersion = message.getVideoVersion();
  141. Map<String, String> uploadFiles = message.getUploadFiles();
  142. String path = message.getPath();
  143. Integer pushChannel = message.getPushChannel();
  144. String pushToken = message.getPushToken();
  145. String prefix = message.getPrefix();
  146. try {
  147. if(!buildSuccess){//建模失败,修改状态为失败状态
  148. log.info("建模失败,修改状态为失败状态");
  149. // TODO: 2022/3/21 plus版本稳定后删除------------------------start
  150. sceneService.updateStatus(sceneCode, SceneStatus.FAILD.code());
  151. sceneProService.updateStatus(sceneCode, SceneStatus.FAILD.code());
  152. // TODO: 2022/3/21 plus版本稳定后删除------------------------send
  153. scenePlusService.update(
  154. new LambdaUpdateWrapper<ScenePlus>()
  155. .set(ScenePlus::getSceneStatus, SceneStatus.FAILD.code())
  156. .eq(ScenePlus::getNum, sceneCode));
  157. SceneFileBuild sceneFileBuild = sceneFileBuildService.findByFileId(fileId);
  158. if(Objects.nonNull(sceneFileBuild)){
  159. sceneFileBuild.setBuildStatus(BuildStatus.fail.code());
  160. sceneFileBuildService.updateById(sceneFileBuild);
  161. }
  162. return;
  163. }
  164. //建模成功走以下逻辑
  165. log.info("cameraType:{}",cameraType);
  166. //计算场景消耗磁盘空间
  167. long space = this.calUseSpace(uploadFiles);
  168. if(cameraType < 3){
  169. // TODO: 2022/3/21 plus版本稳定后删除------------------------start
  170. sceneService.updateTime(sceneCode, space, payStatus);
  171. // TODO: 2022/3/21 plus版本稳定后删除------------------------send
  172. this.updateDb4Sm(sceneCode, space);
  173. uploadToOssUtil.uploadMulFiles(uploadFiles);
  174. return;
  175. }
  176. //读取计算结果文件生成videosJson
  177. JSONObject videosJson = this.getVideosJson(path, videoVersion, sceneCode, cameraType);
  178. //写入数据库
  179. // TODO: 2022/3/21 plus版本稳定后删除------------------------start
  180. ScenePro scenePro = this.updateDbPro(sceneCode, space, payStatus, videosJson.toJSONString(), computeTime, fileId);
  181. SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  182. // TODO: 2022/3/21 plus版本稳定后删除------------------------send
  183. ScenePlus scenePlus = this.updateDbPlus(sceneCode, space, payStatus, videosJson.toJSONString(), computeTime, fileId);
  184. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  185. //变更容量,arrearCap为true时,代表容量不足
  186. boolean arrearCap = this.updateSpace(scenePlus.getCameraId(), space, scenePlus.getId(), scenePro.getId());
  187. //如果相机容量不足,需要把场景的paystatus改为容量不足状态
  188. this.sealScene(arrearCap, scenePlus.getId(), scenePro.getId());
  189. Object[] editInfoArr = this.updateEditInfo(scenePro, scenePlus);
  190. SceneEditInfo sceneEditInfo = (SceneEditInfo)editInfoArr[0];
  191. SceneEditInfoExt sceneEditInfoExt = (SceneEditInfoExt)editInfoArr[1];
  192. SceneEditControls sceneEditControls = (SceneEditControls)editInfoArr[2];
  193. String sceneUrl = mainUrl + "/" + sceneProNewUrl;
  194. String outPathZh = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+sceneCode+".png";
  195. String outPathEn = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+sceneCode+"_en.png";
  196. //生成新的分享的二维码-中文版本
  197. MatrixToImageWriterUtil.createQRCode(sceneUrl + sceneCode, outPathZh, false,null);
  198. //生成新的分享的二维码-英文版本
  199. MatrixToImageWriterUtil.createQRCode(sceneUrl + sceneCode + "&lang=en", outPathEn, false, null);
  200. //上传二维码
  201. uploadToOssUtil.upload(outPathZh, String.format(UploadFilePath.DOWNLOADS_QRCODE, sceneCode) + sceneCode + ".png");
  202. uploadToOssUtil.upload(outPathEn, String.format(UploadFilePath.DOWNLOADS_QRCODE, sceneCode) + sceneCode + "_en.png");
  203. Map<String, String> newUploadFiles = this.uploadFileMapHandler(sceneCode, cameraType, uploadFiles);
  204. //上传计算结果中的caches/images和caches/videos
  205. this.uploadCaches(sceneCode, uploadFiles, path);
  206. //上传文件
  207. uploadToOssUtil.uploadMulFiles(newUploadFiles);
  208. // TODO: 2022/3/11 同时上传一份到旧版本的目录,用于过渡期使用,待重构版本稳定后删除
  209. uploadToOssUtil.uploadMulFiles(uploadFiles);
  210. //拷贝部分文件到编辑目录,用于用户编辑
  211. this.copyToEditDir(sceneCode);
  212. //计算成功,发短信
  213. this.sendSms(pushChannel,pushToken, cameraType, scenePro.getSceneName(), scenePro.getWebSite());
  214. // //上传日志文件
  215. // buildScenePreService.uploadLogFile(sceneCode, sceneProExt.getDataSource());
  216. //写scene.json
  217. this.writeSceneJson(sceneCode, videosJson,
  218. sceneEditInfo, sceneEditInfoExt, sceneEditControls, scenePlus, scenePlusExt, arrearCap);
  219. log.info("场景计算结果处理结束,场景码:{}", message.getSceneCode());
  220. }catch (Exception e){
  221. log.error("场景计算结果处理出错", e);
  222. //发送钉钉
  223. String serverPath = message.getPath().substring(0,message.getPath().lastIndexOf("/")+1).concat(prefix);
  224. // buildSceneDTService.handFail("场景计算结果处理出错", serverPath, sceneCode, BuildScenePreServiceImpl.hostName, BuildScenePreServiceImpl.contentExt);
  225. }
  226. }
  227. private void writeSceneJson(String num, JSONObject videosJson, SceneEditInfo sceneEditInfo,SceneEditInfoExt sceneEditInfoExt,
  228. SceneEditControls sceneEditControls, ScenePlus scenePlus, ScenePlusExt scenePlusExt, boolean arrearCap) throws Exception{
  229. String sceneJsonKey = String.format(UploadFilePath.DATA_VIEW_PATH, num) + "scene.json";
  230. boolean exist = uploadToOssUtil.existKey(sceneJsonKey);
  231. //如果云端没有scene.json文件,生成一份
  232. if(!exist){
  233. SceneJsonBean sceneJson = new SceneJsonBean();
  234. BeanUtil.copyProperties(sceneEditInfoExt, sceneJson);
  235. BeanUtil.copyProperties(sceneEditInfo, sceneJson);
  236. SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class);
  237. sceneJson.setControls(sceneEditControlsVO);
  238. sceneJson.setNum(num);
  239. sceneJson.setCreateTime(scenePlus.getCreateTime());
  240. sceneJson.setSceneResolution(scenePlusExt.getSceneResolution());
  241. sceneJson.setSceneFrom(scenePlusExt.getSceneFrom());
  242. sceneJson.setVideos(JSON.toJSONString(videosJson));
  243. if(arrearCap) {
  244. sceneJson.setPayStatus(PayStatus.NO_CAPACITY.code());
  245. }
  246. String sceneJsonStr = JSON.toJSONString(sceneJson);
  247. //上传sceneJson文件
  248. uploadToOssUtil.upload(sceneJsonStr.getBytes(), sceneJsonKey);
  249. //scenejson写入缓存
  250. redisUtil.set(String.format(RedisKey.SCENE_JSON, num), sceneJsonStr);
  251. }
  252. // TODO: 2022/4/21 plus版本稳定后删除------------------------------------------start
  253. String localSceneJsonPath = String.format(ConstantFilePath.SCENE_PATH_FORMAT, num);
  254. String localSceneJson = FileUtils.readFile(localSceneJsonPath);
  255. if(StrUtil.isNotEmpty(localSceneJson)){
  256. JSONObject sceneJsonObject = JSON.parseObject(localSceneJson);
  257. sceneJsonObject.put("videos", videosJson);
  258. if(arrearCap) {
  259. sceneJsonObject.put("payStatus", PayStatus.NO_CAPACITY.code());
  260. }
  261. FileUtils.writeFile(localSceneJsonPath, sceneJsonObject.toJSONString());
  262. }
  263. // TODO: 2022/4/21 plus版本稳定后删除------------------------------------------end
  264. }
  265. private void sendSms(Integer pushChannel, String pushToken, int cameraType, String sceneName, String webSite){
  266. log.info("推送消息,渠道是 {}, 手机token是 {}", pushChannel, pushToken);
  267. if(Objects.isNull(pushChannel) && StrUtil.isBlank(pushToken))
  268. return;
  269. try{
  270. if(StorageType.AWS.code().equals(ossType)){
  271. PushMsgUtil
  272. .googlePushMsg(ConstantFilePath.BASE_PATH + "/refreshToken.json", pushToken,
  273. sceneName + "计算完成", "您上传的" + sceneName + "计算完成,点击查看", webSite);
  274. return;
  275. }
  276. PushMessageConfig demo = null;
  277. if(pushChannel == 0){
  278. if(cameraType == 10 || cameraType == 13){
  279. //ios
  280. log.info("IOS_KEY:{}, IOS_SECRET:{}", IOS_KEY_Z, IOS_SECRET_Z);
  281. demo = new PushMessageConfig(IOS_KEY_Z, IOS_SECRET_Z);
  282. demo.sendIOSUnicast(pushToken, "四维看看Minion",
  283. sceneName + "计算完成", "您上传的" + sceneName + "计算完成,点击查看", webSite);
  284. }else {
  285. //ios
  286. log.info("IOS_KEY:{}, IOS_SECRET:{}", IOS_KEY, IOS_SECRET);
  287. demo = new PushMessageConfig(IOS_KEY, IOS_SECRET);
  288. demo.sendIOSUnicast(pushToken, "四维看看Pro",
  289. sceneName + "计算完成", "您上传的" + sceneName + "计算完成,点击查看", webSite);
  290. }
  291. }else {
  292. if(cameraType == 10 || cameraType == 13){
  293. //ios
  294. //安卓
  295. log.info("ANDROID_KEY:{}, ANDROID_SECRET:{}", ANDROID_KEY_Z, ANDROID_SECRET_Z);
  296. demo = new PushMessageConfig(ANDROID_KEY_Z, ANDROID_SECRET_Z);
  297. demo.sendAndroidUnicast2(pushToken, "四维看看Minion",
  298. sceneName + "计算完成", "您上传的" + sceneName + "计算完成,点击查看", webSite);
  299. }else {
  300. //安卓
  301. log.info("ANDROID_KEY:{}, ANDROID_SECRET:{}", ANDROID_KEY, ANDROID_SECRET);
  302. demo = new PushMessageConfig(ANDROID_KEY, ANDROID_SECRET);
  303. demo.sendAndroidUnicast(pushToken, "四维看看Pro",
  304. sceneName + "计算完成", "您上传的" + sceneName + "计算完成,点击查看", webSite);
  305. }
  306. }
  307. log.info("消息推送结束!");
  308. }catch (Exception e){
  309. log.error("推送消息失败:", e);
  310. }
  311. }
  312. private void copyToEditDir(String num) throws IOException {
  313. String editImagesPath = String.format(UploadFilePath.IMG_EDIT_PATH, num);
  314. String viewImagesPath = String.format(UploadFilePath.IMG_VIEW_PATH, num);
  315. String editDataPath = String.format(UploadFilePath.DATA_EDIT_PATH, num);
  316. String viewDataPath = String.format(UploadFilePath.DATA_VIEW_PATH, num);
  317. Map<String, String> map = new HashMap<>();
  318. map.put(editImagesPath + "vision.modeldata", viewImagesPath + "vision.modeldata");
  319. map.put(editImagesPath + "vision2.modeldata", viewImagesPath + "vision2.modeldata");
  320. map.put(editDataPath + "floorplan_cad.json", viewDataPath + "floorplan_cad.json");
  321. for (Entry<String, String> entry : map.entrySet()) {
  322. uploadToOssUtil.copyFiles(entry.getValue(), entry.getKey());
  323. }
  324. }
  325. private void uploadCaches(String num, Map<String,String> map, String path){
  326. String sceneNumPAth = String.format("scene/%s", num);
  327. List<String> imagesList = FileUtil.getFileList(path + "/caches/images");
  328. log.info("caches/images_path:{}", path + "/caches/images");
  329. if(CollUtil.isNotEmpty(imagesList)){
  330. log.info("上传的caches/images文件列表:{}", imagesList.toString());
  331. imagesList.stream().forEach(str -> {
  332. if(str.endsWith(".jpg")){
  333. map.put(str, str.replace(path, sceneNumPAth));
  334. }
  335. });
  336. }
  337. List<String> videosList = FileUtil.getFileList(path + "/caches/videos");
  338. if(CollUtil.isNotEmpty(videosList)){
  339. log.info("上传的caches/videos文件列表:{}", videosList.toString());
  340. videosList.stream().forEach(str -> map.put(str, str.replace(path, sceneNumPAth)));
  341. }
  342. }
  343. private Map<String, String> uploadFileMapHandler(String num, int cameraType, Map<String, String> map) {
  344. if(CollUtil.isEmpty(map)){
  345. return null;
  346. }
  347. String dataPath = String.format("data/data%s/", num);
  348. String imagePath = String.format("images/images%s/", num);
  349. String videoPath = String.format("video/video%s/", num);
  350. String dataViewPath = String.format(UploadFilePath.DATA_VIEW_PATH, num);
  351. String imgViewPath = String.format(UploadFilePath.IMG_VIEW_PATH, num);
  352. String videosViewPath = String.format(UploadFilePath.VIDEOS_VIEW_PATH, num);
  353. Map<String, String> newMap = new HashMap<>();
  354. for (String key : map.keySet()) {
  355. String value = map.get(key);
  356. if(value.contains(dataPath)){
  357. newMap.put(key, value.replace(dataPath, dataViewPath));
  358. }
  359. if(value.contains(imagePath)){
  360. newMap.put(key, value.replace(imagePath, imgViewPath));
  361. }
  362. if(value.contains(videoPath)){
  363. newMap.put(key, value.replace(videoPath, videosViewPath));
  364. }
  365. }
  366. return newMap;
  367. }
  368. private JSONObject getVideosJson(String path, Integer videoVersion, String projectNum, int cameraType) throws Exception {
  369. //读取videos_hdr_param.json, 保存点位视频的value
  370. Map<String, Object> videoMap = new HashMap<>();
  371. String videosHdr = FileUtils.readFile(path + File.separator + "results/videos/videos_hdr_param.json");
  372. JSONArray videoArray = null;
  373. if(StringUtils.isNotEmpty(videosHdr)){
  374. videoArray = JSONObject.parseObject(videosHdr).getJSONArray("hdr_param");
  375. }
  376. if(videoArray != null){
  377. for(int i = 0, len = videoArray.size(); i < len; i++) {
  378. videoMap.put(videoArray.getJSONObject(i).getString("name"), videoArray.getJSONObject(i).getString("value"));
  379. if(videoArray.getJSONObject(i).containsKey("fov")){
  380. videoMap.put(videoArray.getJSONObject(i).getString("name") + "_fov", videoArray.getJSONObject(i).getString("fov"));
  381. }
  382. }
  383. }
  384. //获取upload中的video视频名称
  385. String uploadData = FileUtils.readFile(path + File.separator + "results" +File.separator+"upload.json");
  386. JSONObject uploadJson = null;
  387. JSONArray array = null;
  388. if(uploadData!=null) {
  389. uploadJson = JSONObject.parseObject(uploadData);
  390. array = uploadJson.getJSONArray("upload");
  391. }
  392. JSONObject fileJson = null;
  393. String fileName = "";
  394. //计算ts文件的大小,并拼接成json格式
  395. JSONArray jsonArray = new JSONArray();
  396. JSONObject videoJson = null;
  397. JSONObject videosJson = new JSONObject();
  398. long videoSize = 0L;
  399. for(int i = 0, len = array.size(); i < len; i++) {
  400. fileJson = array.getJSONObject(i);
  401. fileName = fileJson.getString("file");
  402. if(fileJson.getIntValue("clazz") == 11 && fileName.contains(".mp4") && !fileName.contains("-ios.mp4")){
  403. videoJson = new JSONObject();
  404. videoJson.put("id", fileName.substring(
  405. 0, fileName.lastIndexOf(".")).replace("videos/", ""));
  406. //如果ts文件存在,就计算ts大小
  407. if(new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).exists()){
  408. videoSize = new File(path + File.separator + "results" +File.separator+ fileName.replace(".mp4", ".ts")).length();
  409. videoJson.put("tsSize", videoSize);
  410. }
  411. if(videoMap.containsKey(videoJson.get("id"))){
  412. videoJson.put("value", videoMap.get(videoJson.get("id")));
  413. }
  414. if(videoMap.containsKey(videoJson.get("id") + "_fov")){
  415. videoJson.put("blend_fov", videoMap.get(videoJson.get("id") + "_fov"));
  416. }else {
  417. videoJson.put("blend_fov", 7);
  418. }
  419. jsonArray.add(videoJson);
  420. }
  421. }
  422. videosJson.put("data", jsonArray);
  423. if(Objects.nonNull(videoVersion) && videoVersion >= 4){
  424. videosJson.put("version", 3);
  425. if(StorageType.OSS.code().equals(ossType)){
  426. videosJson.put("upPath", prefixAli + "data/data" + projectNum + "/Up.xml");
  427. }
  428. if(StorageType.AWS.code().equals(ossType)){
  429. videosJson.put("upPath", ConstantUrl.PREFIX_AWS + "data/data" + projectNum + "/Up.xml");
  430. }
  431. if(cameraType == 13){
  432. //转台相机
  433. videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
  434. }
  435. }else {
  436. videosJson.put("version", 1);
  437. if("oss".equals(ossType)){
  438. videosJson.put("upPath", prefixAli + "data/data" + projectNum + "/Up2.xml");
  439. }
  440. if("aws".equals(ossType)){
  441. videosJson.put("upPath", ConstantUrl.PREFIX_AWS + "data/data" + projectNum + "/Up2.xml");
  442. }
  443. if(cameraType == 13){
  444. //转台相机
  445. videosJson.put("upPath", videosJson.getString("upPath").replace(".xml", ".txt"));
  446. }
  447. }
  448. if(cameraType == 5 || cameraType == 6){
  449. videosJson.put("version", 1);
  450. if("oss".equals(ossType)){
  451. videosJson.put("upPath", prefixAli + "data/data" + projectNum + "/stitch_params.txt");
  452. }
  453. if("aws".equals(ossType)){
  454. videosJson.put("upPath", ConstantUrl.PREFIX_AWS + "data/data" + projectNum + "/stitch_params.txt");
  455. }
  456. }
  457. return videosJson;
  458. }
  459. private Long calUseSpace(Map<String, String> uploadFile) {
  460. File spaceFile = null;
  461. long space = 0L;
  462. for (String key : uploadFile.keySet()) {
  463. spaceFile = new File(key);
  464. if(spaceFile.exists()){
  465. space += spaceFile.length();
  466. }
  467. }
  468. return space;
  469. }
  470. private void sealScene(boolean arrearCap, Long scenePlusId, Long sceneProId){
  471. if(!arrearCap){
  472. return;
  473. }
  474. scenePlusService.update(
  475. new LambdaUpdateWrapper<ScenePlus>()
  476. .set(ScenePlus::getPayStatus, PayStatus.NO_CAPACITY)
  477. .eq(ScenePlus::getId, scenePlusId));
  478. // TODO: 2022/3/22 plus版本稳定后删除---------------------------start
  479. sceneProService.update(
  480. new LambdaUpdateWrapper<ScenePro>()
  481. .set(ScenePro::getPayStatus, PayStatus.NO_CAPACITY)
  482. .eq(ScenePro::getId, sceneProId));
  483. // TODO: 2022/3/22 plus版本稳定后删除---------------------------end
  484. }
  485. /**
  486. * <p>
  487. 双目场景更新数据库
  488. * </p>
  489. * @author dengsixing
  490. * @date 2022/3/21
  491. * @param num
  492. * @param space
  493. **/
  494. private void updateDb4Sm(String num, long space){
  495. List<ScenePlus> ScenePlusList = scenePlusService.list(
  496. new LambdaQueryWrapper<ScenePlus>()
  497. .select(ScenePlus::getId)
  498. .eq(ScenePlus::getNum, num));
  499. if(CollUtil.isNotEmpty(ScenePlusList))
  500. return ;
  501. List<Long> sceneIds = ScenePlusList.stream().map(scene -> {
  502. return scene.getId();
  503. }).collect(Collectors.toList());
  504. //更新场景创建时间
  505. scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>()
  506. .in(ScenePlus::getId, sceneIds)
  507. .set(ScenePlus::getCreateTime, Calendar.getInstance().getTime())
  508. .set(ScenePlus::getSceneStatus, SceneStatus.NO_DISPLAY.code()));
  509. //更新使用容量
  510. scenePlusExtService.update(
  511. new LambdaUpdateWrapper<ScenePlusExt>()
  512. .in(ScenePlusExt::getPlusId, sceneIds)
  513. .set(ScenePlusExt::getSpace, space));
  514. }
  515. private ScenePro updateDbPro(String sceneCode, Long space, Integer payStatus, String videosJson, Long computeTime, String fileId) throws Exception{
  516. ScenePro scenePro = sceneProService.findBySceneNum(sceneCode);
  517. sceneProService.updateTime(sceneCode, space, payStatus, videosJson, computeTime);
  518. SceneFileBuild sceneFileBuild = sceneFileBuildService.findByFileId(fileId);
  519. if(sceneFileBuild != null){
  520. sceneFileBuild.setBuildStatus(BuildStatus.success.code());
  521. sceneFileBuildService.updateById(sceneFileBuild);
  522. }
  523. if(Objects.nonNull(scenePro)){
  524. SceneProExt sceneProExt = sceneProExtService.findBySceneProId(scenePro.getId());
  525. SceneSource sceneSource = SceneSource.get(sceneProExt.getSceneSource());
  526. switch (sceneSource){
  527. case BM:
  528. sceneProExt.setSceneResolution(SceneResolution.TILES_2K.code());
  529. sceneProExt.setSceneFrom(SceneFrom.PRO.code());
  530. break;
  531. case SM:
  532. sceneProExt.setSceneResolution(SceneResolution.TILES_1K.code());
  533. sceneProExt.setSceneFrom(SceneFrom.LITE.code());
  534. break;
  535. case ZT:
  536. sceneProExt.setSceneResolution(SceneResolution.TILES_4K.code());
  537. sceneProExt.setSceneFrom(SceneFrom.MINION.code());
  538. break;
  539. case JG:
  540. sceneProExt.setSceneResolution(SceneResolution.TILES_4K.code());
  541. sceneProExt.setSceneFrom(SceneFrom.LASER.code());
  542. break;
  543. }
  544. sceneProExtService.updateById(sceneProExt);
  545. }
  546. return scenePro;
  547. }
  548. private ScenePlus updateDbPlus(String num, Long space, Integer payStatus, String videosJson, Long computeTime, String fileId) throws Exception{
  549. ScenePlus scenePlus = scenePlusService.getScenePlusByNum(num);
  550. ScenePlusExt scenePlusExt = scenePlusExtService.getScenePlusExtByPlusId(scenePlus.getId());
  551. List<ScenePlus> list = scenePlusService.list(
  552. new LambdaQueryWrapper<ScenePlus>()
  553. .select(ScenePlus::getId)
  554. .eq(ScenePlus::getNum, num));
  555. if(scenePlus == null || CollUtil.isEmpty(list))
  556. return null;
  557. //修改场景状态 空间 支付状态 计算时间
  558. List<Long> scenePlusIds = list.stream().map(plus -> {
  559. return plus.getId();
  560. }).collect(Collectors.toList());
  561. scenePlusService.update(new LambdaUpdateWrapper<ScenePlus>()
  562. .in(ScenePlus::getId, scenePlusIds)
  563. .set(ScenePlus::getSceneStatus, SceneStatus.NO_DISPLAY.code())
  564. .set(ScenePlus::getCreateTime, Calendar.getInstance().getTime())
  565. .set(ScenePlus::getPayStatus, payStatus));
  566. scenePlusExtService.update(new LambdaUpdateWrapper<ScenePlusExt>()
  567. .in(ScenePlusExt::getPlusId, scenePlusIds)
  568. .set(ScenePlusExt::getSpace, space)
  569. .set(ScenePlusExt::getComputeTime, computeTime)
  570. .set(ScenePlusExt::getVideos, videosJson));
  571. SceneFileBuild sceneFileBuild = sceneFileBuildService.findByFileId(fileId);
  572. if(sceneFileBuild != null){
  573. sceneFileBuild.setBuildStatus(BuildStatus.success.code());
  574. sceneFileBuildService.updateById(sceneFileBuild);
  575. }
  576. SceneSource sceneSource = SceneSource.get(scenePlus.getSceneSource());
  577. switch (sceneSource){
  578. case BM:
  579. scenePlusExt.setSceneResolution(SceneResolution.TILES_2K.code());
  580. scenePlusExt.setSceneFrom(SceneFrom.PRO.code());
  581. break;
  582. case SM:
  583. scenePlusExt.setSceneResolution(SceneResolution.TILES_1K.code());
  584. scenePlusExt.setSceneFrom(SceneFrom.LITE.code());
  585. break;
  586. case ZT:
  587. scenePlusExt.setSceneResolution(SceneResolution.TILES_4K.code());
  588. scenePlusExt.setSceneFrom(SceneFrom.MINION.code());
  589. break;
  590. case JG:
  591. scenePlusExt.setSceneResolution(SceneResolution.TILES_4K.code());
  592. scenePlusExt.setSceneFrom(SceneFrom.LASER.code());
  593. break;
  594. }
  595. scenePlusExtService.updateById(scenePlusExt);
  596. return scenePlus;
  597. }
  598. private boolean updateSpace(Long cameraId, Long space, Long scenePlusId, Long sceneProId) throws Exception{
  599. //更新相机使用用量
  600. ResultData<CameraDetail> resultData = platformGoodsClient.updateCameraDetailByCameraIdAndSpace(cameraId, space);
  601. if(resultData.getCode() != ServerCode.SUCCESS.code()){
  602. log.error("调用platform服务updateCameraDetailByCameraIdAndSpace失败,参数{},{}" ,cameraId, space);
  603. throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message());
  604. }
  605. CameraDetail cameraDetail = JSONObject.parseObject(JSONObject.toJSONString(resultData.getData()), CameraDetail.class);
  606. ResultData<UserIncrement> resultData1 = platformUserClient.getUserIncrementByCameraId(cameraId);
  607. if(resultData1.getCode() != ServerCode.SUCCESS.code()){
  608. log.error("调用platform服务getUserIncrementByCameraId失败,参数{}" ,cameraId);
  609. throw new Exception(ServerCode.FEIGN_REQUEST_FAILD.message());
  610. }
  611. if( resultData1.getData() == null || resultData1.getData().getIsExpired() == ExpiredStatus.Expired.code()) {
  612. // 新上传的场景,如果总容量小于使用容量,则该大场景保留在临时存储空间30天
  613. if (cameraDetail.getTotalSpace().compareTo(cameraDetail.getUsedSpace()) == -1){
  614. return true;
  615. }
  616. }
  617. return false;
  618. }
  619. private Object[] updateEditInfo(ScenePro scenePro, ScenePlus scenePlus){
  620. SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
  621. SceneEditControls sceneEditControls = null;
  622. SceneEditInfoExt sceneEditInfoExt = null;
  623. if(sceneEditInfo == null){
  624. sceneEditInfo = new SceneEditInfo();
  625. sceneEditInfo.setScenePlusId(scenePlus.getId());
  626. sceneEditInfo.setSceneProId(scenePro.getId());
  627. sceneEditInfo.setDescription(scenePlus.getDescription());
  628. sceneEditInfo.setTitle(scenePlus.getTitle());
  629. sceneEditInfoService.save(sceneEditInfo);
  630. }else{
  631. sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfo.getId());
  632. sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfo.getId());
  633. }
  634. if(sceneEditControls == null){
  635. sceneEditControls = new SceneEditControls();
  636. sceneEditControls.setEditInfoId(sceneEditInfo.getId());
  637. sceneEditControlsService.save(sceneEditControls);
  638. }
  639. if(sceneEditInfoExt == null){
  640. sceneEditInfoExt = new SceneEditInfoExt();
  641. sceneEditInfoExt.setScenePlusId(scenePlus.getId());
  642. sceneEditInfoExt.setSceneProId(scenePro.getId());
  643. sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId());
  644. sceneEditInfoExtService.save(sceneEditInfoExt);
  645. }
  646. return new Object[]{sceneEditInfo, sceneEditInfoExt, sceneEditControls};
  647. }
  648. }