SceneProServiceImpl.java 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481
  1. package com.fdkankan.manage_jp.service.impl;
  2. import cn.hutool.core.io.FileUtil;
  3. import cn.hutool.core.thread.ThreadUtil;
  4. import cn.hutool.json.JSONUtil;
  5. import com.alibaba.fastjson.JSONObject;
  6. import com.alibaba.nacos.shaded.org.checkerframework.checker.units.qual.A;
  7. import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
  8. import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
  9. import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
  10. import com.fdkankan.common.constant.ErrorCode;
  11. import com.fdkankan.common.constant.SceneConstant;
  12. import com.fdkankan.common.constant.SceneKind;
  13. import com.fdkankan.common.constant.SceneVersionType;
  14. import com.fdkankan.common.util.FileUtils;
  15. import com.fdkankan.common.util.JwtUtil;
  16. import com.fdkankan.fyun.face.FYunFileServiceInterface;
  17. import com.fdkankan.manage_jp.common.OssPath;
  18. import com.fdkankan.manage_jp.common.PageInfo;
  19. import com.fdkankan.manage_jp.common.RedisKeyUtil;
  20. import com.fdkankan.manage_jp.common.ResultCode;
  21. import com.fdkankan.manage_jp.entity.*;
  22. import com.fdkankan.manage_jp.exception.BusinessException;
  23. import com.fdkankan.manage_jp.httpClient.service.LaserService;
  24. import com.fdkankan.manage_jp.mapper.ISceneProMapper;
  25. import com.fdkankan.manage_jp.mapper.ISceneUpgradeMapper;
  26. import com.fdkankan.manage_jp.service.*;
  27. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  28. import com.fdkankan.manage_jp.util.GenerateNumUtil;
  29. import com.fdkankan.manage_jp.util.SceneResourcePath;
  30. import com.fdkankan.manage_jp.util.SnowflakeIdGenerator;
  31. import com.fdkankan.manage_jp.vo.request.SceneParam;
  32. import com.fdkankan.manage_jp.vo.response.SceneVo;
  33. import com.fdkankan.rabbitmq.bean.BuildSceneCallMessage;
  34. import com.fdkankan.rabbitmq.util.RabbitMqProducer;
  35. import com.fdkankan.redis.util.RedisUtil;
  36. import lombok.extern.slf4j.Slf4j;
  37. import org.apache.commons.lang3.StringUtils;
  38. import org.springframework.beans.factory.annotation.Autowired;
  39. import org.springframework.beans.factory.annotation.Value;
  40. import org.springframework.scheduling.annotation.Async;
  41. import org.springframework.stereotype.Service;
  42. import org.springframework.util.ObjectUtils;
  43. import java.io.File;
  44. import java.util.*;
  45. import java.util.concurrent.CompletableFuture;
  46. import java.util.concurrent.ExecutorService;
  47. import java.util.stream.Collectors;
  48. /**
  49. * <p>
  50. * pro场景表 服务实现类
  51. * </p>
  52. *
  53. * @author
  54. * @since 2022-12-30
  55. */
  56. @Service
  57. @Slf4j
  58. public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro> implements ISceneProService {
  59. @Autowired
  60. IScenePlusService scenePlusService;
  61. @Autowired
  62. LaserService laserService;
  63. @Autowired
  64. ISceneProEditService sceneProEditService;
  65. @Autowired
  66. ICompanyService companyService;
  67. @Autowired
  68. IScenePlusExtService scenePlusExtService;
  69. @Autowired
  70. FYunFileServiceInterface fYunFileServiceInterface;
  71. @Autowired
  72. IScene3dNumService scene3dNumService;
  73. @Autowired
  74. SceneCommonService sceneCommonService;
  75. @Autowired
  76. ITmContractorService tmContractorService;
  77. @Autowired
  78. ITmContractorNumService tmContractorNumService;
  79. @Autowired
  80. IUserRoleService userRoleService;
  81. @Autowired
  82. ITmColdStorageService tmColdStorageService;
  83. @Autowired
  84. private RabbitMqProducer mqProducer;
  85. @Value("${queue.modeling.obj.modeling-pre}")
  86. private String queueObjModelingPre;
  87. @Autowired
  88. private ISceneUpgradeMapper sceneUpgradeMapper;
  89. @Autowired
  90. ISceneCopyLogService sceneCopyLogService;
  91. @Autowired
  92. ICameraDetailService cameraDetailService;
  93. @Autowired
  94. ICameraService cameraService;
  95. @Autowired
  96. IUserService userService;
  97. @Autowired
  98. ISceneMoveLogService sceneMoveLogService;
  99. @Override
  100. public Long getCountByUserIds(List<Long> userIds) {
  101. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  102. wrapper.in(ScenePro::getUserId,userIds);
  103. wrapper.eq(ScenePro::getIsUpgrade,0);
  104. long count = this.count(wrapper);
  105. LambdaQueryWrapper<ScenePlus> wrapper2 = new LambdaQueryWrapper<>();
  106. wrapper2.in(ScenePlus::getUserId,userIds);
  107. long count2 = scenePlusService.count(wrapper2);
  108. return count + count2;
  109. }
  110. @Override
  111. public void rebindUser(Long fromUserId, Long toUserId) {
  112. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  113. wrapper.eq(ScenePro::getUserId,fromUserId);
  114. wrapper.set(ScenePro::getUserId,toUserId);
  115. this.update(wrapper);
  116. LambdaUpdateWrapper<ScenePlus> wrapper2 = new LambdaUpdateWrapper<>();
  117. wrapper2.eq(ScenePlus::getUserId,fromUserId);
  118. wrapper2.set(ScenePlus::getUserId,toUserId);
  119. scenePlusService.update(wrapper2);
  120. }
  121. @Override
  122. public Object pageList(SceneParam param, User user) {
  123. Set<Long> roleIds = userRoleService.getByUser(user);
  124. List<String> numList = new ArrayList<>();
  125. List<String> cooNumList = new ArrayList<>();
  126. if(!roleIds.contains(5L)){ //平台管理员
  127. param.setCompanyId(user.getCompanyId());
  128. param.setUserId(user.getId());
  129. param.setNeStatus(-2);
  130. if(!roleIds.contains(6L)){
  131. param.setIsDel(false);
  132. param.setIsMain(false);
  133. param.setIsMigrate(false);
  134. }
  135. numList = tmContractorNumService.getNumListByCompanyId(user.getCompanyId());
  136. param.setCooperateSceneCodes(numList);
  137. }
  138. Page<SceneVo> page = this.getBaseMapper().pageList(new Page<>(param.getPageNum(),param.getPageSize()),param);
  139. List<String> sceneNumList = page.getRecords().stream().map(SceneVo::getNum).collect(Collectors.toList());
  140. if(!sceneNumList.isEmpty()){
  141. cooNumList = tmContractorNumService.getNumList(sceneNumList);
  142. }
  143. HashMap<String,JSONObject> ssSceneMap = new HashMap<>();
  144. if(param.getType() == 2 || param.getType() == 5){
  145. ssSceneMap = laserService.list(sceneNumList);
  146. }
  147. HashMap<String,Boolean> coldMap = tmColdStorageService.getByNumList(sceneNumList);
  148. for (SceneVo record : page.getRecords()) {
  149. if(param.getType() == 2 || param.getType() == 5){
  150. JSONObject ssObj = ssSceneMap.get(record.getNum());
  151. if(ssObj!=null){
  152. record.setSceneName(ssObj.getString("title"));
  153. record.setThumb(ssObj.getString("thumb"));
  154. record.setWebSite(ssObj.getString("webSite"));
  155. record.setIsObj(ssObj.getInteger("buildObjStatus"));
  156. record.setViewCount(ssObj.getInteger("viewCount"));
  157. record.setStatus(toFdStatus(ssObj.getInteger("status")));
  158. }else{
  159. record.setStatus(-1);
  160. record.setWebSite(null);
  161. }
  162. }
  163. if(StringUtils.isNotBlank(record.getSceneVersion()) && record.getSceneVersion().equals("v3")){
  164. record.setIsUpgrade(0);
  165. }
  166. record.setIsDel(param.getIsDel());
  167. record.setIsMain(param.getIsMain());
  168. record.setIsDownload(param.getIsDownload());
  169. record.setIsMigrate(param.getIsMigrate());
  170. if(!roleIds.contains(5L) && numList != null && numList.contains(record.getNum())){
  171. record.setIsMain(false);
  172. record.setIsDel(false);
  173. record.setIsDownload(false);
  174. }
  175. if(cooNumList != null && cooNumList.contains(record.getNum())){
  176. record.setIsMigrate(false);
  177. }
  178. if(coldMap.get(record.getNum())!=null){
  179. record.setIsMigrate(false);
  180. }
  181. }
  182. return PageInfo.PageInfo(page);
  183. }
  184. private Integer toFdStatus(Integer status) {
  185. //深时状态,-1:场景被删 0:计算中 1计算失败 2计算成功 3封存 4生成OBJ中
  186. switch (status) {
  187. case 0:
  188. case 4:
  189. return 0;
  190. case 2:
  191. return -2;
  192. case 3:
  193. return -3;
  194. default:
  195. return -1;
  196. }
  197. }
  198. @Override
  199. public ScenePro getByNum(String num) {
  200. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  201. wrapper.eq(ScenePro::getNum,num);
  202. wrapper.eq(ScenePro::getIsUpgrade,0);
  203. List<ScenePro> list = this.list(wrapper);
  204. if(list == null || list.size() <=0){
  205. return null;
  206. }
  207. return list.get(0);
  208. }
  209. @Override
  210. public HashMap<String, ScenePro> getByNumList(List<String> numList) {
  211. HashMap<String, ScenePro> map = new HashMap<>();
  212. if(numList == null || numList.isEmpty()){
  213. return map;
  214. }
  215. LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
  216. wrapper.in(ScenePro::getNum,numList);
  217. wrapper.eq(ScenePro::getIsUpgrade,0);
  218. List<ScenePro> list = this.list(wrapper);
  219. list.forEach(e -> map.put(e.getNum(),e));
  220. return map;
  221. }
  222. @Override
  223. public void addDownloadNum(String sceneNum) {
  224. if(StringUtils.isBlank(sceneNum)){
  225. throw new BusinessException(ResultCode.PARAM_ERROR);
  226. }
  227. ScenePro scenePro = this.getByNum(sceneNum);
  228. if(scenePro != null){
  229. SceneProEdit sceneProEditEntity = sceneProEditService.getByProId(scenePro.getId());
  230. if(sceneProEditEntity.getDownloadNum() == null){
  231. sceneProEditEntity.setDownloadNum(1);
  232. }else{
  233. sceneProEditEntity.setDownloadNum(sceneProEditEntity.getDownloadNum() + 1);
  234. }
  235. sceneProEditService.updateById(sceneProEditEntity);
  236. }
  237. }
  238. @Override
  239. public void deleteByNum(String num) {
  240. ScenePro scenePro = this.getByNum(num);
  241. if(scenePro!=null){
  242. this.removeById(scenePro.getId());
  243. this.updateOssStatus(String.format(OssPath.v3_statusPath,scenePro.getNum()),"I");
  244. }
  245. ScenePlus scenePlus = scenePlusService.getByNum(num);
  246. if(scenePlus!=null){
  247. scenePlusService.removeById(scenePlus.getId());
  248. scenePlusExtService.delByPlus(scenePlus.getId());
  249. this.updateOssStatus(String.format(OssPath.v4_statusPath,scenePlus.getNum()),"I");
  250. }
  251. Integer sceneSource = scenePro == null ? scenePlus.getSceneSource() : scenePro.getSceneSource();
  252. if(sceneSource == 4 || sceneSource == 5){
  253. laserService.updateStatus(num,-1);
  254. }
  255. tmContractorNumService.deleteByNum(num);
  256. tmContractorService.checkDel();
  257. }
  258. private void updateOssStatus(String path,String recStatus) {
  259. try {
  260. if(!fYunFileServiceInterface.fileExist(path)){
  261. return;
  262. }
  263. String data = fYunFileServiceInterface.getFileContent(path);
  264. if(StringUtils.isBlank(data)){
  265. return;
  266. }
  267. JSONObject jsonObject = JSONObject.parseObject(data);
  268. jsonObject.put("recStatus",recStatus);
  269. String json = JSONUtil.toJsonStr(jsonObject);
  270. FileUtils.writeFile(OssPath.localStatusPath ,json);
  271. log.info("updateOssStatus--localPath:{},ossPath:{}",OssPath.localStatusPath,path);
  272. fYunFileServiceInterface.uploadFile(OssPath.localStatusPath,path);
  273. }catch (Exception e){
  274. e.printStackTrace();
  275. }finally {
  276. FileUtil.del(OssPath.localStatusPath);
  277. }
  278. }
  279. @Override
  280. public void unbind(Long cameraId,Long userId) {
  281. LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
  282. wrapper.eq(ScenePro::getCameraId,cameraId);
  283. wrapper.eq(ScenePro::getIsUpgrade,0);
  284. wrapper.set(ScenePro::getUserId,userId);
  285. this.update(wrapper);
  286. }
  287. @Override
  288. public ScenePro copyScene(ScenePro scenePro) {
  289. if(scenePro.getStatus() !=-2){
  290. throw new BusinessException(ResultCode.SCENE_ERROR);
  291. }
  292. String oldNum = scenePro.getNum();
  293. String newNum = sceneCommonService.getNewNum(oldNum);
  294. Long sceneProId = scenePro.getId();
  295. String oldSceneName = scenePro.getSceneName();
  296. scenePro.setNum(newNum);
  297. scenePro.setId(null);
  298. scenePro.setSceneName(scenePro.getSceneName() +"(copy)");
  299. scenePro.setViewCount(0);
  300. scenePro.setThumb(scenePro.getThumb().replaceAll(oldNum,scenePro.getNum()));
  301. scenePro.setWebSite(scenePro.getWebSite().replaceAll(oldNum,scenePro.getNum()));
  302. scenePro.setStatus(0);
  303. this.save(scenePro);
  304. try {
  305. sceneCommonService.v3Async(scenePro,oldNum,newNum,sceneProId,oldSceneName);
  306. }catch (Exception e){
  307. log.error("复制场景失败-sceneNum:{},error:{}",oldNum,e);
  308. scenePro.setStatus(-1);
  309. this.updateById(scenePro);
  310. throw new BusinessException(ResultCode.SCENE_DATA_ERROR);
  311. }
  312. return scenePro;
  313. }
  314. @Override
  315. public void updateTitleByNum(String sceneNum, String sceneNewTitle) {
  316. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  317. updateWrapper.eq(ScenePro::getNum,sceneNum);
  318. updateWrapper.set(ScenePro::getLaserTitle,sceneNewTitle);
  319. this.update(updateWrapper);
  320. }
  321. @Override
  322. public void generateObjFile(String num) {
  323. ScenePro sceneProEntity = this.getByNum(num);
  324. ScenePlus scenePlusEntity = scenePlusService.getByNum(num);
  325. if(ObjectUtils.isEmpty(sceneProEntity) && ObjectUtils.isEmpty(scenePlusEntity)){
  326. throw new BusinessException(ResultCode.SCENE_NOT_EXIT);
  327. }
  328. List<SceneCopyLog> sceneCopyLogs = sceneCopyLogService.getByNewNum(num);
  329. if(sceneCopyLogs!=null && sceneCopyLogs.size()>0){
  330. throw new BusinessException(ResultCode.GENERATE_OBJ_ERROR);
  331. }
  332. String path = null;
  333. ScenePlusExt scenePlusExt = null;
  334. if(scenePlusEntity != null){
  335. scenePlusExt = scenePlusExtService.getByPlusId(scenePlusEntity.getId());
  336. if(scenePlusExt == null){
  337. throw new BusinessException(ResultCode.SCENE_ERROR);
  338. }
  339. path = scenePlusExt.getDataSource();
  340. }else {
  341. path = sceneProEntity.getDataSource();
  342. }
  343. Integer sceneSource = sceneProEntity == null ? scenePlusEntity.getSceneSource() : sceneProEntity.getSceneSource();
  344. if(sceneSource != 4){
  345. throw new BusinessException(ResultCode.GENERATE_OBJ_ERROR_TYPE);
  346. }
  347. laserService.updateBuildObjStatus(num,2);
  348. String ossPath = path.replace("/mnt/data","home")+"/data.fdage";
  349. if(!fYunFileServiceInterface.fileExist(ossPath)){
  350. throw new BusinessException(ResultCode.GENERATE_OBJ_ERROR);
  351. }
  352. // 获取最新的场景名称
  353. JSONObject sceneInfo = laserService.getSceneByNum(num);
  354. if(sceneProEntity == null){
  355. LambdaUpdateWrapper<ScenePlus> updateWrapper = new LambdaUpdateWrapper<>();
  356. updateWrapper
  357. .set(ScenePlus::getSceneStatus, 0)
  358. .set(ScenePlus::getTitle, sceneInfo.getString("title"))
  359. .eq(ScenePlus::getNum, num);
  360. scenePlusService.update(updateWrapper);
  361. LambdaUpdateWrapper<ScenePlusExt> plusExtUpdateWrapper = new LambdaUpdateWrapper<>();
  362. plusExtUpdateWrapper
  363. .set(ScenePlusExt::getIsObj, 1)
  364. .eq(ScenePlusExt::getPlusId, scenePlusEntity.getId());
  365. scenePlusExtService.update(plusExtUpdateWrapper);
  366. sendMqMsg(num,scenePlusExt.getAlgorithm(),scenePlusExt.getBuildType(),scenePlusExt.getDataSource());
  367. return;
  368. }
  369. LambdaUpdateWrapper<ScenePro> updateWrapper = new LambdaUpdateWrapper<>();
  370. updateWrapper
  371. .set(ScenePro::getStatus, 0)
  372. .set(ScenePro::getIsObj, 1)
  373. .set(ScenePro::getIsUpgrade, 2) // 升级中
  374. .set(ScenePro::getSceneName, sceneInfo.getString("title"))
  375. .eq(ScenePro::getNum, sceneProEntity.getNum());
  376. this.update(updateWrapper);
  377. //同步到scenePlus、scenePlus
  378. //同步到scenePlus、scenePlus
  379. ScenePlus scenePlus = scenePlusService.getByNum(num);
  380. if(Objects.nonNull(scenePlus)){
  381. sceneUpgradeMapper.deleteScenePlus(num);
  382. sceneUpgradeMapper.deleteScenePlusExt(scenePlus.getId());
  383. }
  384. sceneUpgradeMapper.transferScenePlus(num);
  385. scenePlus = scenePlusService.getByNum(num);
  386. String sceneKind = sceneProEntity.getSceneScheme() == 3 ? SceneKind.FACE.code():SceneKind.TILES.code();
  387. sceneUpgradeMapper.transferScenePlusExt(num, scenePlus.getId(), sceneKind);
  388. sendMqMsg(sceneProEntity.getNum(),sceneProEntity.getAlgorithm(),sceneProEntity.getBuildType(),sceneProEntity.getDataSource());
  389. }
  390. public void sendMqMsg(String num ,String algorithm,String buildType,String dataSource) {
  391. log.info("开始发送激光场景生成obj mq消息");
  392. // 发送MQ
  393. BuildSceneCallMessage mqMsg = new BuildSceneCallMessage();
  394. mqMsg.setSceneNum(num);
  395. mqMsg.setAlgorithm(algorithm);
  396. mqMsg.setBuildType(buildType);
  397. mqMsg.setPath(dataSource);
  398. mqProducer.sendByWorkQueue(queueObjModelingPre,mqMsg);
  399. }
  400. @Override
  401. public void move(SceneParam param) {
  402. if(param.getNumList().isEmpty() || param.getCameraId() == null || StringUtils.isBlank(param.getUserName())){
  403. throw new BusinessException(ResultCode.PARAM_ERROR);
  404. }
  405. for (String num : param.getNumList()) {
  406. User user = userService.getByUserName(param.getUserName());
  407. if(user == null || user.getCompanyId() == null){
  408. throw new BusinessException(ResultCode.USER_NOT_EXIST);
  409. }
  410. ScenePlus scenePlus = scenePlusService.getByNum(num);
  411. ScenePro scenePro = this.getByNum(num);
  412. if(scenePlus == null && scenePro == null){
  413. throw new BusinessException(ResultCode.SCENE_NOT_EXIT);
  414. }
  415. Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
  416. Integer sceneSource = scenePro == null ? scenePlus.getSceneSource() : scenePro.getSceneSource();
  417. Camera oldCamera = cameraService.getById(cameraId);
  418. Camera camera = cameraService.getById(param.getCameraId());
  419. CameraDetail oldCameraDetail = cameraDetailService.getByCameraId(cameraId);
  420. CameraDetail cameraDetail = cameraDetailService.getByCameraId(param.getCameraId());
  421. if(oldCamera == null || oldCameraDetail == null || camera == null || cameraDetail == null){
  422. throw new BusinessException(ResultCode.CAMERA_NOT_EXIST);
  423. }
  424. if(!oldCameraDetail.getType().equals(cameraDetail.getType())){
  425. throw new BusinessException(ResultCode.CAMERA_TYPE_NOT_ERROR);
  426. }
  427. String dataSource = scenePro == null ? null :scenePro.getDataSource();
  428. if(scenePlus !=null){
  429. ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
  430. dataSource = scenePlusExt.getDataSource();
  431. }
  432. scenePlusService.updatePayStatus(num,scenePro == null ?"v4":"v3",scenePro == null?scenePlus.getSceneSource():scenePro.getSceneSource(),1,cameraDetail,user);
  433. scenePlusService.move(num,scenePro == null ?"v4":"v3",cameraDetail,user);
  434. if(sceneSource == 4 || sceneSource == 5){
  435. laserService.move(num, oldCamera.getSnCode(), camera.getSnCode(), user.getId(), dataSource);
  436. }
  437. sceneMoveLogService.saveLog(scenePro,scenePlus,camera.getSnCode(),oldCamera.getSnCode(),dataSource,dataSource, null);
  438. }
  439. }
  440. }