SceneProServiceImpl.java 56 KB

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