4dkankan-center-manage/src/main/java/com/fdkankan/manage/httpClient/service/LaserService.java

package com.fdkankan.manage.httpClient.service;
import cn.hutool.core.bean.BeanUtil;
import com.fdkankan.manage.entity.*;
import com.fdkankan.manage.mq.common.MqQueueUtil;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.redis.util.RedisUtil;
import com.google.common.collect.Lists;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fdkankan.manage.common.*;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.manage.httpClient.client.LaserClient;
import com.fdkankan.manage.httpClient.param.LaserSceneMoveParam;
import com.fdkankan.manage.httpClient.param.LaserSceneParam;
import com.fdkankan.manage.httpClient.param.SSDownSceneParam;
import com.fdkankan.manage.httpClient.param.SsBindParam;
import com.fdkankan.manage.httpClient.vo.FdkkResponse;
import com.fdkankan.manage.service.*;
import com.fdkankan.manage.vo.request.SceneParam;
import com.fdkankan.manage.vo.response.SSDownSceneVo;
import com.fdkankan.manage.vo.response.SceneVo;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.*;
import java.util.stream.Collectors;

@Service
@Slf4j
public class LaserService {

   @Autowired
    LaserClient laserClient;
    @Value("${4dkk.laserService.basePath}")
    private String basePath;
    @Autowired
    IUserService userService;
    @Autowired
    ICameraDetailService cameraDetailService;
    @Autowired
    ICameraService cameraService;
    @Autowired
    ISceneBuildProcessLogService sceneBuildProcessLogService;
    @Autowired
    IScenePlusService scenePlusService;
    @Autowired
    IScenePlusExtService scenePlusExtService;
    @Autowired
    RabbitMqProducer rabbitMqProducer;
    @Autowired
    ICommonService commonService;
    @Autowired
    ISceneProService sceneProService;
    @Autowired
    RedisUtil redisUtil;
    @Autowired
    ISceneColdStorageService sceneColdStorageService;


    public PageInfo pageList(SceneParam param) {
        try {
            LaserSceneParam laserSceneParam = getLaserSceneParam(param);
            if(laserSceneParam == null ){
                return PageInfo.PageInfoEmpty(param.getPageNum(),param.getPageSize());
            }
            FdkkResponse response = laserClient.sceneList(laserSceneParam);
            JSONObject jsonObject =response.getData();
            if(jsonObject == null){
                return PageInfo.PageInfoEmpty(param.getPageNum(),param.getPageSize());
            }
            JSONArray list = jsonObject.getJSONArray("list");
            long total =jsonObject.getLong("total");

            List<SceneVo> sceneVoList = new ArrayList<>();
            for (Object o : list) {
                String res = JSONObject.toJSONString(o);
                SceneVo vo = JSONObject.parseObject(res,SceneVo.class);
                //深时状态,-1:场景被删 0:计算中 1计算失败 2计算成功 3封存 4生成OBJ中
                JSONObject obj = (JSONObject) o;
                vo.setStatusString(getLaserStatus(vo.getStatus(),vo.getPayStatus()));
                vo.setStatus(toFdStatus(vo.getStatus()));
                vo.setSceneName(obj.getString("title"));
                vo.setUserName(obj.getString("phone"));
                vo.setThumb(obj.getString("thumb"));
                vo.setWebSite(obj.getString("webSite"));
                vo.setIsObj(obj.getInteger("buildObjStatus"));
                if(vo.getStatus() == -1){  //计算失败
                    SceneBuildProcessLog sceneBuildProcessLog = sceneBuildProcessLogService.getByNum(vo.getNum());
                    if(sceneBuildProcessLog != null){
                        vo.setSceneBuildProcessLog(sceneBuildProcessLog);
                        vo.setBuildErrorReason(SceneBuildProcessLogEnum.getReason(sceneBuildProcessLog.getProcess()));
                        vo.setDataSource(scenePlusService.getDataSourceByNum(vo.getNum()));
                    }
                }
                this.getLaserSceneGps(vo);
                vo.setAddressComponent(commonService.getAddressComponent(vo.getGps()));

                if( vo.getMixture() == 0 && vo.getLocation() != null && vo.getLocation().equals(6)){
                    vo.setShootCount(0);
                }
                sceneVoList.add(vo);
            }

            HashMap<String,SceneColdStorage> coldStorageMap = null;
            if(sceneVoList.size() >0){
                List<String> numList = sceneVoList.parallelStream().map(SceneVo::getNum).collect(Collectors.toList());
                coldStorageMap = sceneColdStorageService.getByNumList(numList);
            }
            Set<String> snCodes = sceneVoList.parallelStream().map(SceneVo::getSnCode).collect(Collectors.toSet());

            HashMap<String,User> snUserMap = userService.getUserBySn(snCodes);




            for (SceneVo sceneVo : sceneVoList) {
                if(coldStorageMap != null){
                    SceneColdStorage sceneColdStorage = coldStorageMap.get(sceneVo.getNum());
                    if(sceneColdStorage != null){
                        sceneVo.setIsColdStorage(true);
                    }
                }
                User user = snUserMap.get(sceneVo.getSnCode());
                if(user != null){
                    sceneVo.setUserId(user.getId());
                    sceneVo.setUserName(user.getUserName());
                }
            }
            Page<SceneVo> voPage = new Page<>(param.getPageNum(),param.getPageSize());
            voPage.setRecords(sceneVoList);
            voPage.setTotal(total);
            return PageInfo.PageInfo(voPage);
        }catch (Exception e){
            log.error("访问深时失败：",e);
            //throw new BusinessException(ResultCode.SS_GET_ERROR);
        }
        Page<SceneVo> voPage = new Page<>(param.getPageNum(),param.getPageSize());
        return PageInfo.PageInfo(voPage);
    }


    private void getLaserSceneGps(SceneVo vo) {
//        String redisKey = String.format(RedisKeyUtil.numGpsKey, vo.getNum());
//        String redisKey2 = String.format(RedisKeyUtil.numShootKey, vo.getNum());
//        if(redisUtil.hasKey(redisKey) && redisUtil.hasKey(redisKey2)){
//            String gps = redisUtil.get(redisKey);
//            String shootNum = redisUtil.get(redisKey2);
//            vo.setGps(gps);
//            vo.setShootCount(Integer.valueOf(shootNum));
//            return;
//        }
        ScenePro pro = sceneProService.getByNum(vo.getNum());
        if(pro != null){
            //redisUtil.set(redisKey,pro.getGps()== null ?"":pro.getGps());
            //redisUtil.set(redisKey2,String.valueOf(pro.getShootCount()== null ? 0 :pro.getShootCount()));
            vo.setGps(pro.getGps());
            vo.setShootCount(pro.getShootCount());
        }
        ScenePlus plus = scenePlusService.getByNum(vo.getNum());
        if(plus != null){
            ScenePlusExt ext = scenePlusExtService.getByPlusId(plus.getId());
            if(ext != null){
                //redisUtil.set(redisKey,ext.getGps()== null?"":ext.getGps());
               // redisUtil.set(redisKey2,String.valueOf(ext.getShootCount()== null ? 0 :ext.getShootCount()));
                vo.setGps(ext.getGps());
                vo.setShootCount(ext.getShootCount());
            }
        }
    }

    private LaserSceneParam getLaserSceneParam(SceneParam param) {
        LaserSceneParam newParam = new LaserSceneParam();
        if(param.getCompanyId()!= null){ //客户场景
            List<CameraDetail> cameraDetails = cameraDetailService.getListByCompanyId(param.getCompanyId());
            param.setSnCodes(this.setSnCodes(cameraDetails));
            if(param.getSnCodes() == null || param.getSnCodes().size() <=0){
                return null;
            }
        }
        if(StringUtils.isNotBlank(param.getUserName())){
            List<CameraDetail> cameraDetails = cameraDetailService.getByUserName(param.getUserName());
            List<String> snCodes = this.setSnCodes(cameraDetails);
            if(param.getCompanyId() == null){
                param.setSnCodes(snCodes);
            }else {
                if(snCodes == null || snCodes.size() <=0){
                    return null;
                }
                //取交集
                List<String> intersection = param.getSnCodes().stream().filter(snCodes::contains).collect(Collectors.toList());
                param.setSnCodes(intersection);
            }
        }
        if(StringUtils.isNotBlank(param.getUserName()) && StringUtils.isBlank(param.getSnCode()) &&
                (param.getSnCodes() == null || param.getSnCodes().size() <=0)){
            param.setSnCode("phoneEmptySelect");
        }
        if(StringUtils.isNotBlank(param.getNum())){
            newParam.setSceneCode(param.getNum());
        }
        BeanUtils.copyProperties(param,newParam);
        newParam.setTitle(param.getSceneName());

        if(StringUtils.isNotBlank(param.getField()) && param.getField().equals("create_time")){
            newParam.setOrderBy("shoot_time");
        }else {
            newParam.setOrderBy(param.getField());
        }
        newParam.setSortBy(param.getOrder());
        if(param.getType() == 6){
            newParam.setSceneSource(5);
        }
        return newParam;
    }

    private List<String> setSnCodes(List<CameraDetail> cameraDetails) {
        if(cameraDetails != null && cameraDetails.size() >0){
            Set<Long> cameraIds = cameraDetails.stream()
                    .filter(entity -> entity.getType() == 10 || entity.getType() == 11).map(CameraDetail::getCameraId).collect(Collectors.toSet());
            if(cameraIds.size() >0){
                List<Camera> cameraList = cameraService.listByIds(cameraIds);
                return cameraList.stream().map(Camera::getSnCode).collect(Collectors.toList());
            }
        }
        return null;
    }

    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;
        }
    }

    public static String getLaserStatus(Integer status,Integer payStatus){
        if(payStatus != null && payStatus == -2){
            return "封存";
        }
        //深时状态,-1:场景被删 0:计算中 1计算失败 2计算成功 3封存 4生成OBJ中
        switch (status){
            case -1 : return "场景已删除";
            case 0 :
            case 4 :
                return "计算中";
            case 1 : return "计算失败";
            case 2 : return "计算成功";
            case 3 : return "封存";
            default: return "";
        }
    }

    public void move(String num, String snCode, String toSnCode,Long userId,String newDataSource) {
        LaserSceneMoveParam param = new LaserSceneMoveParam();
        param.setSceneCode(num);
        //param.setSnCode(snCode);
        param.setToSnCode(toSnCode);
        param.setUserId(userId);
        param.setDataSource(newDataSource+"_laserData/laserData");
        if(userId != null){
            User user = userService.getById(userId);
            if(user != null){
                param.setPhone(user.getUserName());
            }
        }
        Map<String, Object> map = BeanUtil.beanToMap(param);
        rabbitMqProducer.sendByWorkQueue(MqQueueUtil.laserMoveQueue,map);
        //laserClient.migrate(param);
    }
    public void moveWenBao(String num, String snCode, String toSnCode,Long userId,String newDataSource) {
        LaserSceneMoveParam param = new LaserSceneMoveParam();
        param.setSceneCode(num);
        //param.setSnCode(snCode);
        param.setToSnCode(toSnCode);
        param.setUserId(userId);
        param.setDataSource(newDataSource+"_laserData/laserData");
        if(userId != null){
            User user = userService.getById(userId);
            if(user != null){
                param.setPhone(user.getUserName());
            }
        }
        Map<String, Object> map = BeanUtil.beanToMap(param);
        rabbitMqProducer.sendByWorkQueue(MqQueueUtil.laserMoveWenBaoQueue,map);
        //laserClient.migrate(param);
    }


    public void copy(String snCode, String createTime, String newNum, Integer status, String sceneKey, String sceneName, Long userId){
        String phone = null;
        if(userId != null){
            User user = userService.getById(userId);
            if(user != null){
                phone = user.getUserName();
            }
        }
        Map<String,Object> params = new HashMap<>();
        params.put("childName",snCode);
        params.put("createTime", createTime);
        params.put("phone", phone);
        params.put("sceneCode", newNum);
        params.put("snCode",snCode);
        params.put("status", status);
        params.put("password", sceneKey);
        params.put("title", sceneName);
        params.put("userId", userId);
        params.put("copy", true);
        Result result = laserClient.saveOrEdit(newNum, params);
        if( result.getCode() != HttpStatus.OK.value()){
            log.error("激光场景状态同步失败!");
        }
    }

    public void delete(String num) {
        try {
            Map<String,Object> params = new HashMap<>();
            params.put("sceneCode", num);
            params.put("status", -1);
            Result result = laserClient.saveOrEdit(num, params);
            if(result.getCode() != HttpStatus.OK.value()){
                log.error("激光场景状态同步失败!");
            }
        }catch (Exception e){
            log.error("激光场景状态同步失败!",e);
        }

    }


    public void disableCooperation(String snCode,  String cooperationUserName){
        List<Map<String, String>> laserParams = new ArrayList<>();
        Map<String, String> param = new HashMap<>();
        param.put("snCode", snCode);
        param.put("cooperationUserName", cooperationUserName);
        laserParams.add(param);
        this.disableCooperation(laserParams);
    }

    public void disableCooperation(List<Map<String, String>> params) {
        if(params.size() <=0){
            return;
        }
        laserClient.cooperationDisable(params);
    }

    public SSDownSceneVo downOfflineSceneStatus(String num) {
        try {
            SSDownSceneVo vo ;
            SSDownSceneParam param = new SSDownSceneParam();
            param.setSceneCode(num);
            Result responseEntity = laserClient.downOfflineSceneStatus(param);
            if( responseEntity.getCode() != HttpStatus.OK.value()){
                log.error("downOfflineSceneStatus-根据场景码获取激光转台下载状态失败:{}",responseEntity);
                return null;
            }
            vo = JSONObject.parseObject(JSONObject.toJSONString(responseEntity.getData()), SSDownSceneVo.class);
            return vo;

        }catch (Exception e){
            log.error("downOfflineSceneStatus-根据场景码获取激光转台下载状态失败!",e);
        }
        return null ;
    }

    public SSDownSceneVo downOfflineScene(String num) {
        try {
            SSDownSceneVo vo ;
            SSDownSceneParam param = new SSDownSceneParam();
            param.setSceneCode(num);
            Result responseEntity = laserClient.downOfflineScene(param);
            if( responseEntity.getCode() != HttpStatus.OK.value()){
                log.error("downOfflineScene-根据场景码获取激光转台下载失败:{}",responseEntity);
                return null;
            }
            vo = JSONObject.parseObject(JSONObject.toJSONString(responseEntity.getData()), SSDownSceneVo.class);
            return vo ;
        }catch (Exception e){
            log.error("downOfflineScene-根据场景码获取激光转台下载状态失败!",e);
        }
        return null ;
    }
    public void toBind(String snCode) {
        try {
            SsBindParam param = new SsBindParam();
            param.setBind(false);
            param.setSnCode(Lists.newArrayList(snCode));
            Result responseEntity = laserClient.toBind(param);
            if( responseEntity.getCode() != HttpStatus.OK.value()){
                log.error("解绑用户激光转台下载失败:{}",responseEntity);
            }
        }catch (Exception e){
            log.error("解绑用户获取激光转台下载状态失败!",e);
        }
    }

}