SceneProServiceImpl.java 59 KB

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