From ef9bb2954b049859d533fa27f04c782c8b4c3ebb Mon Sep 17 00:00:00 2001 From: luxiaotao1123 <t1341870251@63.com> Date: 星期一, 28 十一月 2022 14:57:43 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/manager/controller/CstmrController.java | 45 +++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 43 insertions(+), 2 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 5fa5d2a..b940a38 100644 --- a/src/main/java/com/zy/crm/manager/controller/CstmrController.java +++ b/src/main/java/com/zy/crm/manager/controller/CstmrController.java @@ -36,8 +36,7 @@ import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; -import java.io.IOException; -import java.io.InputStream; +import java.io.*; import java.util.*; @Slf4j @@ -328,4 +327,46 @@ return R.ok().add(vos); } + @RequestMapping("/cstmr/file/import") + @Transactional + public R fileImport() throws IOException { + File file = new File("D:\\tmp\\璺熻繘浜�.xls"); + InputStream inStream = new FileInputStream(file); + int excelVersion = 2007; + Workbook book = null; + book = new HSSFWorkbook(inStream); + Sheet sheet = book.getSheetAt(0); + int totalRows = sheet.getLastRowNum() + 1; // 鎬� + DataFormatter dataFormatter = new DataFormatter(); + for (int i = 1; i < totalRows; i++) { + System.out.println(i); + Row row = sheet.getRow(i); + // 瀹㈡埛浠e彿 + String username = dataFormatter.formatCellValue(row.getCell(0)); + System.out.println(username); + if (username.equals("鐢樺叏") || username.equals("闄嗘檽涓�") + || username.equals("寮犲ぉ鏋�") || username.equals("闄堥箯") || username.equals("璧甸洩鐢�") + || username.equals("寮犱繆鑹�(绂昏亴)") + ) { continue;} + String cstmrUuid = dataFormatter.formatCellValue(row.getCell(2)); + System.out.println(cstmrUuid); + + User user = userService.selectOne(new EntityWrapper<User>().eq("nickname", username)); + if (null == user) { + throw new CoolException(username + "閿欒"); + } + Cstmr cstmr = cstmrService.selectByUuid(null, cstmrUuid); + if (null == cstmr) { + throw new CoolException(cstmrUuid + "閿欒"); + } + CstmrFoll cstmrFoll = new CstmrFoll(); + cstmrFoll.setCstmrId(cstmr.getId()); + cstmrFoll.setUserId(user.getId()); + if (!cstmrFollService.insert(cstmrFoll)) { + throw new CoolException("娣诲姞澶辫触锛岃鑱旂郴绠$悊鍛�"); + } + } + return R.ok(); + } + } -- Gitblit v1.9.1