4dkankan-center-manage/src/main/java/com/fdkankan/manage/service/impl/SceneProServiceImpl.java

package com.fdkankan.manage.service.impl;

import cn.dev33.satoken.stp.StpUtil;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONUtil;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.common.constant.SceneConstant;
import com.fdkankan.fyun.face.FYunFileServiceInterface;
import com.fdkankan.manage.common.*;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.common.util.*;
import com.fdkankan.manage.httpClient.client.FdKKClient;
import com.fdkankan.manage.httpClient.client.OverallClient;
import com.fdkankan.manage.httpClient.service.LaserService;
import com.fdkankan.manage.entity.*;
import com.fdkankan.manage.httpClient.service.OverallService;
import com.fdkankan.manage.httpClient.vo.OverallVo;
import com.fdkankan.manage.mapper.ISceneProMapper;
import com.fdkankan.manage.mq.common.MqQueueUtil;
import com.fdkankan.manage.mq.param.ScenePayStatusVo;
import com.fdkankan.manage.mq.param.SceneRestStoreVo;
import com.fdkankan.manage.service.*;
import com.fdkankan.manage.util.CameraUtils;
import com.fdkankan.manage.util.DateUtils;
import com.fdkankan.manage.util.SceneStatusUtil;
import com.fdkankan.manage.vo.request.SceneParam;
import com.fdkankan.manage.vo.response.*;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.redis.util.RedisUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * <p>
 * pro场景表 服务实现类
 * </p>
 *
 * @author 
 * @since 2022-06-16
 */
@Service
@Slf4j
public class SceneProServiceImpl extends ServiceImpl<ISceneProMapper, ScenePro> implements ISceneProService {

    @Autowired
    private ICameraService cameraService;
    @Autowired
    ICameraDetailService cameraDetailService;
    @Autowired
    ISceneService sceneService;
    @Autowired
    ISceneCopyLogService copyLogService;
    @Autowired
    IUserIncrementService userIncrementService;
    @Autowired
    IIncrementTypeService incrementTypeService;
    @Autowired
    ISceneProEditService sceneProEditService;
    @Autowired
    FYunFileServiceInterface fYunFileServiceInterface;
    @Autowired
    IScenePlusService scenePlusService;
    @Autowired
    IScene3dNumService scene3dNumService;
    @Autowired
    ISceneCopyLogService sceneCopyLogService;
    @Autowired
    IScenePlusExtService scenePlusExtService;
    @Autowired
    LaserService laserService;
    @Autowired
    FdKKClient fdKKClient;
    @Autowired
    IFolderSceneService folderSceneService;
    @Autowired
    ISceneBuildProcessLogService sceneBuildProcessLogService;
    @Autowired
    ISceneMoveLogService sceneMoveLogService;
    @Autowired
    ISceneResourceCameraService sceneResourceCameraService;
    @Autowired
    ISceneDelLogService sceneDelLogService;
    @Autowired
    RabbitMqProducer rabbitMqProducer;
    @Autowired
    RedisUtil redisUtil;
    @Autowired
    ISceneColdStorageService sceneColdStorageService;
    @Autowired
    ICommonService commonService;
    @Autowired
    IUserService userService;
    @Autowired
    IMqSendLogService mqSendLogService;
    @Autowired
    OverallService overallService;


    @Override
    public ScenePro getByNum(String num) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePro::getNum,num);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        List<ScenePro> list = this.list(wrapper);
        if(list == null || list.size() <=0){
            return null;
        }
        return list.get(0);
    }

    @Override
    public HashMap<Long, Long> getCountGroupByUserId(List<Long> userIdList,Integer isObj) {
        HashMap<Long,Long> map = new HashMap<>();
        List<GroupByCount> result = this.getBaseMapper().getCountGroupByUserId(userIdList,isObj);
        result.forEach(entity ->map.put(entity.getId(),entity.getCount()));
        return map;
    }

    @Override
    public HashMap<Long, Long> getCountGroupByCameraId(List<Long> cameraIds) {
        HashMap<Long,Long> map = new HashMap<>();
        List<GroupByCount> result = this.getBaseMapper().getCountGroupByCameraId(cameraIds);
        result.forEach(entity ->map.put(entity.getId(),entity.getCount()));
        return map;
    }

    @Override
    public void unbindCamera(Long cameraId) {
        LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
        wrapper.set(ScenePro::getUserId,null)
                .eq(ScenePro::getCameraId,cameraId);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        this.update(wrapper);
    }

    @Override
    public List<ScenePro> getListByCameraId(Long cameraId) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePro::getCameraId,cameraId);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        return this.list(wrapper);
    }

    /**
     * @param payStatus -2 封存，为 1 解封
     */
    @Override
    public void lockOrUnLockBySpace(CameraDetail cameraDetail) {
        if(cameraDetail == null ){
            return;
        }
        if(!"aws".equals(CacheUtil.uploadType) && ( cameraDetail.getType() ==10 || cameraDetail.getType() == 11)){
            return;
        }
        ScenePayStatusVo scenePayStatusVo = new ScenePayStatusVo(cameraDetail.getCameraId());
        Map<String, Object> map = BeanUtil.beanToMap(scenePayStatusVo);
        rabbitMqProducer.sendByWorkQueue(MqQueueUtil.ucenterScenePayStatusQueue, map);
    }

    @Override
    public void lockOrUnLockBySpace(Long cameraId) {
        CameraDetail cameraDetail = cameraDetailService.getByCameraId(cameraId);
        this.lockOrUnLockBySpace(cameraDetail);
    }


    /**
     * 从oss中获取文件，并重写,上传替换
     */
    private void updateOssStatus(String path,Integer payStatus,String updateStatus) {
        String localPath = String.format(OssPath.localStatusPath, path);
        try {
            if(!fYunFileServiceInterface.fileExist(path)){
                return;
            }
            String data = fYunFileServiceInterface.getFileContent(path);
            if(StringUtils.isBlank(data)){
                return;
            }
            JSONObject jsonObject = JSONObject.parseObject(data);
            jsonObject.put(updateStatus,payStatus);
            String json = JSONUtil.toJsonStr(jsonObject);
            FileUtils.writeFile(localPath,json);
            log.info("updateOssStatus--localPath:{},ossPath:{}",localPath,path);
            fYunFileServiceInterface.uploadFile(localPath,path);
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            FileUtil.del(localPath);
        }
    }


    @Override
    public PageInfo pageList(SceneParam param) {
        if(param.getType() == 3){ //双目lite
           return sceneService.pageList(param);
        }
        //优化查询，去掉left join
        //snCode 查询，userName 查询，companyId查询
        //snCode回显，userName回显
        if(StringUtils.isNotBlank(param.getSnCode())){
            List<Camera> cameraList = cameraService.getLikeBySnCode(param.getSnCode());
            List<Long> cameraIds = cameraList.stream().map(Camera::getId).collect(Collectors.toList());
            param.setCameraIds(cameraIds);
        }
        if(StringUtils.isNotBlank(param.getUserName())){
            List<User> userList = userService.getLikeByUserName(param.getUserName());
            List<Long> userIds = userList.stream().map(User::getId).collect(Collectors.toList());
            param.setUserIds(userIds);
        }
        if(param.getCompanyId() != null){
            List<CameraDetail> details = cameraDetailService.getListByCompanyId(param.getCompanyId());
            List<Long> cameraIds = details.stream().map(CameraDetail::getCameraId).collect(Collectors.toList());
            if(param.getCameraIds() == null || param.getCameraIds().isEmpty()){
                param.setCameraIds(cameraIds);
            }else {
                List<Long> intersection = cameraIds.stream().filter(param.getCameraIds()::contains).collect(Collectors.toList());
                param.setCameraIds(intersection);
            }
        }
        if((param.getCameraIds() != null && param.getCameraIds().size()<=0) || (param.getUserIds() != null && param.getUserIds().size()<=0)){
            param.setNum("empty");
        }

        Page<SceneVo> page = this.getBaseMapper().pageList(new Page<>(param.getPageNum(),param.getPageSize()),param);
        List<SceneVo> records = page.getRecords();
        HashMap<String,SceneCopyLog> map = null;
        HashMap<String,SceneColdStorage> coldStorageMap = null;
        HashMap<Long, User> userMap = null;
        HashMap<Long, Camera> cameraMap = null;
        HashMap<String,JSONObject> ssSceneMap = null;
        HashMap<String,MqSendLog> mqSendLogMap = null;

        List<String> sceneNumList = page.getRecords().stream().map(SceneVo::getNum).collect(Collectors.toList());
        if(!records.isEmpty()){
            map = copyLogService.getByNewNumList(sceneNumList);
            coldStorageMap = sceneColdStorageService.getByNumList(sceneNumList);
            List<Long> userIds = records.parallelStream().map(SceneVo::getUserId).collect(Collectors.toList());
            userMap = userService.getByIds(userIds);
            List<Long> cameraIds = records.parallelStream().map(SceneVo::getCameraId).collect(Collectors.toList());
            cameraMap = cameraService.getByIds(cameraIds);

            if(param.getType() == 2 ){
                ssSceneMap = laserService.list(sceneNumList,4);
            }
            if(param.getType() == 6){
                ssSceneMap = laserService.list(sceneNumList,5);
            }
            List<String> modelingNumList = page.getRecords().stream().filter(e -> e.getStatus() == 0).map(SceneVo::getNum).collect(Collectors.toList());
            if(!modelingNumList.isEmpty()){
                mqSendLogMap = mqSendLogService.getByNumList(modelingNumList);
            }
        }
        for (SceneVo record : page.getRecords()) {
            if(param.getType() == 2 || param.getType() == 6){
                JSONObject ssObj = ssSceneMap.get(record.getNum());
                if(ssObj!=null){
                    record.setSceneName(ssObj.getString("title"));
                    record.setThumb(ssObj.getString("thumb"));
                    record.setWebSite(ssObj.getString("webSite"));
                    record.setIsObj(ssObj.getInteger("buildObjStatus"));
                    record.setViewCount(ssObj.getInteger("viewCount"));
                    record.setStatus(toFdStatus(ssObj.getInteger("status")));
                }else{
                    record.setStatus(-1);
                    record.setWebSite(null);
                }
            }

            record.setStatusString(SceneStatusUtil.getStatusString(record));
            if(userMap != null){
                User user = userMap.get(record.getUserId());
                if(user != null){
                    record.setUserName(user.getUserName());
                }
            }
            if(cameraMap != null){
                Camera camera = cameraMap.get(record.getCameraId());
                if(camera != null){
                    record.setSnCode(camera.getSnCode());
                }
            }
            if(map !=null ){
                SceneCopyLog sceneCopyLog = map.get(record.getNum());
                if(sceneCopyLog != null){
                    record.setCopyTime(sceneCopyLog.getCreateTime());
                    record.setIsCopy(true);
                }
            }
            if(coldStorageMap != null){
                SceneColdStorage sceneColdStorage = coldStorageMap.get(record.getNum());
                if(sceneColdStorage != null){
                    record.setIsColdStorage(true);
                }
            }
            if(record.getStatus() == -1){  //计算失败
                SceneBuildProcessLog sceneBuildProcessLog = sceneBuildProcessLogService.getByNum(record.getNum());
                if(sceneBuildProcessLog != null){
                    record.setSceneBuildProcessLog(sceneBuildProcessLog);
                    record.setBuildErrorReason(SceneBuildProcessLogEnum.getReason(sceneBuildProcessLog.getProcess()));
                }
            }
            //{"latitude":22.3672085,"longitude":113.595673,"altitude":9.275519,"horizontalAccuracy":65.0,"verticalAccuracy":10.0,"timestamp":1564381147.2775609}
            record.setAddressComponent(commonService.getAddressComponent(record.getGps()));

            if(mqSendLogMap!=null){
                MqSendLog mqSendLog = mqSendLogMap.get(record.getNum());
                if(mqSendLog != null){
                    record.setStatusString("排队中");
                    record.setStatus(-4);
                }
            }
        }
        return PageInfo.PageInfo(page);
    }

    private Integer toFdStatus(Integer status) {
        //深时状态,-1:场景被删 0:计算中 1计算失败 2计算成功 3封存 4生成OBJ中
        switch (status) {
            case 0:
            case 4:
                return 0;
            case 2:
                return -2;
            case 3:
                return -3;
            default:
                return -1;
        }
    }
    @Override
    public synchronized Integer move(SceneParam param) {
        Camera camera = cameraService.getBySnCode(param.getSnCode());
        if(camera == null){
            throw new BusinessException(ResultCode.CAMERA_SN_NOT_EXIST);
        }
        CameraDetail cameraDetail = cameraDetailService.getByCameraId(camera.getId());
        if(cameraDetail == null){
            throw new BusinessException(ResultCode.CAMERA_SN_NOT_EXIST);
        }
        if(StringUtils.isNotBlank(param.getNum())){
           moveScene(param.getNum(),camera,cameraDetail);
           return 1;
        }
        Integer successNum = param.getNumList().size();
        Integer errorNum = 0;
        for (String num : param.getNumList()) {
            try {
                moveScene(num,camera,cameraDetail);
            }catch (Exception e){
                errorNum ++;
                log.info("move-scene-error:num:{},errorNum:{},error:{}",num,errorNum,e.getMessage());
            }
        }
        return successNum - errorNum;
    }

    private void moveScene(String num,Camera camera,CameraDetail cameraDetail){
        ScenePro scenePro = this.getByNum(num);
        ScenePlus scenePlus = scenePlusService.getByNum(num);
        if((scenePro == null || scenePro.getCameraId() == null ) && (scenePlus== null || scenePlus.getCameraId() == null)){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
        }
        Integer status =  scenePro == null ? scenePlus.getSceneStatus() : scenePro.getStatus();
        if(status != -2){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5037, SceneConstant.FAILURE_MSG_5037);
        }

        Long sceneCameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
        Long space = scenePro == null ? 0 :scenePro.getSpace();
        String dataSource = scenePro == null ? null :scenePro.getDataSource();
        if(scenePlus !=null){
            ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
            space = scenePlusExt.getSpace();
            dataSource = scenePlusExt.getDataSource();
        }
        space = space == null ? 0 :space;
        cameraDetailService.initSpace(cameraDetail.getCameraId());
        Boolean checkSpace = cameraDetailService.checkSpace(cameraDetail, space);
        //深时场景无限容量
        if( !checkSpace){
            throw new BusinessException(ResultCode.CAMERA_SPACE_ERROR);
        }

        Camera oldCamera = cameraService.getById(sceneCameraId);
        if(oldCamera == null){
            throw new BusinessException(ResultCode.CAMERA_NOT_EXIST);
        }
        CameraDetail oldCameraDetail = cameraDetailService.getByCameraId(oldCamera.getId());
        if(oldCameraDetail == null){
            throw new BusinessException(ResultCode.CAMERA_NOT_EXIST);
        }
        if(oldCameraDetail.getCameraId().equals(cameraDetail.getCameraId())){
            throw new BusinessException(ResultCode.CAMERA_NOT_MOVE);
        }
        if(!oldCameraDetail.getType().equals(cameraDetail.getType())){
            throw new BusinessException(ResultCode.CAMERA_TYPE_NOT_ERROR);
        }

        Long oldUseSpace = oldCameraDetail.getUsedSpace() - space < 0 ? 0 : oldCameraDetail.getUsedSpace() - space;
        if("GB".equals(oldCameraDetail.getUnit())){
            oldUseSpace = 1L;
        }
        Long subSpace =  oldCameraDetail.getTotalSpace() - oldUseSpace;

        if(scenePro!=null){
            LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
            wrapper.eq(ScenePro::getId,scenePro.getId());
            wrapper.set(ScenePro::getCameraId,camera.getId());
            wrapper.set(ScenePro::getUserId,cameraDetail.getUserId());
            // wrapper.set(ScenePro::getStatus,0);
            // this.updateOssStatus(String.format(OssPath.v3_statusPath,scenePro.getNum()),0,"status");
            this.update(wrapper);
            //场景迁移到另外的相机清除本身在的文件夹
            folderSceneService.delBySceneId(scenePro.getId());
        }
        if(scenePlus!=null){
            LambdaUpdateWrapper<ScenePlus> wrapper = new LambdaUpdateWrapper<>();
            wrapper.eq(ScenePlus::getId,scenePlus.getId());
            wrapper.set(ScenePlus::getCameraId,camera.getId());
            wrapper.set(ScenePlus::getUserId,cameraDetail.getUserId());
            //wrapper.set(ScenePlus::getSceneStatus,0);
            //this.updateOssStatus(String.format(OssPath.v4_statusPath,scenePlus.getNum()),0,"status");
            scenePlusService.update(wrapper);
            folderSceneService.delBySceneId(scenePlus.getId());
        }
        //协作相机
        sceneResourceCameraService.setCooperationUser(cameraDetail,num);

        //String newDataSource = updateFdageNewDataSource(scenePro, scenePlus, oldCamera.getSnCode(), param.getSnCode(), dataSource);

        if(cameraDetail.getType() == 10 || cameraDetail.getType() == 11){       //深时场景
            //迁移深时 dataSource
            //FileUtil.move(new File(dataSource +"_laserData"),new File(newDataSource+"_laserData"),true);
            laserService.move(num,oldCamera.getSnCode(),camera.getSnCode(),cameraDetail.getUserId(),dataSource);
        }
        cameraDetailService.initSpace(oldCameraDetail);
        cameraDetailService.initSpace(cameraDetail);

        if(!"aws".equals(CacheUtil.uploadType) && oldCameraDetail.getType() != 11 && oldCameraDetail.getType() != 10 && subSpace >0){           //有剩余容量解封容量内场景
            this.lockOrUnLockBySpace(oldCameraDetail.getCameraId());
        }
        if(!"aws".equals(CacheUtil.uploadType) && cameraDetail.getType() != 11 && cameraDetail.getType() != 10){           //有剩余容量解封容量内场景
            this.lockOrUnLockBySpace(cameraDetail.getCameraId());
        }

        if("aws".equals(CacheUtil.uploadType) && subSpace >0){           //有剩余容量解封容量内场景
            this.lockOrUnLockBySpace(oldCameraDetail.getCameraId());
        }
        if("aws".equals(CacheUtil.uploadType)){           //有剩余容量解封容量内场景
            this.lockOrUnLockBySpace(cameraDetail.getCameraId());
        }
        sceneMoveLogService.saveLog(scenePro,scenePlus,camera.getSnCode(),oldCamera.getSnCode(),dataSource,dataSource, null);
        laserService.moveWenBao(num,oldCamera.getSnCode(),camera.getSnCode(),cameraDetail.getUserId(),dataSource);
    }



    private JSONObject updateFdageJson(JSONObject jsonObject,String newSnCode){
        JSONObject cam = jsonObject.getJSONObject("cam");
        if(cam != null){
            cam.put("uuid",newSnCode.toLowerCase());      //替换相机sn uuid
        }
        JSONArray points = jsonObject.getJSONArray("points");   //修改点位中的相机sn
        if(points !=null){
            for (Object point : points) {
                JSONObject jobj = (JSONObject) point;
                jobj.put("camera",newSnCode.toLowerCase());
            }
        }
        jsonObject.put("creator",newSnCode.toLowerCase());      //替换相机拍摄sn
        return jsonObject;
    }

    /**
     * 兼容，之前迁移，不使用旧snCode 替换新snCode。重写文件
     */
    public String updateFdageNewDataSource(ScenePro scenePro,ScenePlus scenePlus,String oldSnCode,String newSnCode,String dataSource) {
        String localPathFdage = null;
        try {
            newSnCode = newSnCode.toLowerCase();
            String fdagePaht =  dataSource.replace("/mnt/data","home") +"/data.fdage";
            if(!fYunFileServiceInterface.fileExist(fdagePaht)){
                return dataSource;
            }
            localPathFdage = String.format(OssPath.localFdagePath,fdagePaht);
            String fileContent = fYunFileServiceInterface.getFileContent(fdagePaht);
            JSONObject jsonObject = updateFdageJson(JSONObject.parseObject(fileContent), newSnCode);
            FileUtils.writeFile(localPathFdage ,JSONObject.toJSONString(jsonObject));

            String oldFdagePaht =  dataSource.replace("/mnt/data","home") ;
            String[] split = oldFdagePaht.split("/");
            String newFdagePath = split[0] +"/"+ newSnCode +"/"+ split[2] +"/"+ newSnCode +"_" + split[3].split("_")[1];
            String delPath = null;
            if(!oldFdagePaht.equals(newFdagePath)){
                log.info("updateFdageCopy--复制oss资源--oldFdagePaht:{},newFdagePath:{}",oldFdagePaht,newFdagePath);
                fYunFileServiceInterface.copyFileInBucket(oldFdagePaht,newFdagePath);
                delPath = oldFdagePaht;
            }
            fYunFileServiceInterface.uploadFile(localPathFdage,newFdagePath+"/data.fdage");
            log.info("updateFdage--localPathFdage:{},newFdagePath:{}",localPathFdage,newFdagePath);

            String newDataSource = newFdagePath.replace("home","/mnt/data");
            this.updateDataSource(scenePro,scenePlus,newDataSource);
            if(delPath != null){
                fYunFileServiceInterface.deleteFolder(oldFdagePaht);
            }
            sceneMoveLogService.saveLog(scenePro,scenePlus,oldSnCode,newSnCode,dataSource,newDataSource, fileContent);
            return newDataSource;
        }catch (Exception e){
            log.error("updateFdage-error:oldSnCode:{},newSnCode:{},dataSource:{}",oldSnCode,newSnCode,dataSource);
            log.error("updateFdage-error:",e);
            throw new BusinessException(ResultCode.MOVE_SCENE_ERROR);
        }finally {
            if(localPathFdage != null){
                FileUtil.del(localPathFdage);
            }
            if(scenePro!=null){
                LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
                wrapper.eq(ScenePro::getId,scenePro.getId());
                wrapper.set(ScenePro::getStatus,-2);
                this.update(wrapper);
                this.updateOssStatus(String.format(OssPath.v3_statusPath,scenePro.getNum()),-2,"status");

            }
            if(scenePlus!=null){
                LambdaUpdateWrapper<ScenePlus> wrapper = new LambdaUpdateWrapper<>();
                wrapper.eq(ScenePlus::getId,scenePlus.getId());
                wrapper.set(ScenePlus::getSceneStatus,-2);
                scenePlusService.update(wrapper);
                this.updateOssStatus(String.format(OssPath.v4_statusPath,scenePlus.getNum()),-2,"status");
            }
        }
    }


    private void updateDataSource(ScenePro scenePro,ScenePlus scenePlus, String newDataSource) {
        if(scenePro !=null){
            LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
            wrapper.eq(ScenePro::getId,scenePro.getId());
            wrapper.eq(ScenePro::getIsUpgrade,0);
            wrapper.set(ScenePro::getDataSource,newDataSource);
            this.update(wrapper);
        }
        if(scenePlus !=null){
            LambdaUpdateWrapper<ScenePlusExt> wrapperPlus = new LambdaUpdateWrapper<>();
            wrapperPlus.eq(ScenePlusExt::getPlusId,scenePlus.getId());
            wrapperPlus.set(ScenePlusExt::getDataSource,newDataSource);
            scenePlusExtService.update(wrapperPlus);
        }


    }

    @Override
    public void copy(String sceneNum) throws Exception {
        ScenePro scenePro = this.getByNum(sceneNum);
        ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
        if((scenePro == null || scenePro.getCameraId() == null ) && (scenePlus== null || scenePlus.getCameraId() == null)){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
        }
        Long cameraId = scenePro == null ? scenePlus.getCameraId() : scenePro.getCameraId();
        Camera camera = cameraService.getById(cameraId);
        if(camera == null){
            throw new BusinessException(CameraConstant.FAILURE_CODE_6029, CameraConstant.FAILURE_MSG_6029);
        }
        CameraDetail detailEntity = cameraDetailService.getByCameraId(cameraId);
        if(detailEntity == null){
            throw new BusinessException(CameraConstant.FAILURE_CODE_6029, CameraConstant.FAILURE_MSG_6029);
        }
        Integer sceneStatus = scenePro == null ? scenePlus.getSceneStatus() : scenePro.getStatus();
        if(sceneStatus != -2){
            throw new BusinessException(ResultCode.SCENE_IS_BUILDING);
        }

        Long needSpace = 0L;
        if(scenePro != null){
            needSpace = scenePro.getSpace();
        }
        if(scenePlus != null){
            ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
            if(scenePlusExt != null && scenePlusExt.getSpace() != null){
                needSpace = scenePlusExt.getSpace();
            }
        }
        Boolean checkSpace = cameraDetailService.checkSpace(detailEntity, needSpace);

        if(!checkSpace){
            throw new BusinessException(ResultCode.CAMERA_SPACE_ERROR);
        }

        HashMap<String, Object> param = new HashMap<>();
        param.put("num",sceneNum);
        JSONObject jsonObject = fdKKClient.copyScene(param, "m_a_n_a_g_e");
        Integer code = jsonObject.getInteger("code");
        if(code != 0){
            throw new BusinessException(ResultCode.CAMERA_SPACE_ERROR);
        }
    }

    @Override
    public void deleteByNum(String num) {
        Integer sceneSource = null;
        Long cameraId = null;
        ScenePro scenePro = this.getByNum(num);
        Scene scene = sceneService.getByNum(num);
        ScenePlus scenePlus = scenePlusService.getByNum(num);
        Integer status = null;

        if(scenePro!=null){
            cameraId = scenePro.getCameraId();
            sceneSource = scenePro.getSceneSource();
            status = scenePro.getStatus();
        }
        if(scene!=null){
            cameraId = scene.getCameraId();
            status = scene.getStatus();
        }
        if(scenePlus!=null){
            cameraId = scenePlus.getCameraId();
            sceneSource = scenePlus.getSceneSource();
            status = scenePlus.getSceneStatus();
        }
        HashMap<String, MqSendLog> mqSendLogHashMap = mqSendLogService.getByNumList(Arrays.asList(num));
        String redisKey = "mq-wait:queue:num:"+num;
        if(!mqSendLogHashMap.isEmpty()){
            for (String key : mqSendLogHashMap.keySet()) {
                if(!redisUtil.hasKey(redisKey)){
                    mqSendLogService.removeById(mqSendLogHashMap.get(key).getId());
                    status = -1;
                }
            }
        }
        if(status != null && status == 0){
            throw new BusinessException(ResultCode.CAN_NOT_DEL_MODELING_SCENE);
        }

        if(scenePro!=null){
            this.removeById(scenePro.getId());
        }
        if(scene!=null){
            sceneService.removeById(scene.getId());
        }
        if(scenePlus!=null){
            ScenePlusExt plusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
            if(plusExt != null){
                scenePlusExtService.delByPlus(scenePlus.getId());
            }
            scenePlusService.removeById(scenePlus.getId());
        }
        if(sceneSource != null && (sceneSource == 4 || sceneSource == 5)){
            laserService.delete(num);
        }
        overallService.delete(num);
        if(cameraId != null){
            cameraDetailService.initSpace(cameraId);
            this.lockOrUnLockBySpace(cameraId);
        }
        sceneDelLogService.saveLog(num, StpUtil.getLoginId());
    }

    @Override
    public Long getKkCount(List<String> asList, String startTime) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePro::getSceneSource,asList);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        wrapper.lt(ScenePro::getCreateTime,startTime);
        long count = this.count(wrapper);

        LambdaQueryWrapper<ScenePlus> wrapper2 = new LambdaQueryWrapper<>();
        wrapper2.in(ScenePlus::getSceneSource,asList);
        wrapper2.lt(ScenePlus::getCreateTime,startTime);
        long count1 = scenePlusService.count(wrapper2);
        return count + count1;
    }
    @Override
    public Long getSsCount(List<String> asList, String startTime) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePro::getSceneSource,asList);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        wrapper.lt(ScenePro::getCreateTime,startTime);
        LambdaQueryWrapper<ScenePlus> wrapper2 = new LambdaQueryWrapper<>();
        wrapper2.in(ScenePlus::getSceneSource,asList);
        wrapper2.lt(ScenePlus::getCreateTime,startTime);
        long count = scenePlusService.count(wrapper2);
        return this.count(wrapper) + count;
    }
    @Override
    public Long getSsObjCount(List<String> asList, String startTime) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePro::getSceneSource,asList);
        wrapper.eq(ScenePro::getIsUpgrade,0);
        if(StringUtils.isNotBlank(startTime)){
            wrapper.lt(ScenePro::getCreateTime,startTime);
        }
        wrapper.eq(ScenePro::getIsObj,1);
        LambdaQueryWrapper<ScenePlus> wrapper2 = new LambdaQueryWrapper<>();
        wrapper2.in(ScenePlus::getSceneSource,asList);
        if(StringUtils.isNotBlank(startTime)){
            wrapper2.lt(ScenePlus::getCreateTime,startTime);
        }
        List<ScenePlus> list = scenePlusService.list(wrapper2);
        List<Long> plusIds = list.stream().map(ScenePlus::getId).collect(Collectors.toList());
        long count = 0L;
        if(plusIds.size() >0){
            LambdaQueryWrapper<ScenePlusExt> wrapper3 = new LambdaQueryWrapper<>();
            wrapper3.in(ScenePlusExt::getPlusId,plusIds);
            wrapper3.eq(ScenePlusExt::getIsObj,1);
            count = scenePlusExtService.count(wrapper3);

        }
        return this.count(wrapper) + count;
    }

    @Override
    public HashMap<String, String> getSnCodeByNumList(Set<String> numList) {
        HashMap<String, Long> sceneMap = new HashMap<>();
        HashMap<Long, String> cameraMap = new HashMap<>();
        HashMap<String, String> snCodeMap = new HashMap<>();

        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePro::getIsUpgrade,0);
        wrapper.in(ScenePro::getNum,numList);
        List<ScenePro> list = this.list(wrapper);
        if(list.size() >0){
            list.forEach(entity -> sceneMap.put(entity.getNum(),entity.getCameraId()));
        }

        LambdaQueryWrapper<ScenePlus> wrapper2 = new LambdaQueryWrapper<>();
        wrapper2.in(ScenePlus::getNum,numList);
        List<ScenePlus> list2 = scenePlusService.list(wrapper2);
        if(list2.size() >0){
            list2.forEach(entity -> sceneMap.put(entity.getNum(),entity.getCameraId()));
        }
        if(sceneMap.size() <=0){
            return snCodeMap;
        }
        Set<Long> cameraIds = list.stream().map(ScenePro::getCameraId).collect(Collectors.toSet());
        Set<Long> cameraIds2 = list2.stream().map(ScenePlus::getCameraId).collect(Collectors.toSet());

        cameraIds.addAll(cameraIds2);

        if(cameraIds.size() >0){
            List<Camera> cameraList = cameraService.listByIds(cameraIds);
            cameraList.forEach(entity -> cameraMap.put(entity.getId(),entity.getSnCode()));
            for (String num : numList) {
                Long cameraId = sceneMap.get(num);
                if(cameraId != null){
                    snCodeMap.put(num,cameraMap.get(cameraId));
                }
            }
            List<CameraDetail> cameraIds1 = cameraDetailService.getByCameraIds(cameraList.stream().map(Camera::getId).collect(Collectors.toList()));
            for (CameraDetail cameraDetail : cameraIds1) {
                snCodeMap.put(cameraMap.get(cameraDetail.getCameraId()), CameraUtils.getCameraTypeStr(cameraDetail.getType()));
            }
        }

        return snCodeMap;
    }

    @Override
    public  HashMap<Long,Long > getSpaceGroupByCameraId() {
        List<GroupByCount> proSum = this.getBaseMapper().getProSpaceGroupByCameraId();
        HashMap<Long,Long > map = new HashMap<>();
        for (GroupByCount groupByCount : proSum) {
            map.merge(groupByCount.getId(), groupByCount.getCount()==null?0:groupByCount.getCount(), Long::sum);
        }
        List<GroupByCount > plusSum = this.getBaseMapper().getPlusSpaceGroupByCameraId();
        for (GroupByCount groupByCount : plusSum) {
            map.merge(groupByCount.getId(), groupByCount.getCount()==null?0:groupByCount.getCount(), Long::sum);
        }
        return map;
    }

    @Override
    public void rebuildScene(String num) {
        String redisKey2 = String.format(RedisKeyUtil.numShootKey, num);
        redisUtil.del(redisKey2);
        String dataSource = null;
        ScenePro scenePro = this.getByNum(num);
        if(scenePro!=null && scenePro.getSceneSource() != 4){
            throw new BusinessException(ResultCode.V3_SCENE_REBUILD);
        }
        ScenePlus scenePlus = scenePlusService.getByNum(num);
        if(scenePlus == null && scenePro == null){
            throw new BusinessException(ResultCode.SCENE_NOT_EXIST);
        }
        if(scenePro != null){
            dataSource = scenePro.getDataSource();
        }
        if(scenePlus != null){
            ScenePlusExt scenePlusExt= scenePlusExtService.getByPlusId(scenePlus.getId());
            if(scenePlusExt!= null){
                dataSource = scenePlusExt.getDataSource();
            }
        }
        if(StringUtils.isBlank(dataSource)){
            throw new BusinessException(ResultCode.SCENE_REBUILD_ERROR);
        }
        Long countByNewNum = sceneCopyLogService.getCountByNewNum(num);
        if(countByNewNum >0){
            throw new BusinessException(ResultCode.COPY_NUM_NOTREBUILD);
        }
        SceneColdStorage coldStorage = sceneColdStorageService.getByNum(num);
        if(coldStorage != null){
            throw new BusinessException(ResultCode.SCENE_REBUILD_ERROR2);
        }

        if(!fYunFileServiceInterface.fileExist(dataSource.replace("/mnt/data","home")+"/data.fdage")){
            throw new BusinessException(ResultCode.SCENE_REBUILD_ERROR);
        }
        HashMap<String,Object> paramMap = new HashMap<>();
        paramMap.put("num",num);
        paramMap.put("from","manage");
        try {
            JSONObject jsonObject = fdKKClient.rebuildScene(paramMap);
            Integer code = jsonObject.getInteger("code");
            if(code != 0){
                log.error("场景重算失败：{},{}",num,jsonObject);
                throw new BusinessException(ResultCode.SCENE_REBUILD_ERROR.code(), jsonObject.getString("msg"));
            }
        }catch (Exception e){
            throw new BusinessException(ResultCode.SCENE_REBUILD_ERROR);
        }

    }

    @Override
    public void restStore(String num) {
        if(StringUtils.isBlank(num)){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS);
        }
        String redisKey = String.format(RedisKeyUtil.restStoreKey, num);
        if(redisUtil.hasKey(redisKey)){
            throw new BusinessException(ResultCode.SCENE_STORE_ING);
        }
        SceneColdStorage sceneColdStorage = sceneColdStorageService.getByNum(num);
        if(sceneColdStorage == null){
            throw new BusinessException(ResultCode.SCENE_NOT_STORE);
        }
        String dataSource = scenePlusService.getDataSourceByNum(num);
        if(StringUtils.isBlank(dataSource)){
            throw new BusinessException(ResultCode.SCENE_NOT_EXIST);
        }
        String homePath = dataSource.replace("/mnt/data","home");
        SceneRestStoreVo sceneRestStoreVo = new SceneRestStoreVo(num,sceneColdStorage.getColdBucket(),sceneColdStorage.getBucket(),homePath);
        Map<String, Object> map = BeanUtil.beanToMap(sceneRestStoreVo);
        redisUtil.set(redisKey,homePath);
        rabbitMqProducer.sendByWorkQueue(MqQueueUtil.sceneRestoreQueue,map);
        //threadService.checkRestore(num,sceneColdStorage.getColdBucket(),sceneColdStorage.getBucket(),homePath);
    }

    @Override
    public void restStoreSuccess(String num) {
        Integer sceneResource = null;
        ScenePro scenePro = this.getByNum(num);
        Date time = DateUtils.dateAddOneMonth(new Date(),-11);
        String date = DateUtils.getDate(time);
        if(scenePro!=null ){
            sceneResource = scenePro.getSceneSource();
            LambdaUpdateWrapper<ScenePro> wrapper = new LambdaUpdateWrapper<>();
            wrapper.eq(ScenePro::getId,scenePro.getId());
            wrapper.set(ScenePro::getCreateTime,date);
            this.update(wrapper);
        }
        ScenePlus scenePlus = scenePlusService.getByNum(num);
        if(scenePlus != null ){
            sceneResource = scenePlus.getSceneSource();
            ScenePlusExt scenePlusExt = scenePlusExtService.getByPlusId(scenePlus.getId());
            if(scenePlusExt != null){
                LambdaUpdateWrapper<ScenePlusExt> wrapper = new LambdaUpdateWrapper<>();
                wrapper.eq(ScenePlusExt::getId,scenePlusExt.getId());
                wrapper.set(ScenePlusExt::getAlgorithmTime,date);
                scenePlusExtService.update(wrapper);
            }
        }
        //深时场景通知修改计算完成时间
        if(sceneResource != null && (sceneResource == 4 || sceneResource == 5)){
            HashMap<String,String> map = new HashMap<>();
            map.put("sceneCode",num);
            map.put("algorithmTime",date);
            rabbitMqProducer.sendByWorkQueue(MqQueueUtil.laserUnfreezeScene,map);
        }
    }

    @Override
    public Object getRestStoreProcess(String num) {
        if(StringUtils.isBlank(num)){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS);
        }
        SceneColdStorage sceneColdStorage = sceneColdStorageService.getByNum(num);
        if(sceneColdStorage == null){
            throw new BusinessException(ResultCode.SCENE_NOT_STORE);
        }
        String dataSource = scenePlusService.getDataSourceByNum(num);
        if(StringUtils.isBlank(dataSource)){
            throw new BusinessException(ResultCode.SCENE_NOT_EXIST);
        }
        String homePath = dataSource.replace("/mnt/data","home");
        return fYunFileServiceInterface.getRestoreFolderProcess(sceneColdStorage.getColdBucket(),homePath);
    }

    @Override
    public Long getSpaceSumByCameraId(Long cameraId) {
        return this.getBaseMapper().getSpaceSumByCameraId(cameraId);
    }

    @Override
    public Long getCountByCameraId(Long cameraId) {
        LambdaQueryWrapper<ScenePro> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePro::getIsUpgrade,0);
        wrapper.eq(ScenePro::getCameraId,cameraId);
        wrapper.in(ScenePro::getStatus,0,-2);
        return this.count(wrapper);
    }
}