4dkankan-fusion/src/main/java/com/fdkankan/fusion/service/impl/CaseNumServiceImpl.java

package com.fdkankan.fusion.service.impl;

import cn.hutool.core.io.FileUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.fdkankan.fusion.common.FilePath;
import com.fdkankan.fusion.common.util.OBJToGLBUtil;
import com.fdkankan.fusion.common.util.UploadToOssUtil;
import com.fdkankan.fusion.entity.CaseNumEntity;
import com.fdkankan.fusion.entity.Model;
import com.fdkankan.fusion.mapper.ICaseNumMapper;
import com.fdkankan.fusion.request.SceneNumParam;
import com.fdkankan.fusion.service.ICaseNumService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.fdkankan.fusion.service.IModelService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author 
 * @since 2022-07-27
 */
@Service
public class CaseNumServiceImpl extends ServiceImpl<ICaseNumMapper, CaseNumEntity> implements ICaseNumService {

    @Autowired
    UploadToOssUtil uploadToOssUtil;
    @Value("${local.obj_path}")
    private String OBJ_PATH;
    @Value("${local.glb_path}")
    private String GLB_PATH;
    @Value("${upload.query-path}")
    private String queryPath;
    @Autowired
    IModelService modelService;

    @Override
    public List<CaseNumEntity> getByCaseId(Integer caseId) {
        LambdaQueryWrapper<CaseNumEntity> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(CaseNumEntity::getCaseId,caseId);
        return this.list(wrapper);
    }


    @Override
    public void addBatch(Integer caseId, List<SceneNumParam> sceneNumParam) {

        this.deleteByCaseId(caseId);

        List<CaseNumEntity> newCaseNums = new ArrayList<>();
        List<Model> modelList = new ArrayList<>();
        for (SceneNumParam param : sceneNumParam) {
            List<String> numList = param.getNumList();
            HashSet<String> setNum = new HashSet<>(numList);
            for (String num : setNum) {
                CaseNumEntity caseNumEntity = new CaseNumEntity();
                caseNumEntity.setCaseId(caseId);
                caseNumEntity.setNumType(param.getType());
                caseNumEntity.setNum(num);
                newCaseNums.add(caseNumEntity);
                if(param.getType() == 3){                   //用户上传三维模型跳过
                    continue;
                }
                Model model = modelService.getIsNullNewByNum(num);
                model.setModelDateType("obj");
                model.setType(param.getType());
                model.setModelType("pointcloud");    //深时点云类型
                if(param.getType() != 2){           //不为深时
                    model.setModelObjUrl(String.format(OBJ_PATH ,num) +"/mesh.obj");
                    model.setModelGlbUrl(getGlbUrl(param.getType(),num));
                    model.setModelType("glb");
                }
                model.setNum(num);
                model.setCreateStatus(1);
                modelList.add(model);
            }
        }

        if(newCaseNums.size() >0){
            this.saveBatch(newCaseNums);
        }
        if(modelList.size() >0){
            modelService.saveOrUpdateBatch(modelList);
        }
    }

    private String getGlbUrl(Integer type, String num) {
        if(type == 0 || type == 1){ //看看，看见
            String objPath = String.format(OBJ_PATH ,num);
            String glbPath = String.format(GLB_PATH,num) +"/mesh.glb";
            String glbOssPath = String.format(FilePath.GLB_OSS_PATH , num);
            if(uploadToOssUtil.existKey(glbOssPath)){
                return queryPath + "/"+glbOssPath;
            }
            List<String> fileList = uploadToOssUtil.listKeysFromAli(String.format(FilePath.OBJ_OSS_PATH, num));
            for (String fileName : fileList) {
                File file = new File(objPath);
                if(!file.exists()){
                    file.mkdirs();
                }
                String[] split = fileName.split("/");
                String localPath = objPath +"/"+split[split.length-1];
                uploadToOssUtil.downFormAli(fileName,localPath);
            }
            OBJToGLBUtil.objToGlb(objPath,glbPath );
            uploadToOssUtil.upload(glbPath,glbOssPath);
            FileUtil.del(objPath);
            return queryPath + "/"+glbOssPath;
        }
        return null;
    }

    @Override
    public void deleteByCaseId(Integer caseId) {
        LambdaQueryWrapper<CaseNumEntity> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(CaseNumEntity::getCaseId,caseId);
        List<CaseNumEntity> list = this.list(wrapper);
        if(list.size() >0){
            List<String> numList = list.parallelStream().map(CaseNumEntity::getNum).collect(Collectors.toList());
            modelService.deleteByNum(numList);
            this.remove(wrapper);
        }
    }

    @Override
    public HashMap<Integer, List<String>> getTypeMap(Integer caseId) {
        List<CaseNumEntity> caseNumList = this.getByCaseId(caseId);
        HashMap<Integer,List<String>> typeMap = new HashMap<>();
        for (CaseNumEntity caseNumEntity : caseNumList) {
            List<String> numList ;
            if(typeMap.get(caseNumEntity.getNumType()) == null){
                numList = new ArrayList<>();
            }else {
                numList = typeMap.get(caseNumEntity.getNumType());
            }
            numList.add(caseNumEntity.getNum());
            typeMap.put(caseNumEntity.getNumType(),numList);
        }
        return typeMap;
    }

    @Override
    public List<CaseNumEntity> getByNum(String  num) {
        LambdaQueryWrapper<CaseNumEntity> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(CaseNumEntity::getNum,num);
        return this.list(wrapper);
    }
}