From 9236c9d1512b57f78755747bdabcd8adf4685ade Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@163.com> Date: 星期日, 18 九月 2022 20:41:50 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/manager/controller/CstmrController.java | 192 +++++++++++++++++++++++------------------------ 1 files changed, 95 insertions(+), 97 deletions(-) diff --git a/src/main/java/com/zy/crm/manager/controller/CstmrController.java b/src/main/java/com/zy/crm/manager/controller/CstmrController.java index b56f090..dd2ba66 100644 --- a/src/main/java/com/zy/crm/manager/controller/CstmrController.java +++ b/src/main/java/com/zy/crm/manager/controller/CstmrController.java @@ -19,6 +19,7 @@ import com.zy.crm.system.service.DicService; import lombok.extern.slf4j.Slf4j; import org.apache.poi.hssf.usermodel.HSSFWorkbook; +import org.apache.poi.ss.usermodel.DataFormatter; import org.apache.poi.ss.usermodel.Row; import org.apache.poi.ss.usermodel.Sheet; import org.apache.poi.ss.usermodel.Workbook; @@ -182,104 +183,101 @@ @ManagerAuth(memo = "鐢叉柟鍗曚綅Excel瀵煎叆") @Transactional public R cstmrExcelImport(MultipartFile file) throws IOException { - try (InputStream inStream = file.getInputStream()) { - String fileMime = file.getContentType(); - int excelVersion = 2007; - if ("application/vnd.ms-excel".equals(fileMime)) { - excelVersion = 2003; - } - Workbook book = null; - if (excelVersion == 2003) { - book = new HSSFWorkbook(inStream); - } - else { // 褰� excel 鏄� 2007 鏃� - book = new XSSFWorkbook(inStream); - } - Sheet sheet = book.getSheetAt(0); - int totalRows = sheet.getPhysicalNumberOfRows() - 1; // 鎬� - Long userId = getUserId(); - Long hostId = getHostId(); - Long deptId = getDeptId(); - Date now = new Date(); - for (int i = 2; i < totalRows; i++) { - Cstmr cstmr = new Cstmr(); - Row row = sheet.getRow(i); - // 瀹㈡埛浠e彿 - String uuid = row.getCell(0).getStringCellValue(); - if (!Cools.isEmpty(uuid)) { - if (cstmrService.selectByUuid(hostId, uuid) != null) { - throw new CoolException(uuid + "瀹㈡埛浠e彿宸插瓨鍦紝璇烽噸鏂板鍏�"); - } else { - cstmr.setUuid(uuid); - } - } else { - cstmr.setUuid(cstmrService.getUuid(hostId)); - } - // 瀹㈡埛浠e彿 - String name = row.getCell(1).getStringCellValue(); - if (Cools.isEmpty(name)) { - throw new CoolException("绗�" + i + "琛屽鎴峰悕绉颁笉鑳戒负绌猴紝璇烽噸鏂板鍏�"); - } - cstmr.setName(name); - // 瀹㈡埛绫诲埆 - String type = row.getCell(2).getStringCellValue(); - if (Cools.isEmpty(type)) { - throw new CoolException("绗�" + i + "琛屽鎴风被鍒笉鑳戒负绌猴紝璇烽噸鏂板鍏�"); - } - CstmrType cstmrType = cstmrTypeService.selectByName(hostId, type); - if (null == cstmrType) { - throw new CoolException("绗�" + i + "琛屽鎴风被鍒笉鍚堟硶锛岃閲嶆柊瀵煎叆"); - } - cstmr.setCstmrType(cstmrType.getId()); - // 瀹㈡埛绫诲埆 - cstmr.setRela(row.getCell(3).getStringCellValue()); - // 鐢佃瘽 - cstmr.setTel(row.getCell(4).getStringCellValue()); - // 瀹㈡埛鑱旂郴浜� - cstmr.setContacts(row.getCell(5).getStringCellValue()); - // 澶囨敞 - cstmr.setRemarks(row.getCell(6).getStringCellValue()); - // 鐪� - String province = row.getCell(7).getStringCellValue(); - Dic dicProvince = dicService.selectByName(province); - if (Cools.isEmpty(dicProvince)) { - throw new CoolException("绗�" + i + "琛岀渷鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); - } - cstmr.setProvince(dicProvince.getCode()); - // 甯� - String city = row.getCell(8).getStringCellValue(); - Dic dicCity = dicService.selectByName(city); - if (Cools.isEmpty(dicCity)) { - throw new CoolException("绗�" + i + "琛屽競鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); - } - cstmr.setCity(dicCity.getCode()); - // 鍖� - String district = row.getCell(9).getStringCellValue(); - Dic dicDistrict = dicService.selectByName(district); - if (Cools.isEmpty(dicDistrict)) { - throw new CoolException("绗�" + i + "琛屽幙鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); - } - cstmr.setDistrict(dicDistrict.getCode()); - // 璇︾粏鍦板潃 - cstmr.setAddr(row.getCell(10).getStringCellValue()); - // 鍖哄垎 - cstmr.setType0(row.getCell(11).getStringCellValue()); - cstmr.setHostId(hostId); - cstmr.setDeptId(deptId); - cstmr.setUserId(userId); - cstmr.setStatus(1); - cstmr.setCreateBy(userId); - cstmr.setCreateTime(now); - cstmr.setUpdateBy(userId); - cstmr.setUpdateTime(now); - if (!cstmrService.insert(cstmr)) { - throw new CoolException("淇濆瓨澶辫触锛岃閲嶆柊瀵煎叆"); - } - } - } catch (Exception e) { - e.printStackTrace(); + InputStream inStream = file.getInputStream(); + String fileMime = file.getContentType(); + int excelVersion = 2007; + if ("application/vnd.ms-excel".equals(fileMime)) { + excelVersion = 2003; } - + Workbook book = null; + if (excelVersion == 2003) { + book = new HSSFWorkbook(inStream); + } + else { // 褰� excel 鏄� 2007 鏃� + book = new XSSFWorkbook(inStream); + } + Sheet sheet = book.getSheetAt(0); + int totalRows = sheet.getLastRowNum() + 1; // 鎬� + Long userId = getUserId(); + Long hostId = getHostId(); + Long deptId = getDeptId(); + Date now = new Date(); + DataFormatter dataFormatter = new DataFormatter(); + for (int i = 2; i < totalRows; i++) { + Cstmr cstmr = new Cstmr(); + Row row = sheet.getRow(i); + // 瀹㈡埛浠e彿 + String uuid = dataFormatter.formatCellValue(row.getCell(0)); + if (!Cools.isEmpty(uuid)) { + if (cstmrService.selectByUuid(hostId, uuid) != null) { + throw new CoolException(uuid + "瀹㈡埛浠e彿宸插瓨鍦紝璇烽噸鏂板鍏�"); + } else { + cstmr.setUuid(uuid); + } + } else { + cstmr.setUuid(cstmrService.getUuid(hostId)); + } + // 瀹㈡埛浠e彿 + String name = dataFormatter.formatCellValue(row.getCell(1)); + if (Cools.isEmpty(name)) { + throw new CoolException("绗�" + i + "琛屽鎴峰悕绉颁笉鑳戒负绌猴紝璇烽噸鏂板鍏�"); + } + cstmr.setName(name); + // 瀹㈡埛绫诲埆 + String type = dataFormatter.formatCellValue(row.getCell(2)); + if (Cools.isEmpty(type)) { + throw new CoolException("绗�" + i + "琛屽鎴风被鍒笉鑳戒负绌猴紝璇烽噸鏂板鍏�"); + } + CstmrType cstmrType = cstmrTypeService.selectByName(hostId, type); + if (null == cstmrType) { + throw new CoolException("绗�" + i + "琛屽鎴风被鍒笉鍚堟硶锛岃閲嶆柊瀵煎叆"); + } + cstmr.setCstmrType(cstmrType.getId()); + // 瀹㈡埛鍏崇郴 + cstmr.setRela(dataFormatter.formatCellValue(row.getCell(3))); + // 鐢佃瘽 + cstmr.setTel(dataFormatter.formatCellValue(row.getCell(4))); + // 瀹㈡埛鑱旂郴浜� + cstmr.setContacts(dataFormatter.formatCellValue(row.getCell(5))); + // 澶囨敞 + cstmr.setRemarks(dataFormatter.formatCellValue(row.getCell(6))); + // 鐪� + String province = dataFormatter.formatCellValue(row.getCell(7)); + Dic dicProvince = dicService.selectByName(province); + if (Cools.isEmpty(dicProvince)) { + throw new CoolException("绗�" + i + "琛岀渷鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); + } + cstmr.setProvince(dicProvince.getCode()); + // 甯� + String city = dataFormatter.formatCellValue(row.getCell(8)); + Dic dicCity = dicService.selectByName(city); + if (Cools.isEmpty(dicCity)) { + throw new CoolException("绗�" + i + "琛屽競鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); + } + cstmr.setCity(dicCity.getCode()); + // 鍖� + String district = dataFormatter.formatCellValue(row.getCell(9)); + Dic dicDistrict = dicService.selectByName(district); + if (Cools.isEmpty(dicDistrict)) { + throw new CoolException("绗�" + i + "琛屽幙鍚嶄笉鍚堟硶锛岃閲嶆柊瀵煎叆"); + } + cstmr.setDistrict(dicDistrict.getCode()); + // 璇︾粏鍦板潃 + cstmr.setAddr(dataFormatter.formatCellValue(row.getCell(10))); + // 鍖哄垎 + cstmr.setType0(dataFormatter.formatCellValue(row.getCell(11))); + cstmr.setHostId(hostId); + cstmr.setDeptId(deptId); + cstmr.setUserId(userId); + cstmr.setStatus(1); + cstmr.setCreateBy(userId); + cstmr.setCreateTime(now); + cstmr.setUpdateBy(userId); + cstmr.setUpdateTime(now); + if (!cstmrService.insert(cstmr)) { + throw new CoolException("淇濆瓨澶辫触锛岃閲嶆柊瀵煎叆"); + } + } return R.ok(); } -- Gitblit v1.9.1