damiao/eladmin-system/src/main/java/me/zhengjie/dao/mybatis/FaceCompareRepository.java

package me.zhengjie.dao.mybatis;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import lombok.RequiredArgsConstructor;
import me.zhengjie.application.admin.controller.vo.FaceCompareRsp;
import me.zhengjie.base.util.BeanCopyUtils;
import me.zhengjie.base.util.FileUploadUtil;
import me.zhengjie.dao.mybatis.entity.ContractOrderEntity;
import me.zhengjie.dao.mybatis.entity.FaceCompareEntity;
import me.zhengjie.dao.mybatis.entity.FileInfoEntity;
import me.zhengjie.dao.mybatis.mapper.ContractOrderMapper;
import me.zhengjie.dao.mybatis.mapper.FaceCompareMapper;
import me.zhengjie.dao.mybatis.mapper.FileInfoMapper;
import org.apache.http.util.TextUtils;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

@RequiredArgsConstructor
@Component
public class FaceCompareRepository {
    private final FaceCompareMapper thirdPartyInterfaceMapper;
    private final FileInfoMapper fileInfoMapper;

    public void insert(FaceCompareEntity thirdPartyInterfaceEntity) {
        thirdPartyInterfaceEntity.setCreateTime(new Date());
        thirdPartyInterfaceMapper.insert(thirdPartyInterfaceEntity);
    }

    public List<FaceCompareRsp> page(IPage<FaceCompareEntity> page, QueryWrapper<FaceCompareEntity> qw, ContractOrderMapper contractOrderMapper) {
        IPage<FaceCompareEntity> thirdPartyInterfaceEntityIPage = thirdPartyInterfaceMapper.selectPage(page, qw);
        List<FaceCompareRsp> faceCompareRspList = new ArrayList<>();
        for (FaceCompareEntity thirdPartyInterfaceEntity : thirdPartyInterfaceEntityIPage.getRecords()) {
            FaceCompareRsp faceCompareRsp = BeanCopyUtils.convertObj(thirdPartyInterfaceEntity, FaceCompareRsp.class);
            String fileId = thirdPartyInterfaceEntity.getFileId();
            FileInfoEntity fileInfoEntity = fileInfoMapper.selectById(fileId);
            String previewUrl = null;
            try {
                previewUrl = getPreview(fileInfoEntity.getPath());
            } catch (Exception e) {
                throw new RuntimeException(e);
            }
            faceCompareRsp.setUrl(previewUrl);
            if (!TextUtils.isEmpty(faceCompareRsp.getConsultNo())) {
                QueryWrapper<ContractOrderEntity> orderQw = new QueryWrapper<>();
                orderQw.like("business_no", faceCompareRsp.getBusinessNo());
                ContractOrderEntity entity = contractOrderMapper.selectOne(orderQw);
                if (null != entity) {
                    faceCompareRsp.setOrderStatus(entity.getStatus());
                    faceCompareRsp.setAutoGenerate(entity.getAutoGenerate());
                    faceCompareRsp.setProdId(entity.getProdId());
                    faceCompareRsp.setContractType(entity.getContractType() + "");
                    faceCompareRsp.setNotarizationOrderId(entity.getId()+"");
                }
            }
            faceCompareRspList.add(faceCompareRsp);
        }
        return faceCompareRspList;
    }

    public String getPreview(String fileKey) {
        String url = FileUploadUtil.getPreviewUrl(fileKey);
        return url;
    }

    public FaceCompareEntity getOne(QueryWrapper<FaceCompareEntity> thirdPartyInterfaceEntityQueryWrapper) {
        return thirdPartyInterfaceMapper.selectOne(thirdPartyInterfaceEntityQueryWrapper);
    }

    public List<FaceCompareEntity> getList(QueryWrapper<FaceCompareEntity> thirdPartyInterfaceEntityQueryWrapper) {
        return thirdPartyInterfaceMapper.selectList(thirdPartyInterfaceEntityQueryWrapper);
    }
}