SceneProServiceImpl.java 54 KB

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