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

package com.fdkankan.manage.service.impl;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.fdkankan.manage.common.RedisKeyUtil;
import com.fdkankan.manage.common.ResultCode;
import com.fdkankan.common.constant.SceneConstant;
import com.fdkankan.manage.common.DownloadStatusEnum;
import com.fdkankan.manage.exception.BusinessException;
import com.fdkankan.manage.entity.*;
import com.fdkankan.manage.httpClient.service.LaserService;
import com.fdkankan.manage.service.*;
import com.fdkankan.manage.vo.response.DownVo;
import com.fdkankan.manage.vo.response.DownloadProcessVo;
import com.fdkankan.manage.vo.response.SSDownSceneVo;
import com.fdkankan.redis.constant.RedisKey;
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.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

@Service
@Slf4j
public class DownService implements IDownService {

    @Autowired
    ISceneProService sceneProService;
    @Autowired
    ISceneProEditService sceneProEditService;
    @Autowired
    ISceneDownloadLogService sceneDownloadLogService;
    @Autowired
    IScenePlusService scenePlusService;
    @Autowired
    ISceneEditInfoService sceneEditInfoService;
    @Autowired
    RedisUtil redisUtil;
    @Autowired
    LaserService laserService;


    @Override
    public DownVo checkDownLoad(String sceneNum,Integer isObj) {
        if(StringUtils.isEmpty(sceneNum)){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS.code(),ResultCode.MISSING_REQUIRED_PARAMETERS.message());
        }
        ScenePro scenePro = sceneProService.getByNum(sceneNum);
        ScenePlus plus = scenePlusService.getByNum(sceneNum);
        if(scenePro == null && plus == null){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
        }
        Integer sceneType = scenePro !=null ? scenePro.getSceneSource() : plus.getSceneSource();
        log.info("checkDownLoad--sceneType:{},isObj:{}",sceneType,isObj);
        if((sceneType == 4 || sceneType == 5) && isObj !=1){ //深时场景
            return SSCheckDownload(sceneNum);
        }
        SceneDownloadLog sceneDownloadLog;
        int isUp = 0;
        if(scenePro == null){
            isUp = 1;
        }
        Integer sceneVersion = getSceneVersion(scenePro, plus);

        sceneDownloadLog = sceneDownloadLogService.getByStatusAndNum(sceneNum,0,isUp);
        DownVo downVo = new DownVo();
        if(sceneDownloadLog != null){
            downVo.setDownloadStatus(1);
            return downVo;
        }
        sceneDownloadLog = sceneDownloadLogService.getByStatusAndNum(sceneNum,1,isUp);
        //3下载过，并且没有修改过
        if(sceneDownloadLog != null && sceneDownloadLog.getSceneVersion().intValue() == sceneVersion){
            downVo.setDownloadStatus(3);
            downVo.setDownloadUrl(sceneDownloadLog.getDownloadUrl());
            return downVo;
        }
        //下载过，有更改
        if(sceneDownloadLog != null){
            String redisKey = RedisKey.PREFIX_DOWNLOAD_PROGRESS;
            if(isUp == 1){
                redisKey = RedisKey.PREFIX_DOWNLOAD_PROGRESS_V4;
            }
            downVo.setDownloadStatus(2);
            redisUtil.del(String.format(redisKey,sceneNum));  // 清除旧的下载信息
            return downVo;
        }
        return downVo;
    }

    private Integer getSceneVersion(ScenePro scenePro, ScenePlus scenePlus) {
        Integer version = 0;
        if(scenePro != null){
            SceneProEdit proEdit = sceneProEditService.getByProId(scenePro.getId());
            if(proEdit == null){
                throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
            }
            version = proEdit.getVersion();
        }
        if(scenePro == null && scenePlus !=null){
            String redisKey = String.format(RedisKeyUtil.SCENE_VERSION,scenePlus.getNum());
            if(!redisUtil.hasKey(redisKey) || StringUtils.isBlank(redisUtil.get(redisKey))){
                SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(scenePlus.getId());
                if(sceneEditInfo == null){
                    throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
                }
                version = sceneEditInfo.getVersion();
            }else {
                String redisObj = redisUtil.get(redisKey);
                JSONObject obj = JSONObject.parseObject(redisObj);
                version = obj.getInteger("version");
            }
        }
        return version;
    }

    @Override
    public DownVo down(String sceneNum,Integer isObj) {
        if(StringUtils.isEmpty(sceneNum) ){
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS.code(),ResultCode.MISSING_REQUIRED_PARAMETERS.message());
        }
        ScenePro scenePro = sceneProService.getByNum(sceneNum);
        ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
        if(scenePro == null && scenePlus == null){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
        }
        Long userId = scenePro == null ? scenePlus.getUserId() : scenePro.getUserId();

        Integer sceneType = scenePro !=null ? scenePro.getSceneSource() : scenePlus.getSceneSource();
        log.info("down--sceneType:{},isObj:{}",sceneType,isObj);
        if((sceneType == 4 || sceneType == 5) && isObj !=1){ //深时场景
            return SSDownload(sceneNum,userId);
        }

        DownVo downVo = new DownVo();
        Integer sceneVersion = getSceneVersion(scenePro, scenePlus);

        Map<String,String> params = new HashMap<>(2);
        params.put("type","local");
        params.put("sceneNum",sceneNum);
        String redisKey = RedisKey.DOWNLOAD_TASK;
        String sysVersion = "v3";
        if(scenePro == null){
            params.put("num",sceneNum);
            redisKey = RedisKey.SCENE_DOWNLOADS_TASK_V4;
            sysVersion = "v4";
        }
        redisUtil.lRightPush(redisKey, JSONObject.toJSONString(params));
        SceneDownloadLog sceneDownloadLogEntity = new SceneDownloadLog();
        sceneDownloadLogEntity.setUserId(userId);
        sceneDownloadLogEntity.setSceneNum(sceneNum);
        sceneDownloadLogEntity.setSceneVersion(sceneVersion);
        sceneDownloadLogEntity.setStatus(0);
        sceneDownloadLogEntity.setSysVersion(sysVersion);
        sceneDownloadLogService.save(sceneDownloadLogEntity);
        downVo.setDownloadStatus(1);
        return downVo;
    }

    @Override
    public DownloadProcessVo downloadProcess(String sceneNum,Integer isObj) {
        if (StringUtils.isEmpty(sceneNum)) {
            throw new BusinessException(ResultCode.MISSING_REQUIRED_PARAMETERS.code(),ResultCode.MISSING_REQUIRED_PARAMETERS.message());
        }
        ScenePro scenePro = sceneProService.getByNum(sceneNum);
        ScenePlus scenePlus = scenePlusService.getByNum(sceneNum);
        if(scenePro == null && scenePlus == null){
            throw new BusinessException(SceneConstant.FAILURE_CODE_5005, SceneConstant.FAILURE_MSG_5005);
        }
        Integer sceneType = scenePro !=null ? scenePro.getSceneSource() : scenePlus.getSceneSource();
        log.info("downloadProcess--sceneType:{},isObj:{}",sceneType,isObj);
        if((sceneType == 4 || sceneType == 5) && isObj !=1){ //深时场景
            return SSDownloadProcess(sceneNum);
        }
        String redisKey = RedisKey.PREFIX_DOWNLOAD_PROGRESS;
        if(scenePro == null){
            redisKey = RedisKey.PREFIX_DOWNLOAD_PROGRESS_V4;
        }
        // 获取下载进度
        String result = redisUtil.get(String.format(redisKey,sceneNum));
        if(StringUtils.isEmpty(result)){
            return new DownloadProcessVo();
        }
        int isUp = 0;
        if(scenePro == null){
            isUp = 1;
        }
        SceneDownloadLog sceneDownloadLog = sceneDownloadLogService.getByStatusAndNum(sceneNum,0,isUp);

        DownloadProcessVo downloadProcessVo = JSONObject.parseObject(result, DownloadProcessVo.class);

        if(sceneDownloadLog != null){
            switch (downloadProcessVo.getStatus()) {
                case DownloadStatusEnum.DOWNLOAD_SUCCESS_CODE:
                    String url = downloadProcessVo.getUrl();
                    if (!StringUtils.isEmpty(url)) {
                        sceneDownloadLog.setDownloadUrl(url);
                        sceneDownloadLog.setStatus(1);
                        break;
                    }
                case DownloadStatusEnum.DOWNLOAD_FAILED_CODE:
                    sceneDownloadLog.setStatus(-1);
                    break;
            }
            sceneDownloadLogService.updateById(sceneDownloadLog);
        }
        return downloadProcessVo;
    }

    /**
     * status ：离线包状态是否需要重新生成  0 未生成：1 不需要  2需要   3 生成中
     */
    private DownVo SSCheckDownload(String sceneNum) {
        DownVo downVo = new DownVo();
        SSDownSceneVo vo = laserService.downOfflineSceneStatus(sceneNum);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }
        downVo.setDownloadStatus(0);
        if(vo.getStatus() == 1){
            downVo.setDownloadStatus(3);
            downVo.setDownloadUrl(vo.getUrl());
        }
        if(vo.getStatus() == 2){
            downVo.setDownloadStatus(2);
        }
        if(vo.getStatus() == 3){
            downVo.setDownloadStatus(1);
        }
        return downVo;
    }

    /**
     * downloadStatus -1下载失败  1下载成功
     */
    private DownVo SSDownload(String sceneNum,Long userId) {
        DownVo downVo = new DownVo();

        //status ：0:正在生成 1，初次生成  2，已经生成直接下载 3，重新生成
        SSDownSceneVo vo = laserService.downOfflineScene(sceneNum);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }

        SceneDownloadLog sceneDownloadLogEntity = new SceneDownloadLog();
        sceneDownloadLogEntity.setUserId(userId);
        sceneDownloadLogEntity.setSceneNum(sceneNum);
        sceneDownloadLogEntity.setSceneVersion(0);
        sceneDownloadLogEntity.setStatus(0);
        sceneDownloadLogEntity.setSysVersion("ss");
        sceneDownloadLogService.save(sceneDownloadLogEntity);
        downVo.setDownloadStatus(1);
        return downVo;
    }

    public static HashMap<String,Integer> ssNumProcessNumMap = new HashMap<>();

    private DownloadProcessVo SSDownloadProcess(String sceneNum) {
        DownloadProcessVo downVo = new DownloadProcessVo();
        SSDownSceneVo vo = laserService.downOfflineSceneStatus(sceneNum);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }
        downVo.setStatus(1003);

        if(vo.getStatus() == 0 || vo.getStatus() == 2 || vo.getStatus() == 3){    //下载中
            ssNumProcessNumMap.merge(sceneNum, 1, Integer::sum);
            Integer percent = ssNumProcessNumMap.get(sceneNum);
            percent = percent /2;
            if(percent >50){
                percent = 50;
            }
            downVo.setStatus(1001);
            downVo.setPercent(percent);
        }
        if(vo.getStatus() == 1){    //下载完成
            ssNumProcessNumMap.remove(sceneNum);
            downVo.setPercent(100);
            downVo.setUrl(vo.getUrl());
            downVo.setStatus(1002);

            LambdaUpdateWrapper<SceneDownloadLog> wrapper = new LambdaUpdateWrapper<>();
            wrapper.eq(SceneDownloadLog::getSceneNum,sceneNum);
            wrapper.eq(SceneDownloadLog::getStatus,0);
            wrapper.set(SceneDownloadLog::getDownloadUrl,vo.getUrl());
            wrapper.set(SceneDownloadLog::getStatus,1);
            sceneDownloadLogService.update(wrapper);
        }
        return downVo;
    }


    @Override
    public DownVo checkDownLoadE57(String num) {
        DownVo downVo = new DownVo();
        SSDownSceneVo vo = laserService.downE57Status(num);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }
        downVo.setDownloadStatus(0);
        if(vo.getStatus() == 1){
            downVo.setDownloadStatus(3);
            downVo.setDownloadUrl(vo.getUrl());
        }
        if(vo.getStatus() == 2){
            downVo.setDownloadStatus(2);
        }
        if(vo.getStatus() == 3){
            downVo.setDownloadStatus(1);
        }
        return downVo;
    }

    @Override
    public DownVo downE57(String num) {
        DownVo downVo = new DownVo();

        //status ：0:正在生成 1，初次生成  2，已经生成直接下载 3，重新生成
        SSDownSceneVo vo = laserService.downE57(num);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }
        downVo.setDownloadStatus(1);
        return downVo;
    }

    public static HashMap<String,Integer> ssNumProcessNumE57Map = new HashMap<>();


    @Override
    public DownloadProcessVo downloadProcessE57(String num) {
        DownloadProcessVo downVo = new DownloadProcessVo();
        SSDownSceneVo vo = laserService.downE57Status(num);
        if(vo == null){
            throw new BusinessException(ResultCode.SS_SCENE_DOWN_ERROR);
        }
        downVo.setStatus(1003);
        if(vo.getStatus() == 0 || vo.getStatus() == 2 || vo.getStatus() == 3){    //下载中
            ssNumProcessNumE57Map.merge(num, 1, Integer::sum);
            Integer percent = ssNumProcessNumE57Map.get(num);
            percent = percent /2;
            if(percent >50){
                percent = 50;
            }
            downVo.setStatus(1001);
            downVo.setPercent(percent);
        }
        if(vo.getStatus() == 1){    //下载完成
            ssNumProcessNumE57Map.remove(num);
            downVo.setPercent(100);
            downVo.setUrl(vo.getUrl());
            downVo.setStatus(1002);
        }
        return downVo;
    }
}