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.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.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.mapper.CompanyMapper;
|
import com.ruoyi.company.service.CompanyService;
|
import com.ruoyi.company.service.UserService;
|
import com.ruoyi.company.utils.AliyunCloudAuthUtil;
|
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.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;
|
/**
|
* 获取企业列表
|
*
|
* @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 MgtCompanyVO queryCompanyDetail(Long id) {
|
Company company = this.getById(id);
|
MgtCompanyVO mgtCompanyVO = BeanUtils.copyBean(company, MgtCompanyVO.class);
|
if (Objects.isNull(mgtCompanyVO)) {
|
mgtCompanyVO = new MgtCompanyVO();
|
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()).count();
|
if (count > 0) {
|
throw new ServiceException("该公司账号已存在");
|
}
|
//身份证二要素校验
|
Boolean res = aliyunCloudAuthUtil.verifyIdCard(dto.getLegalPersonName(), dto.getIdCardNumber());
|
if (!res) {
|
throw new ServiceException("身份证信息不匹配");
|
}
|
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())
|
.eq(Company::getCompanyName, company.getCompanyName())
|
.count();
|
if (count > 0) {
|
throw new ServiceException("该公司账号已存在");
|
}
|
}
|
//身份证二要素校验
|
Boolean res = aliyunCloudAuthUtil.verifyIdCard(dto.getLegalPersonName(), dto.getIdCardNumber());
|
if (!res) {
|
throw new ServiceException("身份证信息不匹配");
|
}
|
Company companyUpd = BeanUtils.copyBean(dto, Company.class);
|
companyUpd.setId(company.getId());
|
this.updateById(companyUpd);
|
}
|
|
/**
|
* 删除企业
|
*
|
* @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);
|
|
}
|
|
@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));
|
}
|
}
|