4dkankan-parent/4dkankan-center-scene/src/main/java/com/fdkankan/scene/service/impl/SceneUpgradeToV4Service.java

package com.fdkankan.scene.service.impl;

import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.fdkankan.common.constant.ErrorCode;
import com.fdkankan.common.response.ResultData;
import com.fdkankan.rabbitmq.util.RabbitMqProducer;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.redis.constant.RedisLockKey;
import com.fdkankan.redis.util.RedisLockUtil;
import com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.scene.bean.SceneUpgradeProgressBean;
import com.fdkankan.scene.entity.SceneProEditV3;
import com.fdkankan.scene.entity.SceneProV3;
import com.fdkankan.scene.entity.SceneRepairLog;
import com.fdkankan.scene.mapper.ISceneUpgradeMapper;
import com.fdkankan.scene.service.ISceneProEditV3Service;
import com.fdkankan.scene.service.ISceneProV3Service;
import com.fdkankan.scene.service.ISceneRepairLogService;
import com.fdkankan.scene.service.ISceneUpgradeToV4Service;
import com.fdkankan.scene.vo.UpgradeToV4ParamVO;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

/**
 * <p>
 * TODO
 * </p>
 *
 * @author dengsixing
 * @since 2022/4/21
 **/
@Slf4j
@Service
public class SceneUpgradeToV4Service implements ISceneUpgradeToV4Service {

    @Value("${queue.scene.upgrade-to-v4}")
    private String upgradeToV4;

    @Autowired
    private ISceneProV3Service sceneProV3Service;
    @Autowired
    private ISceneProEditV3Service sceneProEditV3Service;
    @Autowired
    private ISceneUpgradeMapper sceneUpgradeMapper;
    @Autowired
    private RedisLockUtil redisLockUtil;
    @Autowired
    private ISceneRepairLogService sceneRepairLogService;
    @Autowired
    private RabbitMqProducer rabbitMqProducer;
    @Autowired
    private RedisUtil redisUtil;


    @Override
    public ResultData upgradeToV4(UpgradeToV4ParamVO param) {

        SceneProV3 sceneProV3 = JSON.parseObject(param.getScenePro(), SceneProV3.class);
        SceneProEditV3 sceneProEditV3 = JSON.parseObject(param.getSceneProEdit(), SceneProEditV3.class);

        //加锁
        String lockKey = String.format(RedisLockKey.LOCK_SCENE_UPGRADE_V4, sceneProV3.getNum());
        boolean lock = redisLockUtil.lock(lockKey,RedisKey.EXPIRE_TIME_2_HOUR);
        if(!lock){
            return ResultData.error(ErrorCode.FAILURE_CODE_7019);
        }

        try {
            //查询升级日志，如果已经升级成功，不允许重复升级
            SceneRepairLog sceneRepairLog = sceneRepairLogService.getOne(
                new LambdaQueryWrapper<SceneRepairLog>()
                    .eq(SceneRepairLog::getNum, sceneProV3.getNum()));
            if(Objects.nonNull(sceneRepairLog)){
                if(sceneRepairLog.getState() == 0)
                    return ResultData.error(ErrorCode.FAILURE_CODE_7019);
                if(sceneRepairLog.getState() == 1)
                    return ResultData.error(ErrorCode.FAILURE_CODE_7020);
            }

            //写入sceneProV3
            sceneProV3Service.saveOrUpdate(sceneProV3);

            //写入sceneProEditV3
            sceneProEditV3Service.saveOrUpdate(sceneProEditV3);

            //写入写入sceneProEdit、写入sceneProEditExt
            sceneUpgradeMapper.transferSceneProEdit(sceneProEditV3.getId());
            sceneUpgradeMapper.transferSceneProEditExt(sceneProEditV3.getId());

            //同步到scenePro、sceneProExt
            sceneUpgradeMapper.transferScenePro(sceneProV3.getId());
            sceneUpgradeMapper.transferSceneProExt(sceneProV3.getId());

            //同步到scenePlus、scenePlus
            sceneUpgradeMapper.transferScenePlus(sceneProV3.getId());
            sceneUpgradeMapper.transferScenePlusExt(sceneProV3.getId());

            //todo 发送mq
            rabbitMqProducer.sendByWorkQueue(upgradeToV4, sceneProV3.getId());

            // TODO: 2022/4/22 写入进度条
            Map<String, Integer> progress = new HashMap<>();
            progress.put("status", 0);
            progress.put("progress", 0);
            redisUtil.set(RedisKey.scene_upgrade_progress_num, JSON.toJSONString(progress));

        }catch (Exception e){
            return ResultData.error(ErrorCode.FAILURE_CODE_7021);
        }finally {
            //解锁
            redisLockUtil.unlockLua(lockKey);
        }

        return ResultData.ok();
    }

    @Override
    public ResultData getUpgradeToV4Progress(String num) {

        String key = String.format(RedisKey.scene_upgrade_progress_num, num);
        String progress = redisUtil.get(key);
        SceneUpgradeProgressBean progressBean = null;
        if(StrUtil.isNotEmpty(progress)){
            progressBean = JSON.parseObject(progress, SceneUpgradeProgressBean.class);
        }else{
            progressBean = SceneUpgradeProgressBean.builder().num(num).status(0).progress(0).build();
        }
        return ResultData.ok(progressBean);
    }

}