4dkankan-center-user/src/main/java/com/fdkankan/ucenter/service/impl/ScenePlusServiceImpl.java

package com.fdkankan.ucenter.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.thread.ThreadUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.fdkankan.common.constant.ConstantFilePath;
import com.fdkankan.common.constant.UploadFilePath;
import com.fdkankan.common.util.DateExtUtil;
import com.fdkankan.common.util.FileUtils;
import com.fdkankan.common.util.MatrixToImageWriterUtil;
import com.fdkankan.fyun.oss.UploadToOssUtil;
import com.fdkankan.redis.constant.RedisKey;
import com.fdkankan.redis.util.RedisUtil;
import com.fdkankan.ucenter.bean.SceneJsonBean;
import com.fdkankan.ucenter.entity.*;
import com.fdkankan.ucenter.mapper.IScenePlusMapper;
import com.fdkankan.ucenter.service.*;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.ucenter.vo.response.SceneEditControlsVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Service;

import java.io.File;
import java.io.IOException;
import java.time.format.DateTimeFormatter;
import java.util.*;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.ExecutorService;

/**
 * <p>
 * 场景主表 服务实现类
 * </p>
 *
 * @author 
 * @since 2022-07-04
 */
@Slf4j
@Service
public class ScenePlusServiceImpl extends ServiceImpl<IScenePlusMapper, ScenePlus> implements IScenePlusService {

    @Autowired
    private IScenePlusExtService scenePlusExtService;
    @Autowired
    private ISceneEditInfoService sceneEditInfoService;
    @Autowired
    private ISceneEditInfoExtService sceneEditInfoExtService;
    @Autowired
    private ISceneEditControlsService sceneEditControlsService;
    @Autowired
    private ICameraDetailService cameraDetailService;
    @Autowired
    private ISceneCopyLogService sceneCopyLogService;
    @Autowired
    UploadToOssUtil uploadToOssUtil;
    @Autowired
    ISceneProService sceneProService;
    @Autowired
    RedisUtil redisUtil;
    @Autowired
    private ISceneProEditService sceneProEditService;

    @Override
    public Long getCountByUserId(Long userId, List<Integer> sceneSourceList) {
        LambdaQueryWrapper<ScenePlus> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePlus::getUserId, userId);
        wrapper.in(ScenePlus::getSceneSource, sceneSourceList);
        return this.count(wrapper);
    }

    @Override
    public Long getCountByUserId(Long userId, Integer cameraType) {
        return this.getBaseMapper().getCountByUserId(userId,cameraType);
    }


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

    @Override
    public void bindOrUnCamera(List<Long> cameraIds, Long userId) {
        LambdaUpdateWrapper<ScenePlus> wrapper = new LambdaUpdateWrapper<>();
        wrapper.set(ScenePlus::getUserId,userId)
                .in(ScenePlus::getCameraId,cameraIds);
        this.update(wrapper);
    }

    @Override
    public List<ScenePlus> getListByCameraIds(List<Long> cameraIds) {
        LambdaQueryWrapper<ScenePlus> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePlus::getCameraId,cameraIds);
        return this.list(wrapper);
    }

    @Override
    public List<ScenePlus> getListByNums(List<String> numList) {
        LambdaQueryWrapper<ScenePlus> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePlus::getNum,numList);
        return this.list(wrapper);
    }

    @Override
    public HashMap<Long, ScenePlus> getMapByIds(List<Long> sceneIds) {
        LambdaQueryWrapper<ScenePlus> wrapper = new LambdaQueryWrapper<>();
        wrapper.in(ScenePlus::getId,sceneIds);
        List<ScenePlus> list = this.list(wrapper);
        HashMap<Long,ScenePlus> map = new HashMap<>();
        list.forEach(entity -> map.put(entity.getId(),entity));
        return map;
    }

    @Override
    public ScenePlus getByNum(String sceneNum) {
        LambdaQueryWrapper<ScenePlus> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(ScenePlus::getNum,sceneNum);
        List<ScenePlus> list = this.list(wrapper);
        if(list!=null && list.size() >0){
            return list.get(0);
        }
        return null;
    }

    @Override
    public Long copyV4Scene(ScenePlus scenePlus,String newNum ,CameraDetail cameraDetail,String time) throws Exception {
        String num = scenePlus.getNum();
        Long plusId = scenePlus.getId();
        ScenePlusExt plusExt = scenePlusExtService.getByPlusId(plusId);
        String bucket = plusExt.getYunFileBucket();
        String dataSource = plusExt.getDataSource();
        String oldUuid = dataSource.substring(dataSource.lastIndexOf("/") + 1);
        Thread.sleep(RandomUtil.randomInt(10, 100));
        String newUuid = oldUuid.split("_")[0].concat("_").concat(DateUtil.format(Calendar.getInstance().getTime(), "yyyyMMddHHmmssSSS"));
        String newDataSource = plusExt.getDataSource().replace(oldUuid, newUuid);

        ScenePro scenePro = sceneProService.getById(plusId);
        scenePro.setId(null);
        scenePro.setDataSource(newDataSource);
        scenePro.setNum(newNum);
        scenePro.setWebSite(scenePro.getWebSite().replace(num, newNum));
        scenePro.setThumb(scenePro.getThumb().replace(num, newNum));
        if(StrUtil.isNotEmpty(scenePro.getVideos())){
            scenePro.setVideos(scenePro.getVideos().replaceAll(num, newNum));
        }
        scenePro.setSceneName(scenePlus.getTitle());
        sceneProService.save(scenePro);

        SceneProEdit sceneProEdit = sceneProEditService.getByProId(plusId);
        sceneProEdit.setProId(scenePro.getId());
        sceneProEdit.setId(null);
        sceneProEditService.save(sceneProEdit);

        // 拷贝数据
        scenePlus.setNum(newNum);
        scenePlus.setId(scenePro.getId());
        scenePlus.setSceneStatus(0);
        this.save(scenePlus);

        ExecutorService executor = ThreadUtil.newSingleExecutor();
        try {
            CompletableFuture.runAsync(() -> {
                try {
                    // 拷贝场景编辑资源
                    String oldEditPath = String.format(UploadFilePath.EDIT_PATH, num);
                    String newEditPath = String.format(UploadFilePath.EDIT_PATH, newNum);
                    uploadToOssUtil.copyFiles(bucket, oldEditPath, newEditPath);

                    // 拷贝场景展示资源
                    String oldViewPath = String.format(UploadFilePath.VIEW_PATH, num);
                    String newViewPath = String.format(UploadFilePath.VIEW_PATH, newNum);
                    uploadToOssUtil.copyFiles(bucket, oldViewPath, newViewPath);

                    // 拷贝本地资源
                    String oldPath = String.format("/mnt/4Dkankan/scene/%s/caches/images", num);
                    String newPath = String.format("/mnt/4Dkankan/scene/%s/caches/images", newNum);
                    if(new File(oldPath).exists()){
                        FileUtils.copyDirectiory(oldPath, newPath);
                    }

                    String scenePath = ConstantFilePath.SCENE_V4_PATH + num;
                    File file = new File(scenePath);
                    if(file.exists()){
                        String newScenePath = ConstantFilePath.SCENE_V4_PATH + newNum;
                        FileUtils.copyDirectiory(scenePath, newScenePath);
                    }
                    String newVideos = plusExt.getVideos();
                    if(StrUtil.isNotEmpty(newVideos)){
                        newVideos = plusExt.getVideos().replaceAll("/data/data" + num, "/scene_view_data/" + newNum + "/data").replaceAll(num, newNum);
                    }

                    plusExt.setId(null);
                    plusExt.setDataSource(newDataSource);
                    plusExt.setPlusId(scenePlus.getId());
                    plusExt.setWebSite(plusExt.getWebSite().replace(num, newNum));
                    plusExt.setThumb(plusExt.getThumb().replace(num, newNum));
                    plusExt.setVideos(newVideos);
                    plusExt.setViewCount(0);
                    scenePlusExtService.save(plusExt);

                    SceneEditInfo sceneEditInfo = sceneEditInfoService.getByScenePlusId(plusId);
                    Long sceneEditInfoId = sceneEditInfo.getId();

                    sceneEditInfo.setId(null);
                    sceneEditInfo.setScenePlusId(scenePlus.getId());
                    sceneEditInfo.setSceneProId(scenePlus.getId());
                    sceneEditInfo.setTitle(scenePlus.getTitle());
                    sceneEditInfoService.save(sceneEditInfo);

                    SceneEditInfoExt sceneEditInfoExt = sceneEditInfoExtService.getByEditInfoId(sceneEditInfoId);
                    sceneEditInfoExt.setId(null);
                    sceneEditInfoExt.setEditInfoId(sceneEditInfo.getId());
                    sceneEditInfoExt.setScenePlusId(scenePlus.getId());
                    sceneEditInfoExt.setSceneProId(scenePlus.getId());
                    sceneEditInfoExtService.save(sceneEditInfoExt);

                    SceneEditControls sceneEditControls = sceneEditControlsService.getBySceneEditId(sceneEditInfoId);
                    sceneEditControls.setId(null);
                    sceneEditControls.setEditInfoId(sceneEditInfo.getId());
                    sceneEditControlsService.save(sceneEditControls);

                    // 生成scene.json
                    SceneJsonBean sceneJson = new SceneJsonBean();
                    BeanUtil.copyProperties(sceneEditInfoExt, sceneJson);
                    BeanUtil.copyProperties(sceneEditInfo, sceneJson);
                    SceneEditControlsVO sceneEditControlsVO = BeanUtil.copyProperties(sceneEditControls, SceneEditControlsVO.class);
                    sceneJson.setControls(sceneEditControlsVO);
                    sceneJson.setNum(newNum);
                    sceneJson.setCreateTime(scenePlus.getCreateTime());
                    sceneJson.setSceneResolution(plusExt.getSceneResolution());
                    sceneJson.setSceneFrom(plusExt.getSceneFrom());
                    sceneJson.setSceneKind(plusExt.getSceneKind());
                    if(StrUtil.isNotEmpty(plusExt.getVideos())){
                        sceneJson.setVideos(plusExt.getVideos());
                    }

                    String sceneJsonLocalPath = ConstantFilePath.SCENE_PATH + "data" + File.separator + "data" + newNum + File.separator + "scene.json";
                    FileUtils.writeFile(sceneJsonLocalPath, JSON.toJSONString(sceneJson));

                    String sceneJsonPath = String.format(UploadFilePath.DATA_VIEW_PATH+"scene.json", newNum);
                    uploadToOssUtil.upload(bucket, JSON.toJSONBytes(sceneJson), sceneJsonPath);

                    //删除scenejson缓存
                    redisUtil.del(String.format(RedisKey.SCENE_JSON, num));

                    // 生成二维码
                    String outPathZh = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/" + newNum + ".png";
                    String outPathEn = ConstantFilePath.BASE_PATH + File.separator + "sceneQRcode/" + newNum + "_en.png";
                    MatrixToImageWriterUtil.createQRCode(plusExt.getWebSite(), outPathZh, false,null);
                    MatrixToImageWriterUtil.createQRCode(plusExt.getWebSite() + "&lang=en", outPathEn, false, null);
                    uploadToOssUtil.upload(bucket, outPathZh, String.format(UploadFilePath.DOWNLOADS_QRCODE, newNum) + newNum + ".png");
                    uploadToOssUtil.upload(bucket, outPathEn, String.format(UploadFilePath.DOWNLOADS_QRCODE, newNum) + newNum + "_en.png");
                    scenePlus.setSceneStatus(-2);
                    this.updateById(scenePlus);

                    cameraDetail.setUsedSpace(cameraDetail.getUsedSpace() + plusExt.getSpace());
                    cameraDetailService.updateById(cameraDetail);

                    sceneCopyLogService.saveByNum(num,newNum,scenePlus.getUserId());

                }catch (Exception e){
                    this.removeById(scenePlus.getId());
                    log.error("复制场景异常", e);
                }
            }, executor).whenComplete((reslut, e) -> {
                log.info("复制场景oldNum:{},newNum:{}结束-{}",num,newNum, new Date());
            });
        }catch (Exception e){
            log.error("线程错误：{}",e);
        }finally {
            executor.shutdown();
        }

        return scenePlus.getId();
    }
}