郑永安
2023-06-19 2fcd97552d16718cc7997629fd637a73a5a4483f
src/main/java/com/gk/firework/Service/ServiceImpl/RegisterServiceImpl.java
对比新文件
@@ -0,0 +1,88 @@
package com.gk.firework.Service.ServiceImpl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gk.firework.Domain.EnterpriseResource;
import com.gk.firework.Domain.RegisterInfo;
import com.gk.firework.Domain.Utils.PageInfo;
import com.gk.firework.Domain.Utils.StringUtils;
import com.gk.firework.Domain.Vo.RegisterVo;
import com.gk.firework.Mapper.RegisterInfoMapper;
import com.gk.firework.Service.EnterpriseResourceService;
import com.gk.firework.Service.RegisterService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.ArrayList;
import java.util.List;
/**
 * @author : jingjy
 * @date : 2021/6/24 16:55
 */
@Service("RegisterService")
public class RegisterServiceImpl extends ServiceImpl<RegisterInfoMapper, RegisterInfo>  implements RegisterService {
    @Autowired
    private RegisterInfoMapper registerInfoMapper;
    @Autowired
    private EnterpriseResourceService resourceService;
    @Override
    public List<RegisterInfo> needSendSmsList() {
        Byte zero = 0;
        Byte one = 1;
        Byte two = 2;
        LambdaQueryWrapper<RegisterInfo> wrapper = new LambdaQueryWrapper<>();
        wrapper.eq(RegisterInfo::getIssms,zero);
        wrapper.in(RegisterInfo::getReviewresult,one,two);
        wrapper.eq(RegisterInfo::getIsdel,zero);
        return registerInfoMapper.selectList(wrapper);
    }
    @Override
    public void selectDataGrid(PageInfo pageInfo) {
        Page<RegisterVo> page = new Page<>(pageInfo.getPageIndex(), pageInfo.getPageSize());
        List<OrderItem> orderItems = new ArrayList<>();
        OrderItem orderItem = new OrderItem();
        if (StringUtils.isNotBlank(pageInfo.getSort()) && StringUtils.isNotBlank(pageInfo.getOrder())) {
            orderItem.setAsc(pageInfo.getOrder().equalsIgnoreCase("ascending"));
            orderItem.setColumn(pageInfo.getSort());
        }else {
            orderItem.setAsc(false);
            orderItem.setColumn("createdat");
        }
        orderItems.add(orderItem);
        page.setOrders(orderItems);
        if (StringUtils.isBlank(pageInfo.getSort())){
            pageInfo.setSort("createdat");
        }
        if (StringUtils.isBlank(pageInfo.getOrder())){
            pageInfo.setOrder("desc");
        }
        List<RegisterVo> registerVos = registerInfoMapper.selectDataGrid(pageInfo.getCondition(),page);
        List<RegisterVo> registerVoList = new ArrayList<>();
        for (RegisterVo registerVo : registerVos){
            List<String> business =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.BUSINESS);
            List<String> idCard =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.ID_CARD);
            List<String> legalPerson =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.LEGAL_PERSON);
            List<String> operation =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.OPERATION);
            List<String> escortCompany =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.ESCORT_COMPANY);
            List<String> insurance =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.INSURANCE);
            List<String> production =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.PRODUCTION);
            registerVo.setBusinessLicenseUrl(business);
            registerVo.setIdCardUrl(idCard);
            registerVo.setLegalPersonUrl(legalPerson);
            registerVo.setOperationQualificationUrl(operation);
            registerVo.setEscortCompanyUrl(escortCompany);
            registerVo.setInsuranceUrl(insurance);
            registerVo.setSafetyProductionUrl(production);
            registerVoList.add(registerVo);
        }
        pageInfo.setResult(registerVoList);
        pageInfo.setTotalCount(page.getTotal());
    }
}