From 698ae84adaf1b8d0e8dd61d7279863fe17c1e81d Mon Sep 17 00:00:00 2001 From: luodangjia <luodangjia> Date: 星期四, 23 一月 2025 14:20:39 +0800 Subject: [PATCH] 1 --- ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java | 249 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 249 insertions(+), 0 deletions(-) diff --git a/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java b/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java index 8addf56..6b4da84 100644 --- a/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java +++ b/ruoyi-modules/ruoyi-company/src/main/java/com/ruoyi/company/service/impl/CompanyServiceImpl.java @@ -1,11 +1,260 @@ package com.ruoyi.company.service.impl; +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.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; +import org.springframework.transaction.annotation.Transactional; + +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; @Service +@RequiredArgsConstructor(onConstructor_ = {@Lazy}) 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() + .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 MgtCompanyDetailVO queryCompanyDetail(Long id) { + Company company = this.getById(id); + MgtCompanyDetailVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyDetailVO.class); + if (Objects.isNull(mgtCompanyVO)) { + mgtCompanyVO = new MgtCompanyDetailVO(); + return mgtCompanyVO; + } + User user = userService.getById(company.getUserId()); + if (Objects.nonNull(user)) { + mgtCompanyVO.setPhone(user.getPhone()); + mgtCompanyVO.setAccountName(user.getAccountName()); + } + return mgtCompanyVO; + } + + /** + * 新增企业 + * + * @param dto + * @return + */ + @Transactional(rollbackFor = Exception.class) + @Override + public void saveCompany(MgtCompanyDTO dto) { + Long accountCount = userService.lambdaQuery().eq(User::getAccountName, dto.getAccountName()).count(); + if (accountCount > 0) { + throw new ServiceException("账户名重复"); + } + //添加用户信息 + User user = BeanUtils.copyBean(dto, User.class); + user.setPassword(SecurityUtils.encryptPassword(dto.getPassword())); + user.setCreateBy(SecurityUtils.getUserId()); + userService.save(user); + + Company company = BeanUtils.copyBean(dto, Company.class); + company.setUserId(user.getUserId()); + //根据公司名称查询数据库 + Long count = this.lambdaQuery().eq(Company::getCompanyName, company.getCompanyName()).eq(Company::getSocialCode, dto.getSocialCode()).count(); + if (count > 0) { + throw new ServiceException("该企业已存在"); + } + //身份证二要素校验 + identityVerification(dto, user); + this.save(company); + } + + /** + * 编辑企业 + * + * @param dto + */ + @Override + public void editCompany(MgtCompanyDTO dto) { + if (Objects.isNull(dto.getId())) { + throw new ServiceException("企业id不能为空"); + } + //查询企业 + Company company = this.getById(dto.getId()); + if (Objects.isNull(company)) { + throw new ServiceException("该企业不存在"); + } + //查询企业账号 + User user = userService.getById(company.getUserId()); + if (Objects.isNull(user)) { + throw new ServiceException("该企业账号不存在"); + } + 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); + if (StringUtils.isNotBlank(dto.getPassword())) { + userUpd.setPassword(SecurityUtils.encryptPassword(dto.getPassword())); + } + userUpd.setUserId(user.getUserId()); + userUpd.setUpdateBy(SecurityUtils.getUserId()); + userService.updateById(userUpd); + //根据公司名称查询数据库 + 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 + public void deleteCompany(Long id) { + //查询企业 + Company company = this.getById(id); + if (Objects.isNull(company)) { + throw new ServiceException("删除失败,该企业不存在"); + } + //删除企业账号 + 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() + .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"); + String fileName = URLEncoder.encode("企业信息导出数据", "UTF-8").replaceAll("\\+", "%20"); + response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx"); + + FastExcel.write(response.getOutputStream(), MgtCompanyExcel.class) + .sheet("企业信息导出数据") + .doWrite(BeanUtils.copyList(list, MgtCompanyExcel.class)); + } } -- Gitblit v1.7.1