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 | 56 ++++++++++++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 52 insertions(+), 4 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 09ed9c0..b940a38 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.manager.service.CstmrService;
import com.zy.crm.manager.service.CstmrTypeService;
import com.zy.crm.system.entity.Dic;
+import com.zy.crm.system.entity.Role;
import com.zy.crm.system.entity.User;
import com.zy.crm.system.service.DicService;
import com.zy.crm.system.service.UserService;
@@ -35,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
@@ -315,10 +315,58 @@
@RequestMapping("/cstmr/all/get/kv")
@ManagerAuth
public R getDataKV(@RequestParam(required = false) String condition) {
+ User user = getUser(); assert user != null;
+ Role role = getRole(); assert role != null;
+ Page<Cstmr> page = cstmrService.getPage(new Page<>(1, 30)
+ , getHostId()
+ , role.judgeLeader() ? String.valueOf(user.getDeptId()) : null
+ , user.getId()
+ , condition);
List<KeyValueVo> vos = new ArrayList<>();
- Wrapper<Cstmr> wrapper = new EntityWrapper<Cstmr>().andNew().like("name", condition).or().like("uuid", condition).orderBy("create_time", false);
- cstmrService.selectPage(new Page<>(1, 30), wrapper).getRecords().forEach(item -> vos.add(new KeyValueVo(item.getName(), item.getId())));
+ page.getRecords().forEach(item -> vos.add(new KeyValueVo(item.getName(), item.getId())));
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);
+ // 瀹㈡埛浠ｅ彿
+ 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