SceneProServiceImpl.java 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905
  1. package com.fdkankan.ucenter.service.impl;
  2. import com.alibaba.fastjson.JSONObject;
  3. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  4. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  5. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  6. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  7. import com.fdkankan.common.constant.SceneConstant;
  8. import com.fdkankan.common.constant.SceneKind;
  9. import com.fdkankan.common.exception.BusinessException;
  10. import com.fdkankan.common.util.*;
  11. import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
  12. import com.fdkankan.rabbitmq.util.RabbitMqProducer;
  13. import com.fdkankan.redis.constant.RedisKey;
  14. import com.fdkankan.ucenter.common.PageInfo;
  15. import com.fdkankan.fyun.face.FYunFileServiceInterface;
  16. import com.fdkankan.image.MatrixToImageWriterUtil;
  17. import com.fdkankan.ucenter.common.constants.ConstantFilePath;
  18. import com.fdkankan.ucenter.common.constants.NacosProperty;
  19. import com.fdkankan.ucenter.common.constants.UploadFilePath;
  20. import com.fdkankan.redis.util.RedisUtil;
  21. import com.fdkankan.ucenter.common.RedisKeyUtil;
  22. import com.fdkankan.ucenter.common.utils.ShellUtil;
  23. import com.fdkankan.ucenter.constant.CameraConstant;
  24. import com.fdkankan.ucenter.constant.LoginConstant;
  25. import com.fdkankan.ucenter.entity.*;
  26. import com.fdkankan.ucenter.httpClient.service.LaserService;
  27. import com.fdkankan.ucenter.mapper.ISceneProMapper;
  28. import com.fdkankan.ucenter.mapper.ISceneUpgradeMapper;
  29. import com.fdkankan.ucenter.service.*;
  30. import com.fdkankan.ucenter.util.DateUserUtil;
  31. import com.fdkankan.ucenter.vo.request.SceneParam;
  32. import com.fdkankan.ucenter.vo.response.GroupByCount;
  33. import com.fdkankan.ucenter.vo.response.SceneInfoVo;
  34. import com.fdkankan.ucenter.vo.response.SceneNumVo;
  35. import com.fdkankan.ucenter.vo.response.SceneVo;
  36. import java.io.File;
  37. import java.util.ArrayList;
  38. import java.util.Arrays;
  39. import java.util.Collections;
  40. import java.util.Date;
  41. import java.util.HashMap;
  42. import java.util.List;
  43. import java.util.Map;
  44. import java.util.concurrent.CompletableFuture;
  45. import java.util.stream.Collectors;
  46. import javax.annotation.Resource;
  47. import lombok.extern.slf4j.Slf4j;
  48. import org.apache.commons.lang3.StringUtils;
  49. import org.springframework.beans.BeanUtils;
  50. import org.springframework.beans.factory.annotation.Autowired;
  51. import org.springframework.beans.factory.annotation.Value;
  52. import org.springframework.stereotype.Service;
  53. import org.springframework.util.ObjectUtils;
  54. /**
  55. * <p>
  56. * pro场景表 服务实现类
  57. * </p>
  58. *
  59. * @author
  60. * @since 2022-07-04
  61. */
  62. @Service
  63. @Slf4j
  64. public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro> implements ISceneProService {
  65. @Autowired
  66. private IUserService userService;
  67. @Autowired
  68. private ISceneCooperationService sceneCooperationService;
  69. @Autowired
  70. private IScenePlusService scenePlusService;
  71. @Autowired
  72. private ISceneService sceneService;
  73. @Autowired
  74. private LaserService fdkkLaserService;
  75. @Resource
  76. private FYunFileServiceInterface fYunFileService;
  77. @Autowired
  78. private IUserIncrementService userIncrementService;
  79. @Autowired
  80. private IScenePlusExtService scenePlusExtService;
  81. @Autowired
  82. private ICameraDetailService cameraDetailService;
  83. @Autowired
  84. private IFolderService folderService;
  85. @Autowired
  86. private IFolderSceneService folderSceneService;
  87. @Autowired
  88. private ISceneCopyLogService sceneCopyLogService;
  89. @Autowired
  90. private IScene3dNumService scene3dNumService;
  91. @Autowired
  92. private ISceneProEditService sceneProEditService;
  93. @Autowired
  94. private ISceneEditInfoService sceneEditInfoService;
  95. @Autowired
  96. private RedisUtil redisUtil;
  97. @Resource
  98. private ISceneUpgradeMapper sceneUpgradeMapper;
  99. @Autowired
  100. private ICameraService cameraService;
  101. @Value("${main.url}")
  102. private String mainUrl;
  103. @Value("${scene.pro.url}")
  104. private String sceneProUrl;
  105. @Value("${scene.pro.new.url}")
  106. private String sceneProNewUrl;
  107. @Value("${queue.modeling.obj.modeling-pre}")
  108. private String queueObjModelingPre;
  109. @Autowired
  110. private RabbitMqProducer mqProducer;
  111. @Override
  112. public HashMap<String, SceneNumVo> findSceneNumber(String token) {
  113. User user = userService.getByUserName(JwtUtil.getUsername(token));
  114. SceneNumVo sceneNumVoKk = getSceneNumVoByType(Arrays.asList(1,2,12,13,14), user.getId());
  115. Integer oldNum = sceneService.getCountByUserId(user.getId());
  116. sceneNumVoKk.setSceneNum( oldNum + sceneNumVoKk.getSceneNum());
  117. sceneNumVoKk.setTotalNum(oldNum + sceneNumVoKk.getTotalNum());
  118. SceneNumVo sceneNumVoKj = getSceneNumVoByType(Collections.singletonList(3), user.getId());
  119. SceneNumVo sceneNumVoSS = fdkkLaserService.getLaserSceneNumByUser(token);
  120. HashMap<String, SceneNumVo> hashMap = new HashMap<>();
  121. hashMap.put("kk",sceneNumVoKk);
  122. hashMap.put("kJ",sceneNumVoKj);
  123. hashMap.put("SS",sceneNumVoSS);
  124. return hashMap;
  125. }
  126. private SceneNumVo getSceneNumVoByType(List<Integer> sceneSourceList, Long userId){
  127. SceneNumVo sceneNumVo = new SceneNumVo();
  128. LambdaQueryWrapper<ScenePro> queryWrapper = new LambdaQueryWrapper<>();
  129. queryWrapper.eq(ScenePro::getUserId, userId);
  130. queryWrapper.in(ScenePro::getSceneSource, sceneSourceList);
  131. queryWrapper.eq(ScenePro::getIsUpgrade, 0);
  132. Integer sceneNum = Math.toIntExact(this.count(queryWrapper));
  133. Long count = scenePlusService.getCountByUserId(userId,sceneSourceList);
  134. sceneNumVo.setCooperationSceneNum(sceneCooperationService.getCooperationSceneNum(userId,sceneSourceList));
  135. sceneNumVo.setSceneNum(sceneNum + count);
  136. sceneNumVo.setTotalNum(sceneNumVo.getSceneNum() + sceneNumVo.getCooperationSceneNum());
  137. return sceneNumVo;
  138. }
  139. @Override
  140. public HashMap<Long,GroupByCount> findSceneNumByCameraIds(List<Long> cameraIdList) {
  141. HashMap<Long,GroupByCount> map = new HashMap<>();
  142. List<GroupByCount> result = this.getBaseMapper().findSceneProNumByCameraIds(cameraIdList);
  143. List<GroupByCount> result2 = this.getBaseMapper().findScenePlusNumByCameraIds(cameraIdList);
  144. result.forEach(entity ->map.put(entity.getId(),entity));
  145. for (GroupByCount groupByCount : result2) {
  146. if (map.get(groupByCount.getId()) != null) {
  147. Long totalCount = map.get(groupByCount.getId()).getCount() + groupByCount.getCount();
  148. String lastTime = DateUserUtil.getLastTime(map.get(groupByCount.getId()).getLastTime(), groupByCount.getLastTime());
  149. groupByCount.setCount(totalCount);
  150. groupByCount.setLastTime(lastTime);
  151. }
  152. map.put(groupByCount.getId(),groupByCount);
  153. }
  154. return map;
  155. }
  156. /**
  157. * @param payStatus -2 封存,为 1 解封
  158. */
  159. @Override
  160. public void lockOrUnLockBySpace(CameraDetail cameraDetail, Long cameraId, int payStatus) {
  161. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  162. LambdaQueryWrapper<ScenePlus> plusWr = new LambdaQueryWrapper<>();
  163. Long count = 0L;
  164. if(payStatus == 1){ //解封,判断用户权益,用户会员权益无限容量
  165. count = userIncrementService.getValidCountByCameraId(cameraId);
  166. wrapper.orderByAsc(ScenePro::getCreateTime);
  167. plusWr.orderByAsc(ScenePlus::getCreateTime);
  168. wrapper.eq(ScenePro::getPayStatus,-2);
  169. plusWr.eq(ScenePlus::getPayStatus,-2);
  170. }else {
  171. wrapper.orderByDesc(ScenePro::getCreateTime);
  172. plusWr.orderByDesc(ScenePlus::getCreateTime);
  173. wrapper.eq(ScenePro::getPayStatus,1);
  174. plusWr.eq(ScenePlus::getPayStatus,1);
  175. }
  176. wrapper.eq(ScenePro::getCameraId,cameraId)
  177. .eq(ScenePro::getSceneScheme,4)
  178. .eq(ScenePro::getIsUpgrade,0);
  179. plusWr.eq(ScenePlus::getCameraId,cameraId);
  180. List<ScenePro> list = this.list(wrapper);
  181. List<ScenePlus> plusList = scenePlusService.list(plusWr);
  182. Long beyondSpace = Math.abs(cameraDetail.getUsedSpace() - cameraDetail.getTotalSpace());
  183. Long accumulateSpace = 0L;
  184. List<Long> lockedIds = new ArrayList<>();
  185. if(payStatus == 1){
  186. getScenePlusLockedIds(lockedIds,plusList,count,beyondSpace,accumulateSpace);
  187. getSceneLockedIds(lockedIds,list,count,beyondSpace,accumulateSpace);
  188. }else {
  189. getSceneLockedIds(lockedIds,list,count,beyondSpace,accumulateSpace);
  190. getScenePlusLockedIds(lockedIds,plusList,count,beyondSpace,accumulateSpace);
  191. }
  192. lockOrUnLockScenes(lockedIds,payStatus);
  193. }
  194. private void getSceneLockedIds(List<Long> lockedIds ,List<ScenePro> list,Long count,Long beyondSpace,Long accumulateSpace){
  195. if (list != null && list.size() > 0){
  196. for (ScenePro scenePro : list){
  197. accumulateSpace += scenePro.getSpace();
  198. if (count ==0 && accumulateSpace.compareTo(beyondSpace) > 0){
  199. break;
  200. }
  201. lockedIds.add(scenePro.getId());
  202. }
  203. }
  204. }
  205. private void getScenePlusLockedIds(List<Long> lockedIds ,List<ScenePlus> list,Long count,Long beyondSpace,Long accumulateSpace){
  206. if (list != null && list.size() > 0){
  207. List<Long> plusIds = list.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
  208. HashMap<Long, ScenePlusExt> byPlusIds = scenePlusExtService.getByPlusIds(plusIds);
  209. for (ScenePlus scenePlus : list){
  210. ScenePlusExt scenePlusExt = byPlusIds.get(scenePlus.getId());
  211. accumulateSpace += scenePlusExt.getSpace();
  212. if (count ==0 && accumulateSpace.compareTo(beyondSpace) > 0){
  213. break;
  214. }
  215. lockedIds.add(scenePlus.getId());
  216. }
  217. }
  218. }
  219. // payStatus 为 -2 封存,为 1 解封
  220. private void lockOrUnLockScenes(List<Long> lockedIds,Integer payStatus) {
  221. if (lockedIds == null || lockedIds.size() == 0){
  222. return;
  223. }
  224. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  225. updateWrapper.set(ScenePro::getPayStatus,payStatus)
  226. .eq(ScenePro::getIsUpgrade,0)
  227. .in(ScenePro::getId,lockedIds);
  228. this.update(updateWrapper);
  229. LambdaUpdateWrapper<ScenePlus> updatePlusWrapper = new LambdaUpdateWrapper<>();
  230. updatePlusWrapper.set(ScenePlus::getPayStatus,payStatus)
  231. .in(ScenePlus::getId,lockedIds);
  232. scenePlusService.update(updatePlusWrapper);
  233. }
  234. @Override
  235. public List<ScenePro> getListByCameraId(Long cameraId) {
  236. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  237. wrapper.eq(ScenePro::getCameraId,cameraId)
  238. .eq(ScenePro::getIsUpgrade,0);
  239. return this.list(wrapper);
  240. }
  241. @Override
  242. public void bindOrUnCamera(List<Long> cameraIds, Long userId) {
  243. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  244. wrapper.in(ScenePro::getCameraId,cameraIds)
  245. .eq(ScenePro::getIsUpgrade,0)
  246. .set(ScenePro::getUserId,userId);
  247. this.update(wrapper);
  248. }
  249. @Override
  250. public List<ScenePro> getListByCameraIds(List<Long> cameraIds) {
  251. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  252. wrapper.in(ScenePro::getCameraId,cameraIds)
  253. .eq(ScenePro::getIsUpgrade,0);
  254. return this.list(wrapper);
  255. }
  256. @Override
  257. public List<ScenePro> getListByNums(List<String> numList) {
  258. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  259. wrapper.in(ScenePro::getNum,numList)
  260. .eq(ScenePro::getIsUpgrade,0);
  261. return this.list(wrapper);
  262. }
  263. @Override
  264. public Long getCountByUserId(Long userId, List<Integer> resourceList) {
  265. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  266. wrapper.eq(ScenePro::getUserId,userId);
  267. wrapper.eq(ScenePro::getIsUpgrade,0);
  268. wrapper.in(ScenePro::getSceneSource,resourceList);
  269. return this.count(wrapper);
  270. }
  271. @Override
  272. public Long getCountByUserId(Long id, Integer cameraType) {
  273. return this.getBaseMapper().getCountByUserId(id,cameraType);
  274. }
  275. @Override
  276. public void deleteByIds(List<Long> sceneIds) {
  277. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  278. wrapper.in(ScenePro::getId,sceneIds)
  279. .eq(ScenePro::getIsUpgrade,0);
  280. List<ScenePro> proList = this.list(wrapper);
  281. LambdaQueryWrapper<ScenePlus> plusWr = new LambdaQueryWrapper<>();
  282. plusWr.in(ScenePlus::getId,sceneIds);
  283. List<ScenePlus> plusList = scenePlusService.list(plusWr);
  284. this.deleteByList(proList,plusList);
  285. }
  286. @Override
  287. public JSONObject newList(SceneParam param, String username) {
  288. User user = userService.getByUserName(username);
  289. param.setSourceList(getSceneSource(param.getSceneSource()));
  290. param.setUserId(user.getId());
  291. if(param.getFolderId() == -1){ //双目文件夹场景
  292. return sceneService.pageList(param);
  293. }
  294. Page<SceneVo> sceneVoPage = getBaseMapper().pageListAndFolder(new Page<>(param.getPageNum(),param.getPageSize()),param);
  295. List<SceneVo> folderList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 1).collect(Collectors.toList());
  296. Integer folderNum = 0;
  297. if(folderList.size() >0){
  298. List<Long> folderIds = folderList.parallelStream().map(SceneVo::getId).collect(Collectors.toList());
  299. List<Folder> allFolderIds = folderService.getAllFolderIds(folderIds);
  300. folderNum = allFolderIds.size();
  301. HashMap<Long,Long> map = folderSceneService.getGroupByFolderIds(allFolderIds);
  302. for (SceneVo vo : sceneVoPage.getRecords()) {
  303. if(vo.getIsFolder() == 1){
  304. Long sceneNum = map.get(vo.getId()) == null ? 0L : map.get(vo.getId());
  305. vo.setSceneNum(sceneNum);
  306. vo.setCreateTimeStr(vo.getCreateTime());
  307. }
  308. }
  309. }
  310. //设置协作者信息
  311. if(sceneVoPage.getRecords().size() >0){
  312. List<String> numList = sceneVoPage.getRecords().parallelStream().map(SceneVo::getNum).collect(Collectors.toList());
  313. HashMap<String,User> cooMap = sceneCooperationService.getByNumList(numList);
  314. for (SceneVo vo : sceneVoPage.getRecords()) {
  315. if (StringUtils.isNotBlank(vo.getNum())) {
  316. User userVo = cooMap.get(vo.getNum());
  317. if (userVo != null) {
  318. vo.setCooperationUserId(userVo.getId().toString());
  319. vo.setCooperationUserName(userVo.getUserName());
  320. }
  321. }
  322. }
  323. }
  324. if(param.getSceneSource().equals("1") && param.getPageNum() == 1){ //增加双目文件夹
  325. SceneVo sceneVo = new SceneVo();
  326. sceneVo.setId(-1L);
  327. sceneVo.setIsFolder(1);
  328. sceneVo.setName("双目文件夹");
  329. sceneVo.setType("3");
  330. Integer count = sceneService.getCountByUserId(user.getId());
  331. sceneVo.setSceneNum(Long.valueOf(count));
  332. folderList.add(sceneVo);
  333. }
  334. List<SceneVo> sceneList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 0).collect(Collectors.toList());
  335. if(sceneList.size() >0){
  336. folderList.addAll(sceneList);
  337. }
  338. sceneVoPage.setRecords(folderList);
  339. JSONObject jsonObject = new JSONObject();
  340. jsonObject.put("pageInfo",PageInfo.PageInfo(sceneVoPage));
  341. jsonObject.put("sceneNum",sceneVoPage.getTotal() - folderNum);
  342. return jsonObject;
  343. }
  344. public List<String> getSceneSource(String sceneSource){
  345. List<String> sceneSourceList ;
  346. switch (sceneSource){
  347. case "1" : sceneSourceList = Arrays.asList("1","2","12","13","14"); break; //四维看看
  348. case "3" : sceneSourceList = Collections.singletonList("3"); break;//四维看见
  349. default:
  350. if(sceneSource.contains(",")){
  351. sceneSource = sceneSource + ",12,13,14";
  352. sceneSourceList = Arrays.asList(sceneSource.split(","));
  353. break;
  354. }
  355. sceneSourceList = Collections.singletonList(sceneSource);break;
  356. }
  357. return sceneSourceList;
  358. }
  359. @Override
  360. public void copyScene(String sceneNum,String userName) throws Exception {
  361. if(StringUtils.isEmpty(sceneNum)){
  362. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  363. }
  364. ScenePro scenePro = this.getByNum(sceneNum);
  365. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  366. if((scenePro == null || scenePro.getCameraId() == null ) && (scenePlus== null || scenePlus.getCameraId() == null)){
  367. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  368. }
  369. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  370. Long userId = scenePro == null ? scenePlus.getUserId() :scenePro.getUserId();
  371. // 判断是否有绑定会员权益
  372. Long count = userIncrementService.getValidCountByCameraId(cameraId);
  373. if(count <= 0){
  374. throw new BusinessException(LoginConstant.FAILURE_CODE_3030, LoginConstant.FAILURE_MSG_3030);
  375. }
  376. if(userId == null){
  377. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  378. }
  379. //只能复制自己的场景
  380. User user = userService.getByUserName(userName);
  381. if(user == null || !user.getId().equals(userId) ){
  382. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  383. }
  384. this.copySceneNoCheck(sceneNum);
  385. }
  386. @Override
  387. public void copySceneNoCheck(String sceneNum) throws Exception {
  388. ScenePro scenePro = this.getByNum(sceneNum);
  389. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  390. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  391. CameraDetail detailEntity = cameraDetailService.getByCameraId(cameraId);
  392. if(detailEntity == null){
  393. throw new BusinessException(CameraConstant.FAILURE_CODE_6029, CameraConstant.FAILURE_MSG_6029);
  394. }
  395. String newNum = scene3dNumService.generateSceneNum(detailEntity.getType());
  396. String title = scenePro == null ? scenePlus.getTitle() : scenePro.getSceneName();
  397. String newTitle = title.concat("(copy)");
  398. Long sceneId = scenePro == null ? scenePlus.getId() :scenePro.getId();
  399. Long newSceneId = null;
  400. FolderScene folderScene = folderSceneService.getByType(sceneId, null);
  401. String time = DateUtil.date2String(new Date(),DateUtil.YYYYMMDDHHMMSSSSS_DATA_FORMAT);
  402. if(scenePro !=null){ //v3场景复制
  403. log.info("场景复制--V3--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  404. scenePro.getNum(),scenePro.getSceneName(),newNum,newTitle);
  405. scenePro.setSceneName(newTitle);
  406. newSceneId = this.copyV3Scene(scenePro, newNum, detailEntity,time);
  407. }
  408. if(scenePlus != null){ //v4场景复制
  409. log.info("场景复制--V4--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  410. scenePlus.getNum(),scenePlus.getTitle(),newNum,newTitle);
  411. scenePlus.setTitle(newTitle);
  412. newSceneId = scenePlusService.copyV4Scene(scenePlus,newNum,detailEntity,time);
  413. }
  414. log.info("场景复制--完成--sceneId:{}",newSceneId);
  415. if(newSceneId != null && folderScene!= null){
  416. folderScene.setId(null);
  417. folderScene.setSceneId(newSceneId);
  418. folderSceneService.save(folderScene);
  419. }
  420. }
  421. @Override
  422. public ScenePro getByNum(String sceneNum) {
  423. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  424. wrapper.eq(ScenePro::getNum,sceneNum);
  425. wrapper.eq(ScenePro::getIsUpgrade,0);
  426. List<ScenePro> list = this.list(wrapper);
  427. if(list!= null && list.size() >0){
  428. return list.get(0);
  429. }
  430. return null;
  431. }
  432. @Override
  433. public Long copyV3Scene(ScenePro oldScene,String newNum,CameraDetail cameraDetail,String time) throws Exception {
  434. SceneProEdit oldEditScene = sceneProEditService.getByProId(oldScene.getId());
  435. String oldNum = oldScene.getNum();
  436. //复制数据库数据
  437. oldScene.setId(null);
  438. oldScene.setStatus(0);
  439. oldScene.setNum(newNum);
  440. oldScene.setWebSite(oldScene.getWebSite().replace(oldScene.getNum(), newNum));
  441. oldScene.setThumb(oldScene.getThumb().replace(oldScene.getNum(), newNum));
  442. oldScene.setVideos(this.setVideos(oldScene.getVideos(),oldNum,newNum));
  443. oldScene.setViewCount(0);
  444. String preDataSource = oldScene.getDataSource();
  445. String newDataSource = this.setDataSource(preDataSource,time);
  446. oldScene.setDataSource(newDataSource);
  447. this.save(oldScene);
  448. //更新video
  449. Map map = new HashMap();
  450. JSONObject object = new JSONObject();
  451. if(StringUtils.isNotEmpty(oldScene.getVideos())){
  452. object = JSONObject.parseObject(oldScene.getVideos());
  453. if(object.containsKey("upPath")){
  454. String upPath = object.getString("upPath");
  455. upPath = upPath.replace(oldNum,newNum);
  456. object.put("upPath",upPath);
  457. }
  458. oldScene.setVideos(object.toJSONString());
  459. log.info("更新 scene.json");
  460. map.put("videos",oldScene.getVideos());
  461. }
  462. map.put("sceneName",oldScene.getSceneName());
  463. map.put("webSite",oldScene.getWebSite());
  464. map.put("thumb",oldScene.getThumb());
  465. map.put("num",oldScene.getNum());
  466. map.put("id",oldScene.getId());
  467. map.put("dataSource",oldScene.getDataSource());
  468. sceneCopyLogService.saveByNum(oldNum,newNum,oldScene.getUserId());
  469. //复制完成更新相机容量
  470. oldEditScene.setId(null);
  471. oldEditScene.setProId(oldScene.getId());
  472. oldEditScene.setScreencapVoiceSrc(oldEditScene.getScreencapVoiceSrc() == null ? null : oldEditScene.getScreencapVoiceSrc().replace(oldNum, newNum));
  473. oldEditScene.setScreencapVoiceSound(oldEditScene.getScreencapVoiceSound() == null ? null : oldEditScene.getScreencapVoiceSound().replace(oldNum, newNum));
  474. oldEditScene.setScreencapVoiceSoundsync(oldEditScene.getScreencapVoiceSoundsync() == null ? null : oldEditScene.getScreencapVoiceSoundsync().replace(oldNum, newNum));
  475. oldEditScene.setPlayData(oldEditScene.getPlayData() == null ? null : oldEditScene.getPlayData().replace(oldNum, newNum));
  476. oldEditScene.setScreencapThumb(oldEditScene.getScreencapThumb() == null ? null : oldEditScene.getScreencapThumb().replace(oldNum, newNum));
  477. oldEditScene.setFloorPlanPng(oldEditScene.getFloorPlanPng() == null ? null : oldEditScene.getFloorPlanPng().replace(oldNum, newNum));
  478. sceneProEditService.save(oldEditScene);
  479. this.setQrCode(oldScene.getBuildType(), oldScene.getNum());
  480. CompletableFuture.runAsync(() -> {
  481. try {
  482. log.info("开始复制场景-{}", new Date());
  483. String buildModelPath = ConstantFilePath.BUILD_MODEL_PATH;
  484. if(oldScene.getSceneSource().equals(4)){
  485. buildModelPath = ConstantFilePath.BUILD_MODEL_LASER_PATH;
  486. }
  487. this.copyFdage(preDataSource,newDataSource,buildModelPath,time);
  488. //oss复制计算结果资源
  489. downloadFromOldNumFromOss(oldNum, newNum);
  490. // 复制本地secen.json
  491. copyOldSceneLocalToNew(oldNum, newNum);
  492. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  493. //上传资源到oss
  494. uploadNewSceneToOss(newNum);
  495. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  496. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  497. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  498. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  499. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"images/images" + oldNum,ConstantFilePath.SCENE_PATH +"images/images" + newNum);
  500. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"data/data" + oldNum,ConstantFilePath.SCENE_PATH +"data/data" + newNum);
  501. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  502. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",oldNum, newNum);
  503. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",oldNum, newNum);
  504. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"voice/voice" + oldNum,ConstantFilePath.SCENE_PATH +"voice/voice" + newNum);
  505. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"video/video" + oldNum,ConstantFilePath.SCENE_PATH +"video/video" + newNum);
  506. oldScene.setStatus(-2);
  507. this.updateById(oldScene);
  508. log.info("复制场景结束-{}", new Date());
  509. } catch (Exception e) {
  510. this.removeById(oldScene.getId());
  511. log.error("复制场景异常", e);
  512. }
  513. });
  514. return oldScene.getId();
  515. }
  516. /**
  517. * 从旧场景下载资源
  518. * @param sceneNum
  519. * @param newNum
  520. * @throws Exception
  521. */
  522. private void downloadFromOldNumFromOss(String sceneNum, String newNum) throws Exception {
  523. ShellUtil.yunDownload("images/images" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  524. ShellUtil.yunDownload("data/data" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  525. ShellUtil.yunDownload("voice/voice" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  526. ShellUtil.yunDownload("video/video" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  527. }
  528. /**
  529. * 从本地旧场景拷贝资源到新场景
  530. * @param sceneNum
  531. * @param newNum
  532. */
  533. private void copyOldSceneLocalToNew(String sceneNum, String newNum) throws Exception {
  534. FileUtils.copyFolderAllFiles(ConstantFilePath.SCENE_PATH + "data/data" + sceneNum + "/",
  535. ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/", true);
  536. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",sceneNum, newNum);
  537. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",sceneNum, newNum);
  538. }
  539. public void copyFdage(String oldDataSource,String newDataSource,String buildModelPath,String time) throws Exception {
  540. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.replace(buildModelPath, "")+"/", newDataSource);
  541. // 修改data.fdage
  542. String data = FileUtils.readFile(newDataSource + "/data.fdage");
  543. JSONObject jsonObject = JSONObject.parseObject(data);
  544. if(ObjectUtils.isEmpty(jsonObject)){
  545. log.error("data.fdage文件不存在");
  546. throw new BusinessException(-1,"拷贝场景出错,data.fdage文件不存在");
  547. }
  548. jsonObject.put("uuidtime",time);
  549. FileUtils.writeFile(newDataSource + "/data.fdage", jsonObject.toJSONString());
  550. ShellUtil.yunUpload(newDataSource, ConstantFilePath.OSS_PREFIX + newDataSource.replace(buildModelPath, ""));
  551. // 复制计算结果
  552. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.concat("_results/").replace(buildModelPath, ""), newDataSource.concat("_results"));
  553. if(new File(newDataSource.concat("_results")).exists()){
  554. ShellUtil.yunUpload(newDataSource.concat("_results"), ConstantFilePath.OSS_PREFIX + newDataSource.concat("_results").replace(buildModelPath, ""));
  555. FileUtils.delAllFile(newDataSource.concat("_results"));
  556. }
  557. FileUtils.delAllFile(newDataSource);
  558. FileUtils.copyFolderAllFiles(oldDataSource+"/",newDataSource+"/", true);
  559. }
  560. @Override
  561. public String setDataSource(String preDataSource,String time) throws Exception {
  562. SnowflakeIdGenerator snowflakeIdGenerator = new SnowflakeIdGenerator(0,1);
  563. String[] datasource = preDataSource.split("/");
  564. datasource[4] = snowflakeIdGenerator.nextId()+"";
  565. datasource[5] = datasource[5].split("_")[0] + "_" + time;
  566. return Arrays.stream(datasource).collect(Collectors.joining("/"));
  567. }
  568. @Override
  569. public String setVideos(String videos,String oldNum,String newNum) {
  570. JSONObject object = new JSONObject();
  571. if(StringUtils.isNotEmpty(videos)){
  572. object = JSONObject.parseObject(videos);
  573. if(object.containsKey("upPath")){
  574. String upPath = object.getString("upPath");
  575. upPath = upPath.replace(oldNum,newNum);
  576. object.put("upPath",upPath);
  577. }
  578. log.info("更新 scene.json");
  579. }
  580. return object.toJSONString();
  581. }
  582. private void reloadFile(String filePath,String sceneNum, String newNum) throws Exception {
  583. // 修改link-scene.json
  584. String content = FileUtils.readFile(filePath);
  585. if (!ObjectUtils.isEmpty(content)) {
  586. content = content.replaceAll(sceneNum, newNum);
  587. FileUtils.writeFile(filePath,content);
  588. }
  589. }
  590. /**
  591. * 上传新场景资源到oss
  592. * @param newNum
  593. */
  594. private void uploadNewSceneToOss(String newNum) {
  595. ShellUtil.yunUpload("/mnt/4Dkankan/scene/images/images" + newNum, "images/images" + newNum);
  596. ShellUtil.yunUpload("/mnt/4Dkankan/scene/data/data" + newNum, "data/data" + newNum);
  597. ShellUtil.yunUpload("/mnt/4Dkankan/scene/voice/voice" + newNum, "voice/voice" + newNum);
  598. ShellUtil.yunUpload("/mnt/4Dkankan/scene/video/video" + newNum, "video/video" + newNum);
  599. }
  600. @Override
  601. public void delete(String sceneNum) {
  602. if(StringUtils.isEmpty(sceneNum)){
  603. throw new BusinessException(LoginConstant.FAILURE_CODE_3001,LoginConstant.FAILURE_MSG_3001);
  604. }
  605. String[] nums = sceneNum.split(",");
  606. List<String> numList = Arrays.asList(nums);
  607. List<ScenePro> proList = this.getListByNums(numList);
  608. List<ScenePlus> plusList = scenePlusService.getListByNums(numList);
  609. this.deleteByList(proList,plusList);
  610. }
  611. @Override
  612. public void deleteByList(List<ScenePro> proList,List<ScenePlus> plusList) {
  613. HashMap<Long ,Long >cameraMap = new HashMap<>();
  614. if(proList.size() >0){
  615. for (ScenePro scenePro : proList) {
  616. cameraMap.merge(scenePro.getCameraId(), scenePro.getSpace(), Long::sum);
  617. }
  618. List<Long> proIds = proList.parallelStream().map(ScenePro::getId).collect(Collectors.toList());
  619. this.removeByIds(proIds);
  620. }
  621. if(plusList.size() >0){
  622. List<Long> plusIds = plusList.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
  623. HashMap<Long, ScenePlusExt> plusMap = scenePlusExtService.getByPlusIds(plusIds);
  624. for (ScenePlus scenePlus : plusList) {
  625. cameraMap.merge(scenePlus.getCameraId(), plusMap.get(scenePlus.getId()).getSpace(), Long::sum);
  626. }
  627. scenePlusService.removeByIds(plusIds);
  628. }
  629. //恢复相机使用容量
  630. if(cameraMap.size() >0){
  631. cameraDetailService.addUsedSpace(cameraMap);
  632. }
  633. }
  634. @Override
  635. public ScenePro getByUnicode(String unicode) {
  636. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  637. wrapper.like(ScenePro::getDataSource,unicode);
  638. wrapper.eq(ScenePro::getIsUpgrade,0);
  639. wrapper.orderByDesc(ScenePro::getCreateTime);
  640. List<ScenePro> list = this.list(wrapper);
  641. if(list !=null && list.size() >0){
  642. return list.get(0);
  643. }
  644. return null;
  645. }
  646. @Override
  647. public SceneInfoVo getInfo(String num) {
  648. if(StringUtils.isEmpty(num)){
  649. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  650. }
  651. ScenePro scenePro = this.getByNum(num);
  652. ScenePlus scenePlus = scenePlusService.getByNum(num);
  653. if(scenePro == null && scenePlus == null){
  654. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  655. }
  656. if(scenePro!=null && scenePro.getStatus() != 1 && scenePro.getStatus() != -2){
  657. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  658. }
  659. if(scenePlus!=null && scenePlus.getSceneStatus() != 1 && scenePlus.getSceneStatus() != -2){
  660. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  661. }
  662. if(scenePro!=null && scenePro.getPayStatus() != 1){
  663. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  664. }
  665. if(scenePlus!=null && scenePlus.getPayStatus() != 1){
  666. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  667. }
  668. SceneInfoVo infoVo = new SceneInfoVo();
  669. if(scenePro!=null){
  670. SceneProEdit sceneProEdit = sceneProEditService.getByProId(scenePro.getId());
  671. if(sceneProEdit !=null){
  672. BeanUtils.copyProperties(sceneProEdit, infoVo);
  673. }
  674. BeanUtils.copyProperties(scenePro,infoVo);
  675. if(StringUtils.isNotEmpty(scenePro.getGps())){
  676. infoVo.setGps(JSONObject.parseObject(scenePro.getGps()).toJSONString());
  677. }
  678. infoVo.setSceneKey("");
  679. infoVo.setCreateDate(DateUserUtil.getDate(scenePro.getCreateTime()).getTime());
  680. }else {
  681. SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
  682. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
  683. if(sceneEditInfo != null){
  684. BeanUtils.copyProperties(sceneEditInfo,infoVo);
  685. }
  686. if(scenePlusExt !=null){
  687. BeanUtils.copyProperties(scenePlusExt,infoVo);
  688. }
  689. BeanUtils.copyProperties(scenePlus,infoVo);
  690. infoVo.setSceneName(scenePlus.getTitle());
  691. infoVo.setStatus(scenePlus.getSceneStatus());
  692. }
  693. if(StringUtils.isNotEmpty(infoVo.getEntry())){
  694. infoVo.setEntry(JSONObject.parseObject(infoVo.getEntry()).toJSONString());
  695. }
  696. if(StringUtils.isEmpty(infoVo.getSceneKey())){
  697. infoVo.setIsPublic(0);
  698. }else {
  699. infoVo.setIsPublic(1);
  700. }
  701. //查询是否有随心装场景
  702. if(num.contains("vr-")){
  703. infoVo.setVrNum(null);
  704. infoVo.setVideosUser(null);
  705. infoVo.setBgMusicName(null);
  706. infoVo.setBgMusic(null);
  707. }
  708. this.updateViewCount(num);
  709. return infoVo;
  710. }
  711. @Override
  712. public void updateViewCount(String num) {
  713. String redisKey = RedisKey.SCENE_VISIT_CNT;
  714. if(!redisUtil.hHasKey(redisKey,num)){
  715. Integer count = 0;
  716. ScenePro pro = this.getByNum(num);
  717. if(pro !=null){
  718. count = pro.getViewCount();
  719. }else {
  720. ScenePlus plus = scenePlusService.getByNum(num);
  721. if(plus !=null){
  722. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(plus.getId());
  723. count = scenePlusExt.getViewCount();
  724. }
  725. }
  726. redisUtil.hset(redisKey,num,String.valueOf(count));
  727. }
  728. redisUtil.hincr(redisKey,num,1);
  729. }
  730. @Override
  731. public void updateDbViewCount(String num, String count) {
  732. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  733. wrapper.eq(ScenePro::getNum,num);
  734. wrapper.set(ScenePro::getViewCount,count);
  735. this.update(wrapper);
  736. ScenePlus plus = scenePlusService.getByNum(num);
  737. if(plus !=null){
  738. LambdaUpdateWrapper<ScenePlusExt> exWr = new LambdaUpdateWrapper<>();
  739. exWr.eq(ScenePlusExt::getPlusId,plus.getId())
  740. .set(ScenePlusExt::getViewCount,count);
  741. scenePlusExtService.update(exWr);
  742. }
  743. }
  744. @Override
  745. public ScenePro findByFileId(String fileId) {
  746. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  747. wrapper.like(ScenePro::getDataSource,fileId);
  748. List<ScenePro> list = this.list(wrapper);
  749. if(list !=null && list.size() >0){
  750. return list.get(0);
  751. }
  752. return null;
  753. }
  754. @Override
  755. public void setQrCode(String buildType,String num) throws Exception {
  756. String basePath = mainUrl;
  757. if("V2".equals(buildType)){
  758. basePath += sceneProUrl;
  759. }
  760. if("V3".equals(buildType)){
  761. basePath += sceneProNewUrl;
  762. }
  763. MatrixToImageWriterUtil.createQRCode(basePath + num, ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+".png", false,null);
  764. MatrixToImageWriterUtil.createQRCode(basePath + num + "&lang=en", ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+"_en.png", false,null);
  765. }
  766. @Override
  767. public void generateObjFile(String num) {
  768. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  769. wrapper.eq(ScenePro::getNum,num).eq(ScenePro::getRecStatus,"A");
  770. ScenePro sceneProEntity = this.getOne(wrapper);
  771. if(ObjectUtils.isEmpty(sceneProEntity)){
  772. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  773. }
  774. if(sceneProEntity.getSceneSource().intValue() != 4){
  775. throw new BusinessException(LoginConstant.FAILURE_CODE_3003, "只能操作激光场景");
  776. }
  777. // 拷贝文件
  778. String path = sceneProEntity.getDataSource();
  779. if (!new File(path + "/caches/reconstruction/final.bin").exists()
  780. || !new File(path + "/caches/reconstruction/chunk.json").exists()
  781. || !new File(path + "/caches/images").exists()
  782. || !new File(path + "/caches/depthmap").exists()
  783. || !new File(path + "/caches/depthmap_csc").exists()
  784. || !new File(path + "/caches/panorama.json").exists()
  785. || !new File(path + "/results/laserData/laser.ply").exists()) {
  786. log.error("生成obj缺少必要文件,生成失败!");
  787. throw new BusinessException(SceneConstant.FAILURE_CODE_5038, SceneConstant.FAILURE_MSG_5038);
  788. }
  789. // 获取最新的场景名称
  790. JSONObject sceneInfo = fdkkLaserService.getSceneByNum(sceneProEntity.getNum());
  791. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  792. updateWrapper
  793. .set(ScenePro::getStatus, 0)
  794. .set(ScenePro::getIsObj, 1)
  795. .set(ScenePro::getSceneName, sceneInfo.getString("title"))
  796. .eq(ScenePro::getNum, sceneProEntity.getNum());
  797. this.update(updateWrapper);
  798. //同步到scenePlus、scenePlus
  799. sceneUpgradeMapper.deleteScenePlus(sceneProEntity.getId());
  800. sceneUpgradeMapper.transferScenePlus(sceneProEntity.getId());
  801. sceneUpgradeMapper.deleteScenePlusExt(sceneProEntity.getId());
  802. String sceneKind = sceneProEntity.getSceneScheme() == 3 ? SceneKind.FACE.code(): SceneKind.TILES.code();
  803. sceneUpgradeMapper.transferScenePlusExt(sceneProEntity.getId(), sceneKind);
  804. log.info("开始发送激光场景生成obj mq消息");
  805. Camera camera = cameraService.getById(sceneProEntity.getCameraId());
  806. // 发送MQ
  807. BuildSceneCallMessage mqMsg = new BuildSceneCallMessage();
  808. mqMsg.setSceneNum(sceneProEntity.getNum());
  809. mqMsg.setUserId(sceneProEntity.getUserId());
  810. mqMsg.setAlgorithm(sceneProEntity.getAlgorithm());
  811. mqMsg.setBuildType(sceneProEntity.getBuildType());
  812. mqMsg.setDataSource(sceneProEntity.getDataSource());
  813. mqProducer.sendByWorkQueue(queueObjModelingPre,mqMsg);
  814. }
  815. }