1

luodangjia

2025-01-23 698ae84adaf1b8d0e8dd61d7279863fe17c1e81d

 ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java

@@ -3,20 +3,25 @@
import cn.idev.excel.FastExcel;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.constant.CacheConstants;
import com.ruoyi.common.core.exception.ServiceException;
import com.ruoyi.common.core.page.BeanUtils;
import com.ruoyi.common.core.page.PageDTO;
import com.ruoyi.common.core.utils.StringUtils;
import com.ruoyi.common.redis.service.RedisService;
import com.ruoyi.common.security.utils.SecurityUtils;
import com.ruoyi.company.api.domain.Company;
import com.ruoyi.company.api.domain.User;
import com.ruoyi.company.api.domain.dto.MgtCompanyDTO;
import com.ruoyi.company.api.domain.excel.MgtCompanyExcel;
import com.ruoyi.company.api.domain.query.MgtCompanyQuery;
import com.ruoyi.company.api.domain.vo.MgtCompanyVO;
import com.ruoyi.company.domain.vo.MgtCompanyDetailVO;
import com.ruoyi.company.domain.vo.MgtCompanyVO;
import com.ruoyi.company.mapper.CompanyMapper;
import com.ruoyi.company.service.CompanyService;
import com.ruoyi.company.service.UserService;
import com.ruoyi.company.utils.AliyunCloudAuthUtil;
import com.ruoyi.system.api.model.AppUser;
import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
@@ -25,6 +30,7 @@
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.Collection;
import java.util.List;
import java.util.Objects;

@@ -33,35 +39,51 @@
public class CompanyServiceImpl extends ServiceImpl<CompanyMapper, Company> implements CompanyService {
    private final UserService userService;
    private final HttpServletResponse response;

    private final AliyunCloudAuthUtil aliyunCloudAuthUtil;

    private final RedisService redisService;
    /**
     * 获取企业列表
     *
     * @param query
     * @return
     */
    @Override
    public PageDTO<MgtCompanyVO> queryPage(MgtCompanyQuery query) {
        Page<Company> page = this.lambdaQuery()
                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getCompanyName, query.getCompanyInfo())
                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getSocialCode, query.getCompanyInfo())
                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getLegalPersonName, query.getLegalPersonInfo())
                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getIdCardNumber, query.getLegalPersonInfo())
                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactName, query.getContactInfo())
                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactPhone, query.getContactInfo())
                .and(StringUtils.isNotBlank(query.getCompanyInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getCompanyName, query.getCompanyInfo())
                                .or()
                                .like(Company::getSocialCode, query.getCompanyInfo()))
                .and(StringUtils.isNotBlank(query.getLegalPersonInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getLegalPersonName, query.getLegalPersonInfo())
                                .or()
                                .like(Company::getIdCardNumber, query.getLegalPersonInfo()))
                .and(StringUtils.isNotBlank(query.getContactInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getContactName, query.getContactInfo())
                                .or()
                                .like(Company::getContactPhone, query.getContactInfo()))
                .orderByDesc(Company::getCreateTime)
                .page(new Page<>(query.getPageCurr(), query.getPageSize()));
        return PageDTO.of(page,MgtCompanyVO.class);
    }

    /**
     * 获取企业详情
     *
     * @param id
     * @return
     */
    @Override
    public MgtCompanyVO queryCompanyDetail(Long id) {
    public MgtCompanyDetailVO queryCompanyDetail(Long id) {
        Company company = this.getById(id);
        MgtCompanyVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyVO.class);
        MgtCompanyDetailVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyDetailVO.class);
        if (Objects.isNull(mgtCompanyVO)) {
            mgtCompanyVO = new MgtCompanyVO();
            mgtCompanyVO = new MgtCompanyDetailVO();
            return mgtCompanyVO;
        }
        User user = userService.getById(company.getUserId());
@@ -74,6 +96,7 @@

    /**
     * 新增企业
     *
     * @param dto
     * @return
     */
@@ -87,25 +110,29 @@
        //添加用户信息
        User user = BeanUtils.copyBean(dto, User.class);
        user.setPassword(SecurityUtils.encryptPassword(dto.getPassword()));
        user.setCreateBy(SecurityUtils.getUserId());
        userService.save(user);
        // TODO 校验身份证信息

        Company company = BeanUtils.copyBean(dto, Company.class);
        company.setUserId(user.getUserId());
        //根据公司名称查询数据库
        Long count = this.lambdaQuery().eq(Company::getCompanyName, company.getCompanyName()).count();
        Long count = this.lambdaQuery().eq(Company::getCompanyName, company.getCompanyName()).eq(Company::getSocialCode, dto.getSocialCode()).count();
        if (count > 0) {
            throw new ServiceException("该公司账号已存在");
            throw new ServiceException("该企业已存在");
        }
        //身份证二要素校验
        identityVerification(dto, user);
        this.save(company);
    }

    /**
     * 编辑企业
     *
     * @param dto
     */
    @Override
    public void editCompany(MgtCompanyDTO dto) {
        if (Objects.isNull(dto.getId())){
        if (Objects.isNull(dto.getId())) {
            throw new ServiceException("企业id不能为空");
        }
        //查询企业
@@ -118,28 +145,57 @@
        if (Objects.isNull(user)) {
            throw new ServiceException("该企业账号不存在");
        }
        Long accountCount = userService.lambdaQuery().ne(User::getUserId, user.getUserId()).eq(User::getAccountName, dto.getAccountName()).count();
        Long accountCount = userService.lambdaQuery()
                .ne(User::getUserId, user.getUserId())
                .eq(User::getAccountName, dto.getAccountName())
                .count();
        if (accountCount > 0) {
            throw new ServiceException("账户名重复");
        }
        //用户信息
        User userUpd = BeanUtils.copyBean(dto, User.class);
        userUpd.setPassword(SecurityUtils.encryptPassword(dto.getPassword()));
        if (StringUtils.isNotBlank(dto.getPassword())) {
            userUpd.setPassword(SecurityUtils.encryptPassword(dto.getPassword()));
        }
        userUpd.setUserId(user.getUserId());
        userUpd.setUpdateBy(SecurityUtils.getUserId());
        userService.updateById(userUpd);
        //根据公司名称查询数据库
        Long count = this.lambdaQuery().ne(Company::getId,dto.getId()).eq(Company::getCompanyName, company.getCompanyName()).count();
        if (count > 0) {
            throw new ServiceException("该公司账号已存在");
        if (StringUtils.isNotBlank(company.getCompanyName())){
            Long count = this.lambdaQuery()
                    .ne(Company::getId, dto.getId())
                    .and(i -> {
                    i.eq(Company::getSocialCode, dto.getSocialCode());
                    i.or();
                    i.eq(Company::getCompanyName, dto.getCompanyName());
                    })
                    .count();
            if (count > 0) {
                throw new ServiceException("该企业已存在");
            }
        }
        identityVerification(dto, user);
        Company companyUpd = BeanUtils.copyBean(dto, Company.class);
        companyUpd.setId(company.getId());
        this.updateById(companyUpd);
    }

    private void identityVerification(MgtCompanyDTO dto, User user) {
        //身份证二要素校验
        Boolean res = aliyunCloudAuthUtil.verifyIdCard(dto.getLegalPersonName(), dto.getIdCardNumber());
        if (!res) {
            throw new ServiceException("身份证信息不匹配");
        }
        //营业执照二要素校验
        res = aliyunCloudAuthUtil.verifyBusinessLicense(dto.getSocialCode(), dto.getCompanyName(), user.getUserId());
        if (!res) {
            throw new ServiceException("营业执照信息不匹配");
        }
    }

    /**
     * 删除企业
     *
     * @param id
     */
    @Override
@@ -150,26 +206,50 @@
            throw new ServiceException("删除失败，该企业不存在");
        }
        //删除企业账号
         userService.removeById(company.getUserId());
        userService.removeById(company.getUserId());

        //删除企业信息
        this.removeById(id);

        //强退用户
        forceLogout(company.getUserId());

    }

    public void forceLogout(Long userId) {
        Collection<String> keys = redisService.keys(CacheConstants.LOGIN_TOKEN_KEY + "*");
        for (String key : keys) {
            Object user = redisService.getCacheObject(key);
            if (user instanceof AppUser) {
                AppUser appUser = (AppUser) user;
                if (appUser.getUserId().equals(userId)) {
                    redisService.deleteObject(key);
                }
            }
        }
    }

    @Override
    public void export(MgtCompanyQuery query) throws IOException {
        List<Company> list = this.lambdaQuery()
                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getCompanyName, query.getCompanyInfo())
                .like(StringUtils.isNotBlank(query.getCompanyInfo()), Company::getSocialCode, query.getCompanyInfo())
                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getLegalPersonName, query.getLegalPersonInfo())
                .like(StringUtils.isNotBlank(query.getLegalPersonInfo()), Company::getIdCardNumber, query.getLegalPersonInfo())
                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactName, query.getContactInfo())
                .like(StringUtils.isNotBlank(query.getContactInfo()), Company::getContactPhone, query.getContactInfo()).list();
                .and(StringUtils.isNotBlank(query.getCompanyInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getCompanyName, query.getCompanyInfo())
                                .or()
                                .like(Company::getSocialCode, query.getCompanyInfo()))
                .and(StringUtils.isNotBlank(query.getLegalPersonInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getLegalPersonName, query.getLegalPersonInfo())
                                .or()
                                .like(Company::getIdCardNumber, query.getLegalPersonInfo()))
                .and(StringUtils.isNotBlank(query.getContactInfo()),
                        queryWrapper -> queryWrapper
                                .like(Company::getContactName, query.getContactInfo())
                                .or()
                                .like(Company::getContactPhone, query.getContactInfo()))
                .orderByDesc(Company::getCreateTime).list();
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setCharacterEncoding

                ("utf-8");
        response.setCharacterEncoding("utf-8");
        String fileName = URLEncoder.encode("企业信息导出数据", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");