SceneProServiceImpl.java 36 KB

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