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 implements RegisterService { @Autowired private RegisterInfoMapper registerInfoMapper; @Autowired private EnterpriseResourceService resourceService; @Override public List needSendSmsList() { Byte zero = 0; Byte one = 1; Byte two = 2; LambdaQueryWrapper 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 page = new Page<>(pageInfo.getPageIndex(), pageInfo.getPageSize()); List 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 registerVos = registerInfoMapper.selectDataGrid(pageInfo.getCondition(),page); List registerVoList = new ArrayList<>(); for (RegisterVo registerVo : registerVos){ List business =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.BUSINESS); List idCard =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.ID_CARD); List legalPerson =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.LEGAL_PERSON); List operation =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.OPERATION); List escortCompany =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.ESCORT_COMPANY); List insurance =resourceService.getUrlByBelongIdAndType(registerVo.getId(),RegisterVo.INSURANCE); List 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()); } }