SceneProServiceImpl.java 55 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222
  1. package com.fdkankan.ucenter.service.impl;
  2. import cn.hutool.core.io.FileUtil;
  3. import cn.hutool.json.JSONUtil;
  4. import com.alibaba.fastjson.JSONObject;
  5. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  6. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  7. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  8. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  9. import com.fdkankan.common.constant.SceneConstant;
  10. import com.fdkankan.common.constant.SceneKind;
  11. import com.fdkankan.common.exception.BusinessException;
  12. import com.fdkankan.common.util.*;
  13. import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
  14. import com.fdkankan.rabbitmq.util.RabbitMqProducer;
  15. import com.fdkankan.redis.constant.RedisKey;
  16. import com.fdkankan.ucenter.common.OssPath;
  17. import com.fdkankan.ucenter.common.PageInfo;
  18. import com.fdkankan.fyun.face.FYunFileServiceInterface;
  19. import com.fdkankan.image.MatrixToImageWriterUtil;
  20. import com.fdkankan.ucenter.common.constants.ConstantFilePath;
  21. import com.fdkankan.redis.util.RedisUtil;
  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.ResponseScene;
  32. import com.fdkankan.ucenter.vo.request.RequestScene;
  33. import com.fdkankan.ucenter.vo.request.SceneParam;
  34. import com.fdkankan.ucenter.vo.response.GroupByCount;
  35. import com.fdkankan.ucenter.vo.response.SceneInfoVo;
  36. import com.fdkankan.ucenter.vo.response.SceneNumVo;
  37. import com.fdkankan.ucenter.vo.response.SceneVo;
  38. import java.io.File;
  39. import java.util.ArrayList;
  40. import java.util.Arrays;
  41. import java.util.Collections;
  42. import java.util.Date;
  43. import java.util.HashMap;
  44. import java.util.List;
  45. import java.util.Map;
  46. import java.util.concurrent.CompletableFuture;
  47. import java.util.concurrent.locks.Condition;
  48. import java.util.stream.Collectors;
  49. import javax.annotation.Resource;
  50. import lombok.extern.slf4j.Slf4j;
  51. import org.apache.commons.lang3.StringUtils;
  52. import org.joda.time.DateTime;
  53. import org.springframework.beans.BeanUtils;
  54. import org.springframework.beans.factory.annotation.Autowired;
  55. import org.springframework.beans.factory.annotation.Value;
  56. import org.springframework.stereotype.Service;
  57. import org.springframework.util.ObjectUtils;
  58. /**
  59. * <p>
  60. * pro场景表 服务实现类
  61. * </p>
  62. *
  63. * @author
  64. * @since 2022-07-04
  65. */
  66. @Service
  67. @Slf4j
  68. public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro> implements ISceneProService {
  69. @Autowired
  70. private IUserService userService;
  71. @Autowired
  72. private ISceneCooperationService sceneCooperationService;
  73. @Autowired
  74. private IScenePlusService scenePlusService;
  75. @Autowired
  76. private ISceneService sceneService;
  77. @Autowired
  78. private LaserService fdkkLaserService;
  79. @Autowired
  80. private FYunFileServiceInterface fYunFileService;
  81. @Autowired
  82. private IUserIncrementService userIncrementService;
  83. @Autowired
  84. private IIncrementTypeService incrementTypeService;
  85. @Autowired
  86. private IScenePlusExtService scenePlusExtService;
  87. @Autowired
  88. private ISceneEditInfoService sceneEditInfoService;
  89. @Autowired
  90. private ICameraDetailService cameraDetailService;
  91. @Autowired
  92. ICameraService cameraService;
  93. @Autowired
  94. private IFolderService folderService;
  95. @Autowired
  96. private IFolderSceneService folderSceneService;
  97. @Autowired
  98. private ISceneCopyLogService sceneCopyLogService;
  99. @Autowired
  100. private IScene3dNumService scene3dNumService;
  101. @Autowired
  102. private ISceneProEditService sceneProEditService;
  103. @Autowired
  104. private RedisUtil redisUtil;
  105. @Autowired
  106. private ISceneUpgradeMapper sceneUpgradeMapper;
  107. @Autowired
  108. private IInnerService innerService;
  109. @Autowired
  110. LaserService laserService;
  111. @Value("${main.url}")
  112. private String mainUrl;
  113. @Value("${scene.pro.url}")
  114. private String sceneProUrl;
  115. @Value("${scene.pro.new.url}")
  116. private String sceneProNewUrl;
  117. @Value("${queue.modeling.obj.modeling-pre}")
  118. private String queueObjModelingPre;
  119. @Autowired
  120. private RabbitMqProducer mqProducer;
  121. @Override
  122. public HashMap<String, SceneNumVo> findSceneNumber(String token) {
  123. User user = userService.getByUserName(JwtUtil.getUsername(token));
  124. SceneNumVo sceneNumVoKk = getSceneNumVoByType(Arrays.asList(1,2,12,13,14), user.getId());
  125. Integer oldNum = sceneService.getCountByUserId(user.getId());
  126. sceneNumVoKk.setSceneNum( oldNum + sceneNumVoKk.getSceneNum());
  127. sceneNumVoKk.setTotalNum(oldNum + sceneNumVoKk.getTotalNum());
  128. SceneNumVo sceneNumVoKj = getSceneNumVoByType(Collections.singletonList(3), user.getId());
  129. SceneNumVo sceneNumVoSS = fdkkLaserService.getLaserSceneNumByUser(token);
  130. SceneNumVo sceneNumVoSSObj = getSceneNumVoByObjType(Arrays.asList(4), user.getId());
  131. sceneNumVoSS.setSceneNum(sceneNumVoSSObj.getSceneNum() + sceneNumVoSS.getSceneNum());
  132. sceneNumVoSS.setTotalNum(sceneNumVoSSObj.getTotalNum() + sceneNumVoSS.getTotalNum());
  133. HashMap<String, SceneNumVo> hashMap = new HashMap<>();
  134. hashMap.put("kk",sceneNumVoKk);
  135. hashMap.put("kJ",sceneNumVoKj);
  136. hashMap.put("SS",sceneNumVoSS);
  137. return hashMap;
  138. }
  139. private SceneNumVo getSceneNumVoByType(List<Integer> sceneSourceList, Long userId){
  140. SceneNumVo sceneNumVo = new SceneNumVo();
  141. LambdaQueryWrapper<ScenePro> queryWrapper = new LambdaQueryWrapper<>();
  142. queryWrapper.eq(ScenePro::getUserId, userId);
  143. queryWrapper.in(ScenePro::getSceneSource, sceneSourceList);
  144. queryWrapper.eq(ScenePro::getIsUpgrade, 0);
  145. Integer sceneNum = Math.toIntExact(this.count(queryWrapper));
  146. Long count = scenePlusService.getCountByUserId(userId,sceneSourceList);
  147. sceneNumVo.setCooperationSceneNum(sceneCooperationService.getCooperationSceneNum(userId,sceneSourceList));
  148. sceneNumVo.setSceneNum(sceneNum + count);
  149. sceneNumVo.setTotalNum(sceneNumVo.getSceneNum() + sceneNumVo.getCooperationSceneNum());
  150. return sceneNumVo;
  151. }
  152. private SceneNumVo getSceneNumVoByObjType(List<Integer> sceneSourceList, Long userId){
  153. SceneNumVo sceneNumVo = new SceneNumVo();
  154. LambdaQueryWrapper<ScenePro> queryWrapper = new LambdaQueryWrapper<>();
  155. queryWrapper.eq(ScenePro::getUserId, userId);
  156. queryWrapper.in(ScenePro::getSceneSource, sceneSourceList);
  157. queryWrapper.eq(ScenePro::getIsUpgrade, 0);
  158. queryWrapper.eq(ScenePro::getIsObj,1);
  159. Integer sceneNum = Math.toIntExact(this.count(queryWrapper));
  160. LambdaQueryWrapper<ScenePlusExt> wrapper = new LambdaQueryWrapper<>();
  161. wrapper.eq(ScenePlusExt::getIsObj,1);
  162. List<ScenePlusExt> list = scenePlusExtService.list(wrapper);
  163. long plusCount = list.size();
  164. sceneNumVo.setCooperationSceneNum(0L);
  165. sceneNumVo.setSceneNum(sceneNum + plusCount);
  166. sceneNumVo.setTotalNum(sceneNumVo.getSceneNum() + sceneNumVo.getCooperationSceneNum());
  167. return sceneNumVo;
  168. }
  169. @Override
  170. public HashMap<Long,GroupByCount> findSceneNumByCameraIds(List<Long> cameraIdList) {
  171. HashMap<Long,GroupByCount> map = new HashMap<>();
  172. List<GroupByCount> result = this.getBaseMapper().findSceneProNumByCameraIds(cameraIdList);
  173. List<GroupByCount> result2 = this.getBaseMapper().findScenePlusNumByCameraIds(cameraIdList);
  174. result.forEach(entity ->map.put(entity.getId(),entity));
  175. for (GroupByCount groupByCount : result2) {
  176. if (map.get(groupByCount.getId()) != null) {
  177. Long totalCount = map.get(groupByCount.getId()).getCount() + groupByCount.getCount();
  178. String lastTime = DateUserUtil.getLastTime(map.get(groupByCount.getId()).getLastTime(), groupByCount.getLastTime());
  179. groupByCount.setCount(totalCount);
  180. groupByCount.setLastTime(lastTime);
  181. }
  182. map.put(groupByCount.getId(),groupByCount);
  183. }
  184. return map;
  185. }
  186. /**
  187. * @param payStatus -2 封存,为 1 解封
  188. */
  189. @Override
  190. public void lockOrUnLockBySpace(CameraDetail cameraDetail, Long cameraId, Integer payStatus) {
  191. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  192. LambdaQueryWrapper<ScenePlus> plusWr = new LambdaQueryWrapper<>();
  193. Long totalSpace = cameraDetail.getTotalSpace();
  194. UserIncrement userIncrement = userIncrementService.getByCameraId(cameraId);
  195. if(userIncrement!=null){
  196. IncrementType incrementType = incrementTypeService.getById(userIncrement.getIncrementTypeId());
  197. if(incrementType!=null){
  198. if(incrementType.getCameraCapacity() == -1){
  199. totalSpace = -1L;
  200. }else {
  201. totalSpace = incrementType.getCameraCapacity() * 1024 * 1024L;
  202. }
  203. }
  204. }
  205. log.info("lockOrUnLockBySpace---cameraId:{},payStatus:{},totalSpace:{},useSpace{}",
  206. cameraId,payStatus,totalSpace,cameraDetail.getUsedSpace());
  207. if(payStatus == 1){ //解封,判断用户权益,用户会员权益无限容量
  208. wrapper.orderByAsc(ScenePro::getCreateTime);
  209. plusWr.orderByAsc(ScenePlus::getCreateTime);
  210. wrapper.eq(ScenePro::getPayStatus,-2);
  211. plusWr.eq(ScenePlus::getPayStatus,-2);
  212. }else {
  213. if (totalSpace >= cameraDetail.getUsedSpace()) {
  214. // 总容量大于已使用容量,不予封存
  215. return;
  216. }
  217. wrapper.orderByDesc(ScenePro::getCreateTime);
  218. plusWr.orderByDesc(ScenePlus::getCreateTime);
  219. wrapper.eq(ScenePro::getPayStatus,1);
  220. plusWr.eq(ScenePlus::getPayStatus,1);
  221. }
  222. wrapper.eq(ScenePro::getCameraId,cameraId)
  223. .eq(ScenePro::getIsUpgrade,0);
  224. plusWr.eq(ScenePlus::getCameraId,cameraId);
  225. List<ScenePro> list = this.list(wrapper);
  226. List<ScenePlus> plusList = scenePlusService.list(plusWr);
  227. Long beyondSpace = Math.abs(cameraDetail.getUsedSpace() - totalSpace);
  228. Long accumulateSpace = 0L;
  229. List<Long> lockedIds = new ArrayList<>();
  230. if(payStatus == 1){
  231. Long doSpace = getScenePlusLockedIds(lockedIds, plusList, totalSpace, beyondSpace, accumulateSpace);
  232. beyondSpace -= doSpace;
  233. getSceneLockedIds(lockedIds,list,totalSpace,beyondSpace,accumulateSpace);
  234. }else {
  235. Long doSpace = getSceneLockedIds(lockedIds, list, totalSpace, beyondSpace, accumulateSpace);
  236. beyondSpace -= doSpace;
  237. getScenePlusLockedIds(lockedIds,plusList,totalSpace,beyondSpace,accumulateSpace);
  238. }
  239. lockOrUnLockScenes(lockedIds,payStatus);
  240. }
  241. private Long getSceneLockedIds(List<Long> lockedIds ,List<ScenePro> list,Long count,Long beyondSpace,Long accumulateSpace){
  242. if (list != null && list.size() > 0){
  243. for (ScenePro scenePro : list){
  244. accumulateSpace += scenePro.getSpace();
  245. if (count ==-1 && accumulateSpace.compareTo(beyondSpace) > 0){
  246. return accumulateSpace - scenePro.getSpace();
  247. }
  248. lockedIds.add(scenePro.getId());
  249. }
  250. }
  251. return accumulateSpace;
  252. }
  253. private Long getScenePlusLockedIds(List<Long> lockedIds , List<ScenePlus> list, Long count, Long beyondSpace, Long accumulateSpace){
  254. if (list != null && list.size() > 0){
  255. List<Long> plusIds = list.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
  256. HashMap<Long, ScenePlusExt> byPlusIds = scenePlusExtService.getByPlusIds(plusIds);
  257. for (ScenePlus scenePlus : list){
  258. ScenePlusExt scenePlusExt = byPlusIds.get(scenePlus.getId());
  259. accumulateSpace += scenePlusExt.getSpace();
  260. if (count ==-1 && accumulateSpace.compareTo(beyondSpace) > 0){
  261. return accumulateSpace - scenePlusExt.getSpace();
  262. }
  263. lockedIds.add(scenePlus.getId());
  264. }
  265. }
  266. return accumulateSpace;
  267. }
  268. // payStatus 为 -2 封存,为 1 解封
  269. private void lockOrUnLockScenes(List<Long> lockedIds,Integer payStatus) {
  270. if (lockedIds == null || lockedIds.size() == 0){
  271. return;
  272. }
  273. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  274. updateWrapper.set(ScenePro::getPayStatus,payStatus)
  275. .eq(ScenePro::getIsUpgrade,0)
  276. .in(ScenePro::getId,lockedIds);
  277. this.update(updateWrapper);
  278. LambdaUpdateWrapper<ScenePlus> updatePlusWrapper = new LambdaUpdateWrapper<>();
  279. updatePlusWrapper.set(ScenePlus::getPayStatus,payStatus)
  280. .in(ScenePlus::getId,lockedIds);
  281. scenePlusService.update(updatePlusWrapper);
  282. this.updateOssStatus(lockedIds,payStatus);
  283. }
  284. /**
  285. * 修改oss status.json中 payStatus
  286. */
  287. private void updateOssStatus(List<Long> lockedIds, Integer payStatus) {
  288. LambdaQueryWrapper<ScenePro> proWr = new LambdaQueryWrapper<>();
  289. proWr.eq(ScenePro::getIsUpgrade,0);
  290. proWr.in(ScenePro::getId,lockedIds);
  291. List<ScenePro> proList = this.list(proWr);
  292. LambdaQueryWrapper<ScenePlus> pluWr = new LambdaQueryWrapper<>();
  293. pluWr.in(ScenePlus::getId,lockedIds);
  294. List<ScenePlus> plusList = scenePlusService.list(pluWr);
  295. for (ScenePro scenePro : proList) {
  296. this.updateOssStatus(String.format(OssPath.v3_statusPath,scenePro.getNum()),payStatus);
  297. }
  298. for (ScenePlus scenePlus : plusList) {
  299. this.updateOssStatus(String.format(OssPath.v4_statusPath,scenePlus.getNum()),payStatus);
  300. }
  301. }
  302. /**
  303. * 从oss中获取文件,并重写,上传替换
  304. */
  305. private void updateOssStatus(String path,Integer payStatus) {
  306. try {
  307. if(!fYunFileService.fileExist(path)){
  308. return;
  309. }
  310. String data = fYunFileService.getFileContent(path);
  311. if(StringUtils.isBlank(data)){
  312. return;
  313. }
  314. JSONObject jsonObject = JSONObject.parseObject(data);
  315. jsonObject.put("payStatus",payStatus);
  316. String json = JSONUtil.toJsonStr(jsonObject);
  317. FileUtils.writeFile(OssPath.localStatusPath ,json);
  318. log.info("updateOssStatus--localPath:{},ossPath:{}",OssPath.localStatusPath,path);
  319. fYunFileService.uploadFile(OssPath.localStatusPath,path);
  320. }catch (Exception e){
  321. e.printStackTrace();
  322. }finally {
  323. FileUtil.del(OssPath.localStatusPath);
  324. }
  325. }
  326. @Override
  327. public List<ScenePro> getListByCameraId(Long cameraId) {
  328. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  329. wrapper.eq(ScenePro::getCameraId,cameraId)
  330. .eq(ScenePro::getIsUpgrade,0);
  331. return this.list(wrapper);
  332. }
  333. @Override
  334. public void bindOrUnCamera(List<Long> cameraIds, Long userId) {
  335. if(cameraIds.size() >0){
  336. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  337. wrapper.in(ScenePro::getCameraId,cameraIds)
  338. .eq(ScenePro::getIsUpgrade,0)
  339. .set(ScenePro::getUserId,userId);
  340. this.update(wrapper);
  341. }
  342. }
  343. @Override
  344. public List<ScenePro> getListByCameraIds(List<Long> cameraIds) {
  345. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  346. wrapper.in(ScenePro::getCameraId,cameraIds)
  347. .eq(ScenePro::getIsUpgrade,0);
  348. return this.list(wrapper);
  349. }
  350. @Override
  351. public List<ScenePro> getListByNums(List<String> numList) {
  352. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  353. wrapper.in(ScenePro::getNum,numList)
  354. .eq(ScenePro::getIsUpgrade,0);
  355. return this.list(wrapper);
  356. }
  357. @Override
  358. public Long getCountByUserId(Long userId, List<?> resourceList) {
  359. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  360. wrapper.eq(ScenePro::getUserId,userId);
  361. wrapper.eq(ScenePro::getIsUpgrade,0);
  362. wrapper.in(ScenePro::getSceneSource,resourceList);
  363. return this.count(wrapper);
  364. }
  365. @Override
  366. public Long getCountByUserId(Long id, Integer cameraType) {
  367. return this.getBaseMapper().getCountByUserId(id,cameraType);
  368. }
  369. @Override
  370. public void deleteByIds(List<Long> sceneIds) {
  371. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  372. wrapper.in(ScenePro::getId,sceneIds)
  373. .eq(ScenePro::getIsUpgrade,0);
  374. List<ScenePro> proList = this.list(wrapper);
  375. LambdaQueryWrapper<ScenePlus> plusWr = new LambdaQueryWrapper<>();
  376. plusWr.in(ScenePlus::getId,sceneIds);
  377. List<ScenePlus> plusList = scenePlusService.list(plusWr);
  378. this.deleteByList(proList,plusList);
  379. }
  380. @Override
  381. public Page<SceneVo> pageListAndFolder(Page<Object> page, SceneParam param) {
  382. return getBaseMapper().pageListAndFolder(new Page<>(param.getPageNum(),param.getPageSize()),param);
  383. }
  384. @Override
  385. public Object getScenes(String username, RequestScene param) {
  386. User user = userService.getByUserName(username);
  387. log.info("搜索条件是:"+ param.getSearchKey());
  388. //type为12表示一键换装的请求,不查询相机数据
  389. if(StringUtils.isNotEmpty(param.getSearchKey()) && !"11".equals(param.getType())){
  390. List<Long> userIds = userService.getLikeUserName(param.getSearchKey());
  391. if(userIds.size() > 0){
  392. List<String> cooperationList = sceneCooperationService.getNumByUserIds(userIds);
  393. param.setNumList(cooperationList);
  394. }
  395. }
  396. param.setUserId(user.getId());
  397. Page<SceneVo> sceneList = sceneService.getSceneList(param);
  398. for (SceneVo record : sceneList.getRecords()) {
  399. record.setUserName(username);
  400. SceneCooperation sceneCooperationEntity = sceneCooperationService.getByNum(record.getNum());
  401. if(sceneCooperationEntity != null && sceneCooperationEntity.getUserId()!= null){
  402. User user1 = userService.getById(sceneCooperationEntity.getUserId());
  403. if(user1 != null){
  404. record.setCooperationUserId(String.valueOf(sceneCooperationEntity.getUserId()));
  405. record.setCooperationUserName(user1.getUserName());
  406. }
  407. }
  408. }
  409. return PageInfo.PageInfo(sceneList);
  410. }
  411. @Override
  412. public JSONObject newList(SceneParam param, String username) {
  413. User user = userService.getByUserName(username);
  414. param.setSourceList(getSceneSource(param.getSceneSource()));
  415. param.setUserId(user.getId());
  416. Page<SceneVo> sceneVoPage = getBaseMapper().pageListAndFolder(new Page<>(param.getPageNum(),param.getPageSize()),param);
  417. List<SceneVo> folderList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 1).collect(Collectors.toList());
  418. Long totalSceneNum = 0L;
  419. if(folderList.size() >0){
  420. List<Long> folderIds = folderList.parallelStream().map(SceneVo::getId).collect(Collectors.toList());
  421. List<Folder> allFolderIds = folderService.getAllFolderIds(folderIds);
  422. HashMap<Long,Long> map = folderSceneService.getGroupByFolderIds(allFolderIds);
  423. for (SceneVo vo : sceneVoPage.getRecords()) {
  424. if(vo.getIsFolder() == 1 && vo.getType()!= null && vo.getType()!= 3){
  425. Long sceneNum = map.get(vo.getId()) == null ? 0L : map.get(vo.getId());
  426. vo.setSceneNum(sceneNum);
  427. vo.setCreateTimeStr(vo.getCreateTime());
  428. }else if(vo.getIsFolder() == 1 && vo.getType()!= null && vo.getType() == 3){
  429. Integer count = sceneService.getCountByUserId(user.getId());
  430. vo.setSceneNum(Long.valueOf(count));
  431. }
  432. }
  433. }
  434. //设置协作者信息
  435. if(sceneVoPage.getRecords().size() >0){
  436. List<String> numList = sceneVoPage.getRecords().parallelStream().map(SceneVo::getNum).collect(Collectors.toList());
  437. HashMap<String,User> cooMap = sceneCooperationService.getByNumList(numList);
  438. for (SceneVo vo : sceneVoPage.getRecords()) {
  439. if (StringUtils.isNotBlank(vo.getNum())) {
  440. User userVo = cooMap.get(vo.getNum());
  441. if (userVo != null) {
  442. vo.setCooperationUserId(userVo.getId().toString());
  443. vo.setCooperationUserName(userVo.getUserName());
  444. }
  445. }
  446. }
  447. }
  448. List<SceneVo> sceneList = sceneVoPage.getRecords().parallelStream().filter(entity -> entity.getIsFolder() == 0)
  449. .peek(vo->{
  450. if (ObjectUtils.isEmpty(vo.getStatus())) {
  451. return;
  452. }
  453. if (vo.getStatus().equals(500)) { //500状态为数据库中手动修改值,当场景本身异常,算法,算不了
  454. vo.setStatus(-1);
  455. }
  456. if (vo.getStatus().equals(-1)) { //计算失败,钉钉通知之后,判定是否重算,还是修改为 500
  457. vo.setStatus(0);
  458. }
  459. }).collect(Collectors.toList());
  460. if(sceneList.size() >0){
  461. folderList.addAll(sceneList);
  462. }
  463. sceneVoPage.setRecords(folderList);
  464. if(param.getFolderId() == null && param.getIsObj() == null ){
  465. List<String> sourceList = param.getSourceList();
  466. totalSceneNum = this.getCountByUserId(user.getId(),sourceList);
  467. totalSceneNum += scenePlusService.getCountByUserId(user.getId(),sourceList);
  468. totalSceneNum += sceneService.getCountByUserId(user.getId());
  469. }else if(param.getFolderId() != null ){
  470. totalSceneNum = folderSceneService.getCountByFolderAndSon(param.getFolderId());
  471. }else if(param.getIsObj() != null && param.getIsObj() ==1){
  472. totalSceneNum = this.getCountByLaserAndIsObj(user.getId());
  473. }
  474. JSONObject jsonObject = new JSONObject();
  475. jsonObject.put("pageInfo",PageInfo.PageInfo(sceneVoPage));
  476. jsonObject.put("sceneNum",totalSceneNum);
  477. return jsonObject;
  478. }
  479. private Long getCountByLaserAndIsObj(Long userId) {
  480. Long proCount = 0L;
  481. Long plusCount = 0L;
  482. LambdaQueryWrapper<ScenePro> proWr = new LambdaQueryWrapper<>();
  483. proWr.eq(ScenePro::getUserId,userId);
  484. proWr.eq(ScenePro::getSceneSource,4);
  485. proWr.eq(ScenePro::getIsObj,1);
  486. proWr.eq(ScenePro::getIsUpgrade,0);
  487. proCount = this.count(proWr);
  488. LambdaQueryWrapper<ScenePlus> plusWr = new LambdaQueryWrapper<>();
  489. plusWr.eq(ScenePlus::getUserId,userId);
  490. plusWr.eq(ScenePlus::getSceneSource,4);
  491. List<ScenePlus> list = scenePlusService.list(plusWr);
  492. if(list.size() >0){
  493. List<Long> plusIds = list.stream().map(ScenePlus::getId).collect(Collectors.toList());
  494. LambdaQueryWrapper<ScenePlusExt> extWr = new LambdaQueryWrapper<>();
  495. extWr.in(ScenePlusExt::getPlusId,plusIds);
  496. extWr.eq(ScenePlusExt::getIsObj,1);
  497. plusCount = scenePlusExtService.count(extWr);
  498. }
  499. return proCount +plusCount;
  500. }
  501. public List<String> getSceneSource(String sceneSource){
  502. List<String> sceneSourceList ;
  503. if(StringUtils.isNotBlank(sceneSource)){
  504. switch (sceneSource){
  505. case "1" : sceneSourceList = Arrays.asList("1","2","12","13","14"); break; //四维看看
  506. case "3" : sceneSourceList = Collections.singletonList("3"); break;//四维看见
  507. case "4" : sceneSourceList = Collections.singletonList("4"); break;//深时场景
  508. default:
  509. if(sceneSource.contains(",")){
  510. sceneSource = sceneSource + ",12,13,14";
  511. sceneSourceList = Arrays.asList(sceneSource.split(","));
  512. break;
  513. }
  514. sceneSourceList = Collections.singletonList(sceneSource);break;
  515. }
  516. return sceneSourceList;
  517. }
  518. return Arrays.asList("0","1","2","3","4","5","6","7","8","9","10","11","12","13","14","15");
  519. }
  520. @Override
  521. public void copyScene(String sceneNum,String userName) throws Exception {
  522. if(StringUtils.isEmpty(sceneNum)){
  523. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  524. }
  525. ScenePro scenePro = this.getByNum(sceneNum);
  526. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  527. if((scenePro == null || scenePro.getCameraId() == null ) && (scenePlus== null || scenePlus.getCameraId() == null)){
  528. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  529. }
  530. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  531. Long userId = scenePro == null ? scenePlus.getUserId() :scenePro.getUserId();
  532. // 判断是否有绑定会员权益
  533. Long count = userIncrementService.getValidCountByCameraId(cameraId);
  534. if(count <= 0){
  535. throw new BusinessException(LoginConstant.FAILURE_CODE_3030, LoginConstant.FAILURE_MSG_3030);
  536. }
  537. if(userId == null){
  538. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  539. }
  540. //只能复制自己的场景
  541. User user = userService.getByUserName(userName);
  542. if(user == null || !user.getId().equals(userId) ){
  543. throw new BusinessException(SceneConstant.FAILURE_CODE_5013, SceneConstant.FAILURE_MSG_5013);
  544. }
  545. this.copySceneNoCheck(sceneNum);
  546. }
  547. @Override
  548. public void copySceneNoCheck(String sceneNum) throws Exception {
  549. ScenePro scenePro = this.getByNum(sceneNum);
  550. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  551. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  552. CameraDetail detailEntity = cameraDetailService.getByCameraId(cameraId);
  553. if(detailEntity == null){
  554. throw new BusinessException(CameraConstant.FAILURE_CODE_6029, CameraConstant.FAILURE_MSG_6029);
  555. }
  556. String newNum = scene3dNumService.generateSceneNum(detailEntity.getType());
  557. String title = scenePro == null ? scenePlus.getTitle() : scenePro.getSceneName();
  558. String newTitle = title.concat("(copy)");
  559. Long sceneId = scenePro == null ? scenePlus.getId() :scenePro.getId();
  560. Long newSceneId = null;
  561. FolderScene folderScene = folderSceneService.getByType(sceneId, null);
  562. String time = DateUtil.date2String(new Date(),DateUtil.YYYYMMDDHHMMSSSSS_DATA_FORMAT);
  563. if(scenePro !=null){ //v3场景复制
  564. log.info("场景复制--V3--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  565. scenePro.getNum(),scenePro.getSceneName(),newNum,newTitle);
  566. scenePro.setSceneName(newTitle);
  567. newSceneId = this.copyV3Scene(scenePro, newNum, detailEntity,time);
  568. }
  569. if(scenePlus != null){ //v4场景复制
  570. log.info("场景复制--V4--OldNum:{},oldTitle:{},newNum:{},newTitle:{}",
  571. scenePlus.getNum(),scenePlus.getTitle(),newNum,newTitle);
  572. scenePlus.setTitle(newTitle);
  573. newSceneId = scenePlusService.copyV4Scene(scenePlus,newNum,detailEntity,time);
  574. }
  575. log.info("场景复制--完成--sceneId:{}",newSceneId);
  576. if(newSceneId != null && folderScene!= null){
  577. folderScene.setId(null);
  578. folderScene.setSceneId(newSceneId);
  579. folderSceneService.save(folderScene);
  580. }
  581. }
  582. @Override
  583. public ScenePro getByNum(String sceneNum) {
  584. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  585. wrapper.eq(ScenePro::getNum,sceneNum);
  586. wrapper.eq(ScenePro::getIsUpgrade,0);
  587. List<ScenePro> list = this.list(wrapper);
  588. if(list!= null && list.size() >0){
  589. return list.get(0);
  590. }
  591. return null;
  592. }
  593. @Override
  594. public Long copyV3Scene(ScenePro oldScene,String newNum,CameraDetail cameraDetail,String time) throws Exception {
  595. SceneProEdit oldEditScene = sceneProEditService.getByProId(oldScene.getId());
  596. String oldNum = oldScene.getNum();
  597. //复制数据库数据
  598. oldScene.setId(null);
  599. oldScene.setStatus(0);
  600. oldScene.setNum(newNum);
  601. oldScene.setWebSite(oldScene.getWebSite().replace(oldNum, newNum));
  602. oldScene.setThumb(oldScene.getThumb().replace(oldNum, newNum));
  603. oldScene.setVideos(this.setVideos(oldScene.getVideos(),oldNum,newNum));
  604. oldScene.setViewCount(0);
  605. String preDataSource = oldScene.getDataSource();
  606. String newDataSource = this.setDataSource(preDataSource,time);
  607. oldScene.setDataSource(newDataSource);
  608. this.save(oldScene);
  609. if(oldScene.getSceneSource() == 4){ //深时复制
  610. laserService.copy(oldScene.getCameraId(),oldScene.getCreateTime(),newNum,0,null,
  611. oldEditScene.getSceneKey(),oldScene.getSceneName(),oldScene.getUserId(),"V3");
  612. }
  613. //更新video
  614. Map map = new HashMap();
  615. JSONObject object = new JSONObject();
  616. if(StringUtils.isNotEmpty(oldScene.getVideos())){
  617. object = JSONObject.parseObject(oldScene.getVideos());
  618. if(object.containsKey("upPath")){
  619. String upPath = object.getString("upPath");
  620. upPath = upPath.replace(oldNum,newNum);
  621. object.put("upPath",upPath);
  622. }
  623. oldScene.setVideos(object.toJSONString());
  624. log.info("更新 scene.json");
  625. map.put("videos",oldScene.getVideos());
  626. }
  627. map.put("sceneName",oldScene.getSceneName());
  628. map.put("webSite",oldScene.getWebSite());
  629. map.put("thumb",oldScene.getThumb());
  630. map.put("num",oldScene.getNum());
  631. map.put("id",oldScene.getId());
  632. map.put("dataSource",oldScene.getDataSource());
  633. //复制完成更新相机容量
  634. oldEditScene.setId(null);
  635. oldEditScene.setProId(oldScene.getId());
  636. oldEditScene.setScreencapVoiceSrc(oldEditScene.getScreencapVoiceSrc() == null ? null : oldEditScene.getScreencapVoiceSrc().replace(oldNum, newNum));
  637. oldEditScene.setScreencapVoiceSound(oldEditScene.getScreencapVoiceSound() == null ? null : oldEditScene.getScreencapVoiceSound().replace(oldNum, newNum));
  638. oldEditScene.setScreencapVoiceSoundsync(oldEditScene.getScreencapVoiceSoundsync() == null ? null : oldEditScene.getScreencapVoiceSoundsync().replace(oldNum, newNum));
  639. oldEditScene.setPlayData(oldEditScene.getPlayData() == null ? null : oldEditScene.getPlayData().replace(oldNum, newNum));
  640. oldEditScene.setScreencapThumb(oldEditScene.getScreencapThumb() == null ? null : oldEditScene.getScreencapThumb().replace(oldNum, newNum));
  641. oldEditScene.setFloorPlanPng(oldEditScene.getFloorPlanPng() == null ? null : oldEditScene.getFloorPlanPng().replace(oldNum, newNum));
  642. sceneProEditService.save(oldEditScene);
  643. this.setQrCode(oldScene.getBuildType(), oldScene.getNum());
  644. CompletableFuture.runAsync(() -> {
  645. try {
  646. log.info("开始复制场景-{}", new Date());
  647. String buildModelPath = ConstantFilePath.BUILD_MODEL_PATH;
  648. if(oldScene.getSceneSource().equals(4)){
  649. buildModelPath = ConstantFilePath.BUILD_MODEL_LASER_PATH;
  650. }
  651. this.copyFdage(preDataSource,newDataSource,buildModelPath,time);
  652. //oss复制计算结果资源
  653. downloadFromOldNumFromOss(oldNum, newNum);
  654. // 复制本地secen.json
  655. copyOldSceneLocalToNew(oldNum, newNum);
  656. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  657. //上传资源到oss
  658. uploadNewSceneToOss(newNum);
  659. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  660. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  661. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  662. FileUtils.delFolder(ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  663. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"images/images" + oldNum,ConstantFilePath.SCENE_PATH +"images/images" + newNum);
  664. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"data/data" + oldNum,ConstantFilePath.SCENE_PATH +"data/data" + newNum);
  665. FileUtils.writeJsonFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + File.separator + "scene.json", map);
  666. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",oldNum, newNum);
  667. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",oldNum, newNum);
  668. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"voice/voice" + oldNum,ConstantFilePath.SCENE_PATH +"voice/voice" + newNum);
  669. FileUtils.copyDirectiory(ConstantFilePath.SCENE_PATH +"video/video" + oldNum,ConstantFilePath.SCENE_PATH +"video/video" + newNum);
  670. oldScene.setStatus(-2);
  671. this.updateById(oldScene);
  672. log.info("复制场景结束-{}", new Date());
  673. if(oldScene.getSceneSource() == 4){ //深时复制
  674. String laserPath = laserService.copyDataSource(preDataSource,oldScene.getDataSource());
  675. laserService.copy(oldScene.getCameraId(),oldScene.getCreateTime(),newNum,2,laserPath,
  676. oldEditScene.getSceneKey(),oldScene.getSceneName(),oldScene.getUserId(),"V3");
  677. }
  678. sceneCopyLogService.saveByNum(oldNum,newNum,oldScene.getUserId());
  679. } catch (Exception e) {
  680. this.removeById(oldScene.getId());
  681. log.error("复制场景异常", e);
  682. }
  683. });
  684. return oldScene.getId();
  685. }
  686. /**
  687. * 从旧场景下载资源
  688. * @param sceneNum
  689. * @param newNum
  690. * @throws Exception
  691. */
  692. private void downloadFromOldNumFromOss(String sceneNum, String newNum) {
  693. ShellUtil.yunDownload("images/images" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "images/images" + newNum);
  694. ShellUtil.yunDownload("data/data" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "data/data" + newNum);
  695. ShellUtil.yunDownload("voice/voice" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "voice/voice" + newNum);
  696. ShellUtil.yunDownload("video/video" + sceneNum + "/", ConstantFilePath.SCENE_PATH + "video/video" + newNum);
  697. }
  698. /**
  699. * 从本地旧场景拷贝资源到新场景
  700. * @param sceneNum
  701. * @param newNum
  702. */
  703. private void copyOldSceneLocalToNew(String sceneNum, String newNum) throws Exception {
  704. FileUtils.copyFolderAllFiles(ConstantFilePath.SCENE_PATH + "data/data" + sceneNum + "/",
  705. ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/", true);
  706. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/link-scene.json",sceneNum, newNum);
  707. reloadFile(ConstantFilePath.SCENE_PATH + "data/data" + newNum + "/hot.json",sceneNum, newNum);
  708. }
  709. public void copyFdage(String oldDataSource,String newDataSource,String buildModelPath,String time) throws Exception {
  710. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.replace(buildModelPath, "")+"/", newDataSource);
  711. // 修改data.fdage
  712. String data = FileUtils.readFile(newDataSource + "/data.fdage");
  713. JSONObject jsonObject = JSONObject.parseObject(data);
  714. if(ObjectUtils.isEmpty(jsonObject)){
  715. log.error("data.fdage文件不存在");
  716. throw new BusinessException(-1,"拷贝场景出错,data.fdage文件不存在");
  717. }
  718. jsonObject.put("uuidtime",time);
  719. FileUtils.writeFile(newDataSource + "/data.fdage", jsonObject.toJSONString());
  720. ShellUtil.yunUpload(newDataSource, ConstantFilePath.OSS_PREFIX + newDataSource.replace(buildModelPath, ""));
  721. // 复制计算结果
  722. ShellUtil.yunDownload(ConstantFilePath.OSS_PREFIX + oldDataSource.concat("_results/").replace(buildModelPath, ""), newDataSource.concat("_results"));
  723. if(new File(newDataSource.concat("_results")).exists()){
  724. ShellUtil.yunUpload(newDataSource.concat("_results"), ConstantFilePath.OSS_PREFIX + newDataSource.concat("_results").replace(buildModelPath, ""));
  725. FileUtils.delAllFile(newDataSource.concat("_results"));
  726. }
  727. FileUtils.delAllFile(newDataSource);
  728. try {
  729. FileUtils.copyFolderAllFiles(oldDataSource+"/",newDataSource+"/", true);
  730. }catch (Exception e){
  731. log.error("dataSource复制失败,{}不存在",oldDataSource);
  732. }
  733. }
  734. @Override
  735. public String setDataSource(String preDataSource,String time) throws Exception {
  736. SnowflakeIdGenerator snowflakeIdGenerator = new SnowflakeIdGenerator(0,1);
  737. String[] datasource = preDataSource.split("/");
  738. datasource[4] = snowflakeIdGenerator.nextId()+"";
  739. datasource[5] = datasource[5].split("_")[0] + "_" + time;
  740. return Arrays.stream(datasource).collect(Collectors.joining("/"));
  741. }
  742. @Override
  743. public String setVideos(String videos,String oldNum,String newNum) {
  744. JSONObject object = new JSONObject();
  745. if(StringUtils.isNotEmpty(videos)){
  746. object = JSONObject.parseObject(videos);
  747. if(object.containsKey("upPath")){
  748. String upPath = object.getString("upPath");
  749. upPath = upPath.replace(oldNum,newNum);
  750. object.put("upPath",upPath);
  751. }
  752. log.info("更新 scene.json");
  753. }
  754. return object.toJSONString();
  755. }
  756. private void reloadFile(String filePath,String sceneNum, String newNum) throws Exception {
  757. // 修改link-scene.json
  758. String content = FileUtils.readFile(filePath);
  759. if (!ObjectUtils.isEmpty(content)) {
  760. content = content.replaceAll(sceneNum, newNum);
  761. FileUtils.writeFile(filePath,content);
  762. }
  763. }
  764. /**
  765. * 上传新场景资源到oss
  766. * @param newNum
  767. */
  768. private void uploadNewSceneToOss(String newNum) {
  769. ShellUtil.yunUpload("/mnt/4Dkankan/scene/images/images" + newNum, "images/images" + newNum);
  770. ShellUtil.yunUpload("/mnt/4Dkankan/scene/data/data" + newNum, "data/data" + newNum);
  771. ShellUtil.yunUpload("/mnt/4Dkankan/scene/voice/voice" + newNum, "voice/voice" + newNum);
  772. ShellUtil.yunUpload("/mnt/4Dkankan/scene/video/video" + newNum, "video/video" + newNum);
  773. }
  774. @Override
  775. public void delete(String sceneNum) {
  776. if(StringUtils.isEmpty(sceneNum)){
  777. throw new BusinessException(LoginConstant.FAILURE_CODE_3001,LoginConstant.FAILURE_MSG_3001);
  778. }
  779. String[] nums = sceneNum.split(",");
  780. List<String> numList = Arrays.asList(nums);
  781. List<ScenePro> proList = this.getListByNums(numList);
  782. List<ScenePlus> plusList = scenePlusService.getListByNums(numList);
  783. this.deleteByList(proList,plusList);
  784. }
  785. @Override
  786. public void deleteByList(List<ScenePro> proList,List<ScenePlus> plusList) {
  787. HashMap<Long ,Long >cameraMap = new HashMap<>();
  788. if(proList.size() >0){
  789. for (ScenePro scenePro : proList) {
  790. cameraMap.merge(scenePro.getCameraId(), scenePro.getSpace(), Long::sum);
  791. }
  792. List<Long> proIds = proList.parallelStream().map(ScenePro::getId).collect(Collectors.toList());
  793. this.removeByIds(proIds);
  794. }
  795. if(plusList.size() >0){
  796. List<Long> plusIds = plusList.parallelStream().map(ScenePlus::getId).collect(Collectors.toList());
  797. HashMap<Long, ScenePlusExt> plusMap = scenePlusExtService.getByPlusIds(plusIds);
  798. for (ScenePlus scenePlus : plusList) {
  799. if(scenePlus.getCameraId()!= null && plusMap.get(scenePlus.getId())!=null){
  800. cameraMap.merge(scenePlus.getCameraId(), plusMap.get(scenePlus.getId()).getSpace(), Long::sum);
  801. }
  802. }
  803. scenePlusService.removeByIds(plusIds);
  804. }
  805. //恢复相机使用容量
  806. if(cameraMap.size() >0){
  807. cameraDetailService.addUsedSpace(cameraMap);
  808. }
  809. }
  810. @Override
  811. public ScenePro getByUnicode(String unicode) {
  812. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  813. wrapper.like(ScenePro::getDataSource,unicode);
  814. wrapper.eq(ScenePro::getIsUpgrade,0);
  815. wrapper.orderByDesc(ScenePro::getCreateTime);
  816. List<ScenePro> list = this.list(wrapper);
  817. if(list !=null && list.size() >0){
  818. return list.get(0);
  819. }
  820. return null;
  821. }
  822. @Override
  823. public SceneInfoVo getInfo(String num) {
  824. if(StringUtils.isEmpty(num)){
  825. throw new BusinessException(LoginConstant.FAILURE_CODE_3001, LoginConstant.FAILURE_MSG_3001);
  826. }
  827. ScenePro scenePro = this.getByNum(num);
  828. ScenePlus scenePlus = scenePlusService.getByNum(num);
  829. if(scenePro == null && scenePlus == null){
  830. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  831. }
  832. if(scenePro!=null && scenePro.getStatus() != 1 && scenePro.getStatus() != -2){
  833. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  834. }
  835. if(scenePlus!=null && scenePlus.getSceneStatus() != 1 && scenePlus.getSceneStatus() != -2){
  836. throw new BusinessException(SceneConstant.FAILURE_CODE_5033, SceneConstant.FAILURE_MSG_5033);
  837. }
  838. if(scenePro!=null && scenePro.getPayStatus() != 1){
  839. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  840. }
  841. if(scenePlus!=null && scenePlus.getPayStatus() != 1){
  842. throw new BusinessException(SceneConstant.FAILURE_CODE_5034, SceneConstant.FAILURE_MSG_5034);
  843. }
  844. SceneInfoVo infoVo = new SceneInfoVo();
  845. if(scenePro!=null){
  846. SceneProEdit sceneProEdit = sceneProEditService.getByProId(scenePro.getId());
  847. if(sceneProEdit !=null){
  848. BeanUtils.copyProperties(sceneProEdit, infoVo);
  849. }
  850. BeanUtils.copyProperties(scenePro,infoVo);
  851. if(StringUtils.isNotEmpty(scenePro.getGps())){
  852. infoVo.setGps(JSONObject.parseObject(scenePro.getGps()).toJSONString());
  853. }
  854. infoVo.setSceneKey("");
  855. infoVo.setCreateDate(DateUserUtil.getDate(scenePro.getCreateTime()).getTime());
  856. }else {
  857. SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
  858. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
  859. if(sceneEditInfo != null){
  860. BeanUtils.copyProperties(sceneEditInfo,infoVo);
  861. }
  862. if(scenePlusExt !=null){
  863. BeanUtils.copyProperties(scenePlusExt,infoVo);
  864. }
  865. BeanUtils.copyProperties(scenePlus,infoVo);
  866. infoVo.setSceneName(scenePlus.getTitle());
  867. infoVo.setStatus(scenePlus.getSceneStatus());
  868. }
  869. if(StringUtils.isNotEmpty(infoVo.getEntry())){
  870. infoVo.setEntry(JSONObject.parseObject(infoVo.getEntry()).toJSONString());
  871. }
  872. if(StringUtils.isEmpty(infoVo.getSceneKey())){
  873. infoVo.setIsPublic(0);
  874. }else {
  875. infoVo.setIsPublic(1);
  876. }
  877. //查询是否有随心装场景
  878. if(num.contains("vr-")){
  879. infoVo.setVrNum(null);
  880. infoVo.setVideosUser(null);
  881. infoVo.setBgMusicName(null);
  882. infoVo.setBgMusic(null);
  883. }
  884. infoVo.setSceneVer(innerService.getSceneNumVersion(num));
  885. this.updateViewCount(num);
  886. return infoVo;
  887. }
  888. @Override
  889. public void updateViewCount(String num) {
  890. String redisKey = RedisKey.SCENE_VISIT_CNT;
  891. if(!redisUtil.hHasKey(redisKey,num)){
  892. Integer count = 0;
  893. ScenePro pro = this.getByNum(num);
  894. if(pro !=null){
  895. count = pro.getViewCount() == null ? 0 : pro.getViewCount();
  896. }else {
  897. ScenePlus plus = scenePlusService.getByNum(num);
  898. if(plus !=null){
  899. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(plus.getId());
  900. if(scenePlusExt != null){
  901. count = scenePlusExt.getViewCount() == null ? 0 :scenePlusExt.getViewCount();
  902. }
  903. }
  904. }
  905. redisUtil.hset(redisKey,num,String.valueOf(count));
  906. }
  907. redisUtil.hincr(redisKey,num,1);
  908. }
  909. @Override
  910. public void updateDbViewCount(String num, String count) {
  911. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  912. wrapper.eq(ScenePro::getNum,num);
  913. wrapper.set(ScenePro::getViewCount,count);
  914. this.update(wrapper);
  915. ScenePlus plus = scenePlusService.getByNum(num);
  916. if(plus !=null){
  917. LambdaUpdateWrapper<ScenePlusExt> exWr = new LambdaUpdateWrapper<>();
  918. exWr.eq(ScenePlusExt::getPlusId,plus.getId())
  919. .set(ScenePlusExt::getViewCount,count);
  920. scenePlusExtService.update(exWr);
  921. }
  922. }
  923. @Override
  924. public ScenePro findByFileId(String fileId) {
  925. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  926. wrapper.like(ScenePro::getDataSource,fileId);
  927. List<ScenePro> list = this.list(wrapper);
  928. if(list !=null && list.size() >0){
  929. return list.get(0);
  930. }
  931. return null;
  932. }
  933. @Override
  934. public void setQrCode(String buildType,String num) throws Exception {
  935. String basePath = mainUrl;
  936. if("V2".equals(buildType)){
  937. basePath += sceneProUrl;
  938. }
  939. if("V3".equals(buildType)){
  940. basePath += sceneProNewUrl;
  941. }
  942. MatrixToImageWriterUtil.createQRCode(basePath + num, ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+".png", false,null);
  943. MatrixToImageWriterUtil.createQRCode(basePath + num + "&lang=en", ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/"+num+"_en.png", false,null);
  944. }
  945. @Override
  946. public void generateObjFile(String num) {
  947. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  948. wrapper.eq(ScenePro::getNum,num).eq(ScenePro::getRecStatus,"A");
  949. ScenePro sceneProEntity = this.getOne(wrapper);
  950. if(ObjectUtils.isEmpty(sceneProEntity)){
  951. generatePlusObjFile(num);
  952. return;
  953. }
  954. if(sceneProEntity.getSceneSource() != 4){
  955. throw new BusinessException(LoginConstant.FAILURE_CODE_3003, "只能操作激光场景");
  956. }
  957. // 拷贝文件
  958. String path = sceneProEntity.getDataSource();
  959. if (!new File(path + "/caches/reconstruction/final.bin").exists()
  960. || !new File(path + "/caches/reconstruction/chunk.json").exists()
  961. || !new File(path + "/caches/images").exists()
  962. || !new File(path + "/caches/depthmap").exists()
  963. || !new File(path + "/caches/depthmap_csc").exists()
  964. || !new File(path + "/caches/panorama.json").exists()
  965. || !new File(path + "/results/laserData/laser.ply").exists()) {
  966. log.error("生成obj缺少必要文件,生成失败!");
  967. throw new BusinessException(SceneConstant.FAILURE_CODE_5038, SceneConstant.FAILURE_MSG_5038);
  968. }
  969. // 获取最新的场景名称
  970. JSONObject sceneInfo = fdkkLaserService.getSceneByNum(sceneProEntity.getNum());
  971. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  972. updateWrapper
  973. .set(ScenePro::getStatus, 0)
  974. .set(ScenePro::getIsObj, 1)
  975. .set(ScenePro::getIsUpgrade, 2) // 升级中
  976. .set(ScenePro::getSceneName, sceneInfo.getString("title"))
  977. .eq(ScenePro::getNum, sceneProEntity.getNum());
  978. this.update(updateWrapper);
  979. //同步到scenePlus、scenePlus
  980. sceneUpgradeMapper.deleteScenePlus(sceneProEntity.getId());
  981. sceneUpgradeMapper.transferScenePlus(sceneProEntity.getId());
  982. sceneUpgradeMapper.deleteScenePlusExt(sceneProEntity.getId());
  983. String sceneKind = sceneProEntity.getSceneScheme() == 3 ? SceneKind.FACE.code(): SceneKind.TILES.code();
  984. sceneUpgradeMapper.transferScenePlusExt(sceneProEntity.getId(), sceneKind);
  985. log.info("开始发送激光场景生成obj mq消息");
  986. // 发送MQ
  987. BuildSceneCallMessage mqMsg = new BuildSceneCallMessage();
  988. mqMsg.setSceneNum(sceneProEntity.getNum());
  989. mqMsg.setAlgorithm(sceneProEntity.getAlgorithm());
  990. mqMsg.setBuildType(sceneProEntity.getBuildType());
  991. mqMsg.setPath(sceneProEntity.getDataSource());
  992. mqProducer.sendByWorkQueue(queueObjModelingPre,mqMsg);
  993. }
  994. public void generatePlusObjFile(String num) {
  995. ScenePlus scenePlus = scenePlusService.getByNum(num);
  996. if(ObjectUtils.isEmpty(scenePlus)){
  997. throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
  998. }
  999. if(scenePlus.getSceneSource() != 4){
  1000. throw new BusinessException(LoginConstant.FAILURE_CODE_3003, "只能操作激光场景");
  1001. }
  1002. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
  1003. // 拷贝文件
  1004. String path = scenePlusExt.getDataSource();
  1005. if (!new File(path + "/caches/reconstruction/final.bin").exists()
  1006. || !new File(path + "/caches/reconstruction/chunk.json").exists()
  1007. || !new File(path + "/caches/images").exists()
  1008. || !new File(path + "/caches/depthmap").exists()
  1009. || !new File(path + "/caches/depthmap_csc").exists()
  1010. || !new File(path + "/caches/panorama.json").exists()
  1011. || !new File(path + "/results/laserData/laser.ply").exists()) {
  1012. log.error("生成obj缺少必要文件,生成失败!");
  1013. throw new BusinessException(SceneConstant.FAILURE_CODE_5038, SceneConstant.FAILURE_MSG_5038);
  1014. }
  1015. // 获取最新的场景名称
  1016. JSONObject sceneInfo = fdkkLaserService.getSceneByNum(num);
  1017. LambdaUpdateWrapper<ScenePlus> updateWrapper = new LambdaUpdateWrapper<>();
  1018. updateWrapper
  1019. .set(ScenePlus::getSceneStatus, 0)
  1020. .set(ScenePlus::getTitle, sceneInfo.getString("title"))
  1021. .eq(ScenePlus::getNum, num);
  1022. scenePlusService.update(updateWrapper);
  1023. LambdaUpdateWrapper<ScenePlusExt> plusExtUpdateWrapper = new LambdaUpdateWrapper<>();
  1024. plusExtUpdateWrapper
  1025. .set(ScenePlusExt::getIsObj, 1)
  1026. .eq(ScenePlusExt::getPlusId, scenePlus.getId());
  1027. scenePlusExtService.update(plusExtUpdateWrapper);
  1028. log.info("开始发送激光场景生成obj mq消息");
  1029. // 发送MQ
  1030. BuildSceneCallMessage mqMsg = new BuildSceneCallMessage();
  1031. mqMsg.setSceneNum(num);
  1032. mqMsg.setAlgorithm(scenePlusExt.getAlgorithm());
  1033. mqMsg.setBuildType(scenePlusExt.getBuildType());
  1034. mqMsg.setPath(scenePlusExt.getDataSource());
  1035. mqProducer.sendByWorkQueue(queueObjModelingPre,mqMsg);
  1036. }
  1037. @Override
  1038. public ResponseScene getSceneDetail(String sceneNum) {
  1039. if(StringUtils.isBlank(sceneNum)){
  1040. return null;
  1041. }
  1042. ResponseScene vo = new ResponseScene();
  1043. Scene sceneEntity = sceneService.getByNum(sceneNum);
  1044. if(sceneEntity != null){
  1045. return getResponseScene(vo, sceneEntity);
  1046. }
  1047. ScenePro sceneProEntity = this.getByNum(sceneNum);
  1048. if(sceneProEntity != null){
  1049. SceneProEdit sceneProEditEntity = sceneProEditService.getByProId(sceneProEntity.getId());
  1050. return getResponseProScene(vo, sceneProEntity, sceneProEditEntity);
  1051. }
  1052. ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
  1053. if(scenePlus != null){
  1054. ScenePlusExt ext = scenePlusExtService.getByPlusId(scenePlus.getId());
  1055. SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
  1056. return getResponseProScene(vo, scenePlus, ext,sceneEditInfo);
  1057. }
  1058. return vo;
  1059. }
  1060. private ResponseScene getResponseScene(ResponseScene vo, Scene sceneEntity) {
  1061. if (sceneEntity != null){
  1062. BeanUtils.copyProperties(sceneEntity, vo);
  1063. vo.setThumbImg(sceneEntity.getThumbStatus());
  1064. vo.setCreateTime(new DateTime(sceneEntity.getCreateTime()).toString("yyyy-MM-dd HH:mm"));
  1065. vo.setSceneIndex(sceneEntity.getStyle());
  1066. vo.setHasBGM(sceneEntity.getBgMusic());
  1067. vo.setCameraType(sceneEntity.getSceneScheme());
  1068. vo.setIsPublic(StringUtils.isEmpty(sceneEntity.getSceneKey()) ? 0 : 1);
  1069. }
  1070. return vo;
  1071. }
  1072. private ResponseScene getResponseProScene(ResponseScene vo, ScenePro sceneProEntity, SceneProEdit sceneProEditEntity) {
  1073. if (sceneProEntity != null){
  1074. BeanUtils.copyProperties(sceneProEditEntity, vo);
  1075. BeanUtils.copyProperties(sceneProEntity, vo);
  1076. vo.setCreateTime(DateUserUtil.getMDate(vo.getCreateTime()));
  1077. vo.setCameraType(sceneProEntity.getSceneScheme());
  1078. vo.setThumbImg(sceneProEditEntity.getThumbStatus());
  1079. vo.setHasBGM(sceneProEditEntity.getBgMusic());
  1080. vo.setIsPublic(StringUtils.isEmpty(sceneProEditEntity.getSceneKey()) ? 0 : 1);
  1081. }
  1082. return vo;
  1083. }
  1084. private ResponseScene getResponseProScene(ResponseScene vo, ScenePlus plus, ScenePlusExt ext,SceneEditInfo editInfo) {
  1085. if (plus != null){
  1086. BeanUtils.copyProperties(plus, vo);
  1087. BeanUtils.copyProperties(ext, vo);
  1088. BeanUtils.copyProperties(editInfo, vo);
  1089. vo.setNum(plus.getNum());
  1090. vo.setSceneName(plus.getTitle());
  1091. vo.setCreateTime(DateUserUtil.getMDate(vo.getCreateTime()));
  1092. vo.setCameraType(ext.getSceneScheme());
  1093. vo.setThumbImg(1);
  1094. vo.setHasBGM(editInfo.getMusic());
  1095. vo.setIsPublic(StringUtils.isEmpty(editInfo.getScenePassword()) ? 0 : 1);
  1096. }
  1097. return vo;
  1098. }
  1099. }