SceneProServiceImpl.java 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901
  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. Page<SceneVo> sceneVoPage = getBaseMapper().pageListAndFolder(new Page<>(param.getPageNum(),param.getPageSize()),param);
  292. List<SceneVo> folderList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 1).collect(Collectors.toList());
  293. Integer folderNum = 0;
  294. if(folderList.size() >0){
  295. List<Long> folderIds = folderList.parallelStream().map(SceneVo::getId).collect(Collectors.toList());
  296. List<Folder> allFolderIds = folderService.getAllFolderIds(folderIds);
  297. folderNum = allFolderIds.size();
  298. HashMap<Long,Long> map = folderSceneService.getGroupByFolderIds(allFolderIds);
  299. for (SceneVo vo : sceneVoPage.getRecords()) {
  300. if(vo.getIsFolder() == 1){
  301. Long sceneNum = map.get(vo.getId()) == null ? 0L : map.get(vo.getId());
  302. vo.setSceneNum(sceneNum);
  303. vo.setCreateTimeStr(vo.getCreateTime());
  304. }
  305. }
  306. }
  307. //设置协作者信息
  308. if(sceneVoPage.getRecords().size() >0){
  309. List<String> numList = sceneVoPage.getRecords().parallelStream().map(SceneVo::getNum).collect(Collectors.toList());
  310. HashMap<String,User> cooMap = sceneCooperationService.getByNumList(numList);
  311. for (SceneVo vo : sceneVoPage.getRecords()) {
  312. if (StringUtils.isNotBlank(vo.getNum())) {
  313. User userVo = cooMap.get(vo.getNum());
  314. if (userVo != null) {
  315. vo.setCooperationUserId(userVo.getId().toString());
  316. vo.setCooperationUserName(userVo.getUserName());
  317. }
  318. }
  319. }
  320. }
  321. if(param.getSceneSource().equals("1")){ //增加双目文件夹
  322. SceneVo sceneVo = new SceneVo();
  323. sceneVo.setId(-1L);
  324. sceneVo.setIsFolder(1);
  325. sceneVo.setName("双目文件夹");
  326. sceneVo.setType("0");
  327. Integer count = sceneService.getCountByUserId(user.getId());
  328. sceneVo.setSceneNum(Long.valueOf(count));
  329. folderList.add(sceneVo);
  330. }
  331. List<SceneVo> sceneList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 0).collect(Collectors.toList());
  332. if(sceneList.size() >0){
  333. folderList.addAll(sceneList);
  334. }
  335. sceneVoPage.setRecords(folderList);
  336. JSONObject jsonObject = new JSONObject();
  337. jsonObject.put("pageInfo",PageInfo.PageInfo(sceneVoPage));
  338. jsonObject.put("sceneNum",sceneVoPage.getTotal() - folderNum);
  339. return jsonObject;
  340. }
  341. public List<String> getSceneSource(String sceneSource){
  342. List<String> sceneSourceList ;
  343. switch (sceneSource){
  344. case "1" : sceneSourceList = Arrays.asList("1","2","12","13","14"); break; //四维看看
  345. case "3" : sceneSourceList = Collections.singletonList("3"); break;//四维看见
  346. default:
  347. if(sceneSource.contains(",")){
  348. sceneSource = sceneSource + ",12,13,14";
  349. sceneSourceList = Arrays.asList(sceneSource.split(","));
  350. break;
  351. }
  352. sceneSourceList = Collections.singletonList(sceneSource);break;
  353. }
  354. return sceneSourceList;
  355. }
  356. @Override
  357. public void copyScene(String sceneNum,String userName) throws Exception {
  358. if(StringUtils.isEmpty(sceneNum)){
  359. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  360. }
  361. ScenePro scenePro = this.getByNum(sceneNum);
  362. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  363. if((scenePro == null || scenePro.getCameraId() == null ) && (scenePlus== null || scenePlus.getCameraId() == null)){
  364. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  365. }
  366. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  367. Long userId = scenePro == null ? scenePlus.getUserId() :scenePro.getUserId();
  368. // 判断是否有绑定会员权益
  369. Long count = userIncrementService.getValidCountByCameraId(cameraId);
  370. if(count <= 0){
  371. throw new BusinessException(LoginConstant.FAILURE_CODE_3030, LoginConstant.FAILURE_MSG_3030);
  372. }
  373. if(userId == null){
  374. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  375. }
  376. //只能复制自己的场景
  377. User user = userService.getByUserName(userName);
  378. if(user == null || !user.getId().equals(userId) ){
  379. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  380. }
  381. this.copySceneNoCheck(sceneNum);
  382. }
  383. @Override
  384. public void copySceneNoCheck(String sceneNum) throws Exception {
  385. ScenePro scenePro = this.getByNum(sceneNum);
  386. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  387. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  388. CameraDetail detailEntity = cameraDetailService.getByCameraId(cameraId);
  389. if(detailEntity == null){
  390. throw new BusinessException(CameraConstant.FAILURE_CODE_6029, CameraConstant.FAILURE_MSG_6029);
  391. }
  392. String newNum = scene3dNumService.generateSceneNum(detailEntity.getType());
  393. String title = scenePro == null ? scenePlus.getTitle() : scenePro.getSceneName();
  394. String newTitle = title.concat("(copy)");
  395. Long sceneId = scenePro == null ? scenePlus.getId() :scenePro.getId();
  396. Long newSceneId = null;
  397. FolderScene folderScene = folderSceneService.getByType(sceneId, null);
  398. String time = DateUtil.date2String(new Date(),DateUtil.YYYYMMDDHHMMSSSSS_DATA_FORMAT);
  399. if(scenePro !=null){ //v3场景复制
  400. log.info("场景复制--V3--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  401. scenePro.getNum(),scenePro.getSceneName(),newNum,newTitle);
  402. scenePro.setSceneName(newTitle);
  403. newSceneId = this.copyV3Scene(scenePro, newNum, detailEntity,time);
  404. }
  405. if(scenePlus != null){ //v4场景复制
  406. log.info("场景复制--V4--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  407. scenePlus.getNum(),scenePlus.getTitle(),newNum,newTitle);
  408. scenePlus.setTitle(newTitle);
  409. newSceneId = scenePlusService.copyV4Scene(scenePlus,newNum,detailEntity,time);
  410. }
  411. log.info("场景复制--完成--sceneId:{}",newSceneId);
  412. if(newSceneId != null && folderScene!= null){
  413. folderScene.setId(null);
  414. folderScene.setSceneId(newSceneId);
  415. folderSceneService.save(folderScene);
  416. }
  417. }
  418. @Override
  419. public ScenePro getByNum(String sceneNum) {
  420. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  421. wrapper.eq(ScenePro::getNum,sceneNum);
  422. wrapper.eq(ScenePro::getIsUpgrade,0);
  423. List<ScenePro> list = this.list(wrapper);
  424. if(list!= null && list.size() >0){
  425. return list.get(0);
  426. }
  427. return null;
  428. }
  429. @Override
  430. public Long copyV3Scene(ScenePro oldScene,String newNum,CameraDetail cameraDetail,String time) throws Exception {
  431. SceneProEdit oldEditScene = sceneProEditService.getByProId(oldScene.getId());
  432. String oldNum = oldScene.getNum();
  433. //复制数据库数据
  434. oldScene.setId(null);
  435. oldScene.setStatus(0);
  436. oldScene.setNum(newNum);
  437. oldScene.setWebSite(oldScene.getWebSite().replace(oldScene.getNum(), newNum));
  438. oldScene.setThumb(oldScene.getThumb().replace(oldScene.getNum(), newNum));
  439. oldScene.setVideos(this.setVideos(oldScene.getVideos(),oldNum,newNum));
  440. oldScene.setViewCount(0);
  441. String preDataSource = oldScene.getDataSource();
  442. String newDataSource = this.setDataSource(preDataSource,time);
  443. oldScene.setDataSource(newDataSource);
  444. this.save(oldScene);
  445. //更新video
  446. Map map = new HashMap();
  447. JSONObject object = new JSONObject();
  448. if(StringUtils.isNotEmpty(oldScene.getVideos())){
  449. object = JSONObject.parseObject(oldScene.getVideos());
  450. if(object.containsKey("upPath")){
  451. String upPath = object.getString("upPath");
  452. upPath = upPath.replace(oldNum,newNum);
  453. object.put("upPath",upPath);
  454. }
  455. oldScene.setVideos(object.toJSONString());
  456. log.info("更新 scene.json");
  457. map.put("videos",oldScene.getVideos());
  458. }
  459. map.put("sceneName",oldScene.getSceneName());
  460. map.put("webSite",oldScene.getWebSite());
  461. map.put("thumb",oldScene.getThumb());
  462. map.put("num",oldScene.getNum());
  463. map.put("id",oldScene.getId());
  464. map.put("dataSource",oldScene.getDataSource());
  465. sceneCopyLogService.saveByNum(oldNum,newNum,oldScene.getUserId());
  466. //复制完成更新相机容量
  467. oldEditScene.setId(null);
  468. oldEditScene.setProId(oldScene.getId());
  469. oldEditScene.setScreencapVoiceSrc(oldEditScene.getScreencapVoiceSrc() == null ? null : oldEditScene.getScreencapVoiceSrc().replace(oldNum, newNum));
  470. oldEditScene.setScreencapVoiceSound(oldEditScene.getScreencapVoiceSound() == null ? null : oldEditScene.getScreencapVoiceSound().replace(oldNum, newNum));
  471. oldEditScene.setScreencapVoiceSoundsync(oldEditScene.getScreencapVoiceSoundsync() == null ? null : oldEditScene.getScreencapVoiceSoundsync().replace(oldNum, newNum));
  472. oldEditScene.setPlayData(oldEditScene.getPlayData() == null ? null : oldEditScene.getPlayData().replace(oldNum, newNum));
  473. oldEditScene.setScreencapThumb(oldEditScene.getScreencapThumb() == null ? null : oldEditScene.getScreencapThumb().replace(oldNum, newNum));
  474. oldEditScene.setFloorPlanPng(oldEditScene.getFloorPlanPng() == null ? null : oldEditScene.getFloorPlanPng().replace(oldNum, newNum));
  475. sceneProEditService.save(oldEditScene);
  476. this.setQrCode(oldScene.getBuildType(), oldScene.getNum());
  477. CompletableFuture.runAsync(() -> {
  478. try {
  479. log.info("开始复制场景-{}", new Date());
  480. String buildModelPath = ConstantFilePath.BUILD_MODEL_PATH;
  481. if(oldScene.getSceneSource().equals(4)){
  482. buildModelPath = ConstantFilePath.BUILD_MODEL_LASER_PATH;
  483. }
  484. this.copyFdage(preDataSource,newDataSource,buildModelPath,time);
  485. //oss复制计算结果资源
  486. downloadFromOldNumFromOss(oldNum, newNum);
  487. // 复制本地secen.json
  488. copyOldSceneLocalToNew(oldNum, newNum);
  489. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  490. //上传资源到oss
  491. uploadNewSceneToOss(newNum);
  492. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  493. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  494. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  495. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  496. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"images/images" + oldNum,ConstantFilePath.SCENE_PATH +"images/images" + newNum);
  497. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"data/data" + oldNum,ConstantFilePath.SCENE_PATH +"data/data" + newNum);
  498. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  499. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",oldNum, newNum);
  500. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",oldNum, newNum);
  501. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"voice/voice" + oldNum,ConstantFilePath.SCENE_PATH +"voice/voice" + newNum);
  502. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"video/video" + oldNum,ConstantFilePath.SCENE_PATH +"video/video" + newNum);
  503. oldScene.setStatus(-2);
  504. this.updateById(oldScene);
  505. log.info("复制场景结束-{}", new Date());
  506. } catch (Exception e) {
  507. this.removeById(oldScene.getId());
  508. log.error("复制场景异常", e);
  509. }
  510. });
  511. return oldScene.getId();
  512. }
  513. /**
  514. * 从旧场景下载资源
  515. * @param sceneNum
  516. * @param newNum
  517. * @throws Exception
  518. */
  519. private void downloadFromOldNumFromOss(String sceneNum, String newNum) throws Exception {
  520. ShellUtil.yunDownload("images/images" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  521. ShellUtil.yunDownload("data/data" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  522. ShellUtil.yunDownload("voice/voice" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  523. ShellUtil.yunDownload("video/video" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  524. }
  525. /**
  526. * 从本地旧场景拷贝资源到新场景
  527. * @param sceneNum
  528. * @param newNum
  529. */
  530. private void copyOldSceneLocalToNew(String sceneNum, String newNum) throws Exception {
  531. FileUtils.copyFolderAllFiles(ConstantFilePath.SCENE_PATH + "data/data" + sceneNum + "/",
  532. ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/", true);
  533. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",sceneNum, newNum);
  534. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",sceneNum, newNum);
  535. }
  536. public void copyFdage(String oldDataSource,String newDataSource,String buildModelPath,String time) throws Exception {
  537. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.replace(buildModelPath, "")+"/", newDataSource);
  538. // 修改data.fdage
  539. String data = FileUtils.readFile(newDataSource + "/data.fdage");
  540. JSONObject jsonObject = JSONObject.parseObject(data);
  541. if(ObjectUtils.isEmpty(jsonObject)){
  542. log.error("data.fdage文件不存在");
  543. throw new BusinessException(-1,"拷贝场景出错,data.fdage文件不存在");
  544. }
  545. jsonObject.put("uuidtime",time);
  546. FileUtils.writeFile(newDataSource + "/data.fdage", jsonObject.toJSONString());
  547. ShellUtil.yunUpload(newDataSource, NacosProperty.bucket.concat("/" + ConstantFilePath.OSS_PREFIX + newDataSource.replace(buildModelPath, "")));
  548. // 复制计算结果
  549. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.concat("_results/").replace(buildModelPath, ""), newDataSource.concat("_results"));
  550. if(new File(newDataSource.concat("_results")).exists()){
  551. ShellUtil.yunUpload(newDataSource.concat("_results"),NacosProperty.bucket.concat("/" + ConstantFilePath.OSS_PREFIX + newDataSource.concat("_results").replace(buildModelPath, "")));
  552. FileUtils.delAllFile(newDataSource.concat("_results"));
  553. }
  554. FileUtils.delAllFile(newDataSource);
  555. FileUtils.copyFolderAllFiles(oldDataSource+"/",newDataSource+"/", true);
  556. }
  557. @Override
  558. public String setDataSource(String preDataSource,String time) throws Exception {
  559. SnowflakeIdGenerator snowflakeIdGenerator = new SnowflakeIdGenerator(0,1);
  560. String[] datasource = preDataSource.split("/");
  561. datasource[4] = snowflakeIdGenerator.nextId()+"";
  562. datasource[5] = datasource[5].split("_")[0] + "_" + time;
  563. return Arrays.stream(datasource).collect(Collectors.joining("/"));
  564. }
  565. @Override
  566. public String setVideos(String videos,String oldNum,String newNum) {
  567. JSONObject object = new JSONObject();
  568. if(StringUtils.isNotEmpty(videos)){
  569. object = JSONObject.parseObject(videos);
  570. if(object.containsKey("upPath")){
  571. String upPath = object.getString("upPath");
  572. upPath = upPath.replace(oldNum,newNum);
  573. object.put("upPath",upPath);
  574. }
  575. log.info("更新 scene.json");
  576. }
  577. return object.toJSONString();
  578. }
  579. private void reloadFile(String filePath,String sceneNum, String newNum) throws Exception {
  580. // 修改link-scene.json
  581. String content = FileUtils.readFile(filePath);
  582. if (!ObjectUtils.isEmpty(content)) {
  583. content = content.replaceAll(sceneNum, newNum);
  584. FileUtils.writeFile(filePath,content);
  585. }
  586. }
  587. /**
  588. * 上传新场景资源到oss
  589. * @param newNum
  590. */
  591. private void uploadNewSceneToOss(String newNum) {
  592. ShellUtil.yunUpload("/mnt/4Dkankan/scene/images/images" + newNum, NacosProperty.bucket.concat("/images/images") + newNum);
  593. ShellUtil.yunUpload("/mnt/4Dkankan/scene/data/data" + newNum, NacosProperty.bucket.concat("/data/data") + newNum);
  594. ShellUtil.yunUpload("/mnt/4Dkankan/scene/voice/voice" + newNum, NacosProperty.bucket.concat("/voice/voice") + newNum);
  595. ShellUtil.yunUpload("/mnt/4Dkankan/scene/video/video" + newNum, NacosProperty.bucket.concat("/video/video") + newNum);
  596. }
  597. @Override
  598. public void delete(String sceneNum) {
  599. if(StringUtils.isEmpty(sceneNum)){
  600. throw new BusinessException(LoginConstant.FAILURE_CODE_3001,LoginConstant.FAILURE_MSG_3001);
  601. }
  602. String[] nums = sceneNum.split(",");
  603. List<String> numList = Arrays.asList(nums);
  604. List<ScenePro> proList = this.getListByNums(numList);
  605. List<ScenePlus> plusList = scenePlusService.getListByNums(numList);
  606. this.deleteByList(proList,plusList);
  607. }
  608. @Override
  609. public void deleteByList(List<ScenePro> proList,List<ScenePlus> plusList) {
  610. HashMap<Long ,Long >cameraMap = new HashMap<>();
  611. if(proList.size() >0){
  612. for (ScenePro scenePro : proList) {
  613. cameraMap.merge(scenePro.getCameraId(), scenePro.getSpace(), Long::sum);
  614. }
  615. List<Long> proIds = proList.parallelStream().map(ScenePro::getId).collect(Collectors.toList());
  616. this.removeByIds(proIds);
  617. }
  618. if(plusList.size() >0){
  619. List<Long> plusIds = plusList.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
  620. HashMap<Long, ScenePlusExt> plusMap = scenePlusExtService.getByPlusIds(plusIds);
  621. for (ScenePlus scenePlus : plusList) {
  622. cameraMap.merge(scenePlus.getCameraId(), plusMap.get(scenePlus.getId()).getSpace(), Long::sum);
  623. }
  624. scenePlusService.removeByIds(plusIds);
  625. }
  626. //恢复相机使用容量
  627. if(cameraMap.size() >0){
  628. cameraDetailService.addUsedSpace(cameraMap);
  629. }
  630. }
  631. @Override
  632. public ScenePro getByUnicode(String unicode) {
  633. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  634. wrapper.like(ScenePro::getDataSource,unicode);
  635. wrapper.eq(ScenePro::getIsUpgrade,0);
  636. wrapper.orderByDesc(ScenePro::getCreateTime);
  637. List<ScenePro> list = this.list(wrapper);
  638. if(list !=null && list.size() >0){
  639. return list.get(0);
  640. }
  641. return null;
  642. }
  643. @Override
  644. public SceneInfoVo getInfo(String num) {
  645. if(StringUtils.isEmpty(num)){
  646. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  647. }
  648. ScenePro scenePro = this.getByNum(num);
  649. ScenePlus scenePlus = scenePlusService.getByNum(num);
  650. if(scenePro == null && scenePlus == null){
  651. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  652. }
  653. if(scenePro!=null && scenePro.getStatus() != 1 && scenePro.getStatus() != -2){
  654. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  655. }
  656. if(scenePlus!=null && scenePlus.getSceneStatus() != 1 && scenePlus.getSceneStatus() != -2){
  657. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  658. }
  659. if(scenePro!=null && scenePro.getPayStatus() != 1){
  660. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  661. }
  662. if(scenePlus!=null && scenePlus.getPayStatus() != 1){
  663. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  664. }
  665. SceneInfoVo infoVo = new SceneInfoVo();
  666. if(scenePro!=null){
  667. SceneProEdit sceneProEdit = sceneProEditService.getByProId(scenePro.getId());
  668. if(sceneProEdit !=null){
  669. BeanUtils.copyProperties(sceneProEdit, infoVo);
  670. }
  671. BeanUtils.copyProperties(scenePro,infoVo);
  672. if(StringUtils.isNotEmpty(scenePro.getGps())){
  673. infoVo.setGps(JSONObject.parseObject(scenePro.getGps()).toJSONString());
  674. }
  675. infoVo.setSceneKey("");
  676. infoVo.setCreateDate(DateUserUtil.getDate(scenePro.getCreateTime()).getTime());
  677. }else {
  678. SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
  679. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
  680. if(sceneEditInfo != null){
  681. BeanUtils.copyProperties(sceneEditInfo,infoVo);
  682. }
  683. if(scenePlusExt !=null){
  684. BeanUtils.copyProperties(scenePlusExt,infoVo);
  685. }
  686. BeanUtils.copyProperties(scenePlus,infoVo);
  687. infoVo.setSceneName(scenePlus.getTitle());
  688. infoVo.setStatus(scenePlus.getSceneStatus());
  689. }
  690. if(StringUtils.isNotEmpty(infoVo.getEntry())){
  691. infoVo.setEntry(JSONObject.parseObject(infoVo.getEntry()).toJSONString());
  692. }
  693. if(StringUtils.isEmpty(infoVo.getSceneKey())){
  694. infoVo.setIsPublic(0);
  695. }else {
  696. infoVo.setIsPublic(1);
  697. }
  698. //查询是否有随心装场景
  699. if(num.contains("vr-")){
  700. infoVo.setVrNum(null);
  701. infoVo.setVideosUser(null);
  702. infoVo.setBgMusicName(null);
  703. infoVo.setBgMusic(null);
  704. }
  705. this.updateViewCount(num);
  706. return infoVo;
  707. }
  708. @Override
  709. public void updateViewCount(String num) {
  710. String redisKey = RedisKey.SCENE_VISIT_CNT;
  711. if(!redisUtil.hHasKey(redisKey,num)){
  712. Integer count = 0;
  713. ScenePro pro = this.getByNum(num);
  714. if(pro !=null){
  715. count = pro.getViewCount();
  716. }else {
  717. ScenePlus plus = scenePlusService.getByNum(num);
  718. if(plus !=null){
  719. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(plus.getId());
  720. count = scenePlusExt.getViewCount();
  721. }
  722. }
  723. redisUtil.hset(redisKey,num,String.valueOf(count));
  724. }
  725. redisUtil.hincr(redisKey,num,1);
  726. }
  727. @Override
  728. public void updateDbViewCount(String num, String count) {
  729. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  730. wrapper.eq(ScenePro::getNum,num);
  731. wrapper.set(ScenePro::getViewCount,count);
  732. this.update(wrapper);
  733. ScenePlus plus = scenePlusService.getByNum(num);
  734. if(plus !=null){
  735. LambdaUpdateWrapper<ScenePlusExt> exWr = new LambdaUpdateWrapper<>();
  736. exWr.eq(ScenePlusExt::getPlusId,plus.getId())
  737. .set(ScenePlusExt::getViewCount,count);
  738. scenePlusExtService.update(exWr);
  739. }
  740. }
  741. @Override
  742. public ScenePro findByFileId(String fileId) {
  743. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  744. wrapper.like(ScenePro::getDataSource,fileId);
  745. List<ScenePro> list = this.list(wrapper);
  746. if(list !=null && list.size() >0){
  747. return list.get(0);
  748. }
  749. return null;
  750. }
  751. @Override
  752. public void setQrCode(String buildType,String num) throws Exception {
  753. String basePath = mainUrl;
  754. if("V2".equals(buildType)){
  755. basePath += sceneProUrl;
  756. }
  757. if("V3".equals(buildType)){
  758. basePath += sceneProNewUrl;
  759. }
  760. MatrixToImageWriterUtil.createQRCode(basePath + num, ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+".png", false,null);
  761. MatrixToImageWriterUtil.createQRCode(basePath + num + "&lang=en", ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+"_en.png", false,null);
  762. }
  763. @Override
  764. public void generateObjFile(String num) {
  765. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  766. wrapper.eq(ScenePro::getNum,num).eq(ScenePro::getRecStatus,"A");
  767. ScenePro sceneProEntity = this.getOne(wrapper);
  768. if(ObjectUtils.isEmpty(sceneProEntity)){
  769. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  770. }
  771. if(sceneProEntity.getSceneSource().intValue() != 4){
  772. throw new BusinessException(LoginConstant.FAILURE_CODE_3003, "只能操作激光场景");
  773. }
  774. // 拷贝文件
  775. String path = sceneProEntity.getDataSource();
  776. if (!new File(path + "/caches/reconstruction/final.bin").exists()
  777. || !new File(path + "/caches/reconstruction/chunk.json").exists()
  778. || !new File(path + "/caches/images").exists()
  779. || !new File(path + "/caches/depthmap").exists()
  780. || !new File(path + "/caches/depthmap_csc").exists()
  781. || !new File(path + "/caches/panorama.json").exists()
  782. || !new File(path + "/results/laserData/laser.ply").exists()) {
  783. log.error("生成obj缺少必要文件,生成失败!");
  784. throw new BusinessException(SceneConstant.FAILURE_CODE_5038, SceneConstant.FAILURE_MSG_5038);
  785. }
  786. // 获取最新的场景名称
  787. JSONObject sceneInfo = fdkkLaserService.getSceneByNum(sceneProEntity.getNum());
  788. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  789. updateWrapper
  790. .set(ScenePro::getStatus, 0)
  791. .set(ScenePro::getIsObj, 1)
  792. .set(ScenePro::getSceneName, sceneInfo.getString("title"))
  793. .eq(ScenePro::getNum, sceneProEntity.getNum());
  794. this.update(updateWrapper);
  795. //同步到scenePlus、scenePlus
  796. sceneUpgradeMapper.deleteScenePlus(sceneProEntity.getId());
  797. sceneUpgradeMapper.transferScenePlus(sceneProEntity.getId());
  798. sceneUpgradeMapper.deleteScenePlusExt(sceneProEntity.getId());
  799. String sceneKind = sceneProEntity.getSceneScheme() == 3 ? SceneKind.FACE.code(): SceneKind.TILES.code();
  800. sceneUpgradeMapper.transferScenePlusExt(sceneProEntity.getId(), sceneKind);
  801. log.info("开始发送激光场景生成obj mq消息");
  802. Camera camera = cameraService.getById(sceneProEntity.getCameraId());
  803. // 发送MQ
  804. BuildSceneCallMessage mqMsg = new BuildSceneCallMessage();
  805. mqMsg.setSceneNum(sceneProEntity.getNum());
  806. mqMsg.setUserId(sceneProEntity.getUserId());
  807. mqMsg.setAlgorithm(sceneProEntity.getAlgorithm());
  808. mqMsg.setBuildType(sceneProEntity.getBuildType());
  809. mqMsg.setDataSource(sceneProEntity.getDataSource());
  810. mqProducer.sendByWorkQueue(queueObjModelingPre,mqMsg);
  811. }
  812. }