package com.ruoyi.system.web;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
import com.ruoyi.common.config.RuoYiConfig;
|
import com.ruoyi.common.core.controller.BaseController;
|
import com.ruoyi.common.core.domain.AjaxResult;
|
import com.ruoyi.common.utils.StringUtils;
|
import com.ruoyi.common.utils.file.FileUploadUtils;
|
import com.ruoyi.common.utils.poi.ExcelUtil;
|
import com.ruoyi.system.domain.company.AreaInfo;
|
import com.ruoyi.system.domain.company.CompanyInfo;
|
import com.ruoyi.system.domain.dto.CompanyInfoDto;
|
import com.ruoyi.system.domain.vo.CompanyInfoVO;
|
import com.ruoyi.system.service.company.IAreaInfoService;
|
import com.ruoyi.system.service.company.ICompanyInfoService;
|
import com.ruoyi.system.utils.MyServerConfig;
|
import org.apache.commons.lang3.BooleanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.util.*;
|
import java.util.stream.Collectors;
|
|
@RestController
|
public class CompanyInfoController extends BaseController {
|
|
@Autowired
|
private ICompanyInfoService companyInfoService;
|
|
@Autowired
|
private MyServerConfig serverConfig;
|
|
@Autowired
|
private IAreaInfoService areaInfoService;
|
|
|
@RequestMapping("/companyInfo/svarInfo")
|
public AjaxResult savaInfo(CompanyInfoVO companyInfo, MultipartFile companyImageFile,
|
MultipartFile businessLicenseFile, MultipartFile securityCertificateFile) {
|
if (!checkIsExit(Collections.singletonList(companyInfo.getCertificateNum()), null)) {
|
return AjaxResult.error("数据已存在，请检查证书编号是否填写正确！");
|
}
|
initFileInfo(companyInfo, companyImageFile, businessLicenseFile, securityCertificateFile);
|
companyInfo.setCreateTime(new Date());
|
companyInfo.setCreateBy(getUsername());
|
companyInfoService.save(companyInfo);
|
return AjaxResult.success("添加成功！");
|
}
|
|
@RequestMapping("/companyInfo/updateInfo")
|
public AjaxResult updateInfo(CompanyInfoVO companyInfo, MultipartFile companyImageFile,
|
MultipartFile businessLicenseFile, MultipartFile securityCertificateFile) {
|
|
if (!checkIsExit(Collections.singletonList(companyInfo.getCertificateNum()), Collections.singletonList(companyInfo.getId()))) {
|
return AjaxResult.error("数据已存在，请检查证书编号是否填写正确！");
|
}
|
initFileInfo(companyInfo, companyImageFile, businessLicenseFile, securityCertificateFile);
|
companyInfo.setUpdateTime(new Date());
|
companyInfo.setUpdateBy(getUsername());
|
companyInfoService.updateById(companyInfo);
|
return AjaxResult.success("修改成功！");
|
}
|
|
@RequestMapping("/companyInfo/deleteById")
|
public AjaxResult deleteById(@RequestBody CompanyInfo companyInfo) {
|
companyInfo.setId(Long.valueOf(companyInfo.getIdStr()));
|
companyInfo.setIsDelete(1);
|
companyInfo.setUpdateTime(new Date());
|
companyInfo.setUpdateBy(getUsername());
|
companyInfoService.updateById(companyInfo);
|
return AjaxResult.success("删除成功！");
|
}
|
|
|
@RequestMapping("/companyInfo/findCompanyInfo")
|
public AjaxResult findCompanyInfo(@RequestBody CompanyInfoVO companyInfoVO) {
|
return companyInfoService.findCompanyInfoPage(companyInfoVO);
|
}
|
|
@RequestMapping("/companyInfo/getOne")
|
public AjaxResult getOne(@RequestBody CompanyInfoVO companyInfoVO) {
|
if (companyInfoVO.getId() != null) {
|
CompanyInfo companyInfo = companyInfoService.getById(companyInfoVO.getId());
|
if(companyInfo!=null){
|
List<CompanyInfoDto> companyInfoDtoList = companyInfoService.initCompanyCity(Collections.singletonList(companyInfo));
|
return AjaxResult.success(companyInfoDtoList.get(0));
|
}else{
|
return AjaxResult.error("数据不存在!");
|
}
|
|
} else {
|
QueryWrapper queryWrapper = new QueryWrapper();
|
queryWrapper.eq(CompanyInfo.IS_DELETE, 0);
|
queryWrapper.orderByDesc(CompanyInfo.CREATE_TIME);
|
queryWrapper.last("limit 1");
|
return AjaxResult.success(companyInfoService.getOne(queryWrapper));
|
}
|
}
|
|
@PostMapping("/companyInfo/export")
|
public void export(HttpServletResponse response, @RequestBody CompanyInfoVO companyInfoVO) {
|
List<CompanyInfoDto> list = companyInfoService.findExportData(companyInfoVO);
|
ExcelUtil<CompanyInfoDto> util = new ExcelUtil<>(CompanyInfoDto.class);
|
util.exportExcel(response, list, "企业数据");
|
}
|
|
@PostMapping("/companyInfo/downLoadTemplate")
|
public void downLoadTemplate(HttpServletResponse response, @RequestBody CompanyInfoVO companyInfoVO) {
|
List<CompanyInfoDto> list = new ArrayList<>();
|
ExcelUtil<CompanyInfoDto> util = new ExcelUtil<>(CompanyInfoDto.class);
|
util.exportExcel(response, list, "导入模板");
|
}
|
|
@RequestMapping("/companyInfo/importData")
|
public AjaxResult importData(MultipartFile file) {
|
List<AreaInfo> areaInfoList = areaInfoService.list();
|
Map<String,List<AreaInfo>> areaMap = areaInfoList.stream().collect(Collectors.groupingBy(o->o.getCity()+o.getArea()));
|
ExcelUtil<CompanyInfo> util = new ExcelUtil<CompanyInfo>(CompanyInfo.class);
|
List<CompanyInfo> companyInfos = null;
|
List<String> importFlaseData = new ArrayList<>();
|
try {
|
Date time = new Date();
|
companyInfos = util.importExcel(file.getInputStream());
|
final Integer[] companyCityIsEmpty = {0};
|
companyInfos.forEach(companyInfo -> {
|
companyInfo.setCreateTime(time);
|
companyInfo.setCreateBy(getUsername());
|
if(CollectionUtils.isNotEmpty(areaMap.get(companyInfo.getCity()+companyInfo.getArea()))){
|
companyInfo.setCompanyCity(areaMap.get(companyInfo.getCity()+companyInfo.getArea()).get(0).getId());
|
}else{
|
companyCityIsEmpty[0] = 1;
|
}
|
});
|
if(companyCityIsEmpty[0] == 1){
|
return AjaxResult.error("导入失败！所在地州市有误，请检查修改后重试！");
|
}
|
//获取证书编号集合
|
List<String> companyNumList = companyInfos.stream()
|
.filter(companyInfo -> StringUtils.isNotEmpty(companyInfo.getCertificateNum()))
|
.map(CompanyInfo::getCertificateNum).collect(Collectors.toList());
|
//去重判断是否重复
|
Set<String> companySet = companyNumList.stream().collect(Collectors.toSet());
|
|
if (companySet.size() != companyNumList.size() || !checkIsExit(companyNumList, null)) {
|
return AjaxResult.error("导入失败！数据重复，请检查导入数据！");
|
}
|
|
//过滤出不完整的数据
|
Iterator<CompanyInfo> iterator = companyInfos.iterator();
|
while (iterator.hasNext()) {
|
CompanyInfo companyInfo = iterator.next();
|
if (StringUtils.isEmpty(companyInfo.getCompanyName()) || StringUtils.isEmpty(companyInfo.getCompanyDirector())
|
|| StringUtils.isEmpty(companyInfo.getCompanyDirector()) || StringUtils.isEmpty(companyInfo.getCertificateNum())
|
|| StringUtils.isEmpty(companyInfo.getCompanyCity())) {
|
importFlaseData.add(companyInfo.getCompanyName());
|
iterator.remove();
|
}
|
}
|
|
if (StringUtils.isNotEmpty(companyInfos)) {
|
companyInfoService.saveBatch(companyInfos);
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
return AjaxResult.error("导入失败，请检查导入文件格式是否正确");
|
}
|
List<String> successIds = companyInfos.stream()
|
.map(CompanyInfo::getIdStr).collect(Collectors.toList());
|
if (CollectionUtils.isEmpty(importFlaseData)) {
|
return AjaxResult.success("导入成功，成功插入" + companyInfos.size() + "条数据", successIds);
|
} else {
|
return AjaxResult.success("导入成功，成功插入" + companyInfos.size() + "条数据," + importFlaseData + "数据不完整，请修改后重新导入！", successIds);
|
}
|
}
|
|
public boolean checkIsExit(List<String> companyNum, List<Long> idList) {
|
if (CollectionUtils.isNotEmpty(companyNum)) {
|
QueryWrapper queryWrapper = new QueryWrapper();
|
queryWrapper.in(CompanyInfo.CERTIFICATE_NUM, companyNum);
|
queryWrapper.eq(CompanyInfo.IS_DELETE, 0);
|
//如果idList不为空，则是修改方法，判断重复去除本事数据
|
if (CollectionUtils.isNotEmpty(idList)) {
|
queryWrapper.notIn(CompanyInfo.ID, idList);
|
}
|
List<CompanyInfo> companyInfoList = companyInfoService.list(queryWrapper);
|
if (CollectionUtils.isNotEmpty(companyInfoList)) {
|
return false;
|
}
|
}
|
return true;
|
}
|
|
public void initFileInfo(CompanyInfoVO companyInfo, MultipartFile companyImage,
|
MultipartFile businessLicense, MultipartFile securityCertificate) {
|
|
if (BooleanUtils.isTrue(companyInfo.getDeleteCompanyImage())) {
|
companyInfo.setCompanyImage("");
|
}
|
if (BooleanUtils.isTrue(companyInfo.getDeleteBusinessLicense())) {
|
companyInfo.setBusinessLicense("");
|
}
|
if (BooleanUtils.isTrue(companyInfo.getDeleteSecurityCertificate())) {
|
companyInfo.setSecurityCertificate("");
|
}
|
|
if (companyImage != null) {
|
// 上传文件路径
|
String filePath = RuoYiConfig.getUploadPath();
|
// 上传并返回新文件名称
|
String fileName = null;
|
try {
|
fileName = FileUploadUtils.upload(filePath, companyImage);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
companyInfo.setCompanyImage(fileName);
|
}
|
|
if (businessLicense != null) {
|
// 上传文件路径
|
String filePath = RuoYiConfig.getUploadPath();
|
// 上传并返回新文件名称
|
String fileName = null;
|
try {
|
fileName = FileUploadUtils.upload(filePath, businessLicense);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
companyInfo.setBusinessLicense(fileName);
|
}
|
|
if (securityCertificate != null) {
|
// 上传文件路径
|
String filePath = RuoYiConfig.getUploadPath();
|
// 上传并返回新文件名称
|
String fileName = null;
|
try {
|
fileName = FileUploadUtils.upload(filePath, securityCertificate);
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
companyInfo.setSecurityCertificate(fileName);
|
}
|
}
|
}
|
