From e16ecb7e0f8ce8d0ca7199f6867aa01d4c82ae62 Mon Sep 17 00:00:00 2001 From: whycq <you@example.com> Date: 星期三, 21 十二月 2022 22:21:36 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/manager/controller/CstmrController.java | 91 +++++++++++++++++++++++++++++++++++++-------- 1 files changed, 75 insertions(+), 16 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 14bc5e0..995f0d7 100644 --- a/src/main/java/com/zy/crm/manager/controller/CstmrController.java +++ b/src/main/java/com/zy/crm/manager/controller/CstmrController.java @@ -8,21 +8,23 @@ import com.core.common.BaseRes; import com.core.common.Cools; import com.core.common.R; +import com.core.domain.KeyValueVo; import com.core.exception.CoolException; import com.zy.crm.common.web.BaseController; import com.zy.crm.manager.controller.result.FollowerTableVo; import com.zy.crm.manager.entity.Cstmr; import com.zy.crm.manager.entity.CstmrFoll; import com.zy.crm.manager.entity.CstmrType; -import com.zy.crm.manager.entity.result.KeyValueVo; import com.zy.crm.manager.service.CstmrFollService; 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; import lombok.extern.slf4j.Slf4j; +import lombok.val; import org.apache.poi.hssf.usermodel.HSSFWorkbook; import org.apache.poi.ss.usermodel.DataFormatter; import org.apache.poi.ss.usermodel.Row; @@ -35,8 +37,8 @@ import org.springframework.web.multipart.MultipartFile; import javax.servlet.http.HttpServletResponse; -import java.io.IOException; -import java.io.InputStream; +import java.io.*; +import java.lang.reflect.Array; import java.util.*; @Slf4j @@ -299,6 +301,24 @@ return R.ok("娣诲姞鎴愬姛"); } + + @RequestMapping(value = "/cstmr/followers/add/json") + @ManagerAuth + @Transactional + public R cstmrFollowersAddJson(@RequestBody() Map map) { + Object cstmrId = map.get("cstmrId"); + Object followerIds = map.get("followerIds"); + long cstmrId1 = Long.valueOf(String.valueOf(cstmrId)); + long followerIds1 = Long.valueOf(String.valueOf(followerIds)); + CstmrFoll cstmrFoll = new CstmrFoll(); + cstmrFoll.setCstmrId(cstmrId1); + cstmrFoll.setUserId(followerIds1); + if (!cstmrFollService.insert(cstmrFoll)) { + throw new CoolException("娣诲姞澶辫触锛岃鑱旂郴绠$悊鍛�"); + } + return R.ok("娣诲姞鎴愬姛"); + } + @RequestMapping(value = "/cstmr/followers/remove/auth") @ManagerAuth public R cstmrFollowersRemove(@RequestParam("cstmrId") Long cstmrId, @@ -312,22 +332,61 @@ return R.ok("鍒犻櫎鎴愬姛"); } - // xm-select @RequestMapping("/cstmr/all/get/kv") @ManagerAuth - public R getCstmrDataKV(@RequestParam(required = false) String condition) { - Wrapper<Cstmr> wrapper = new EntityWrapper<Cstmr>() - .andNew().like("name", condition).or().like("uuid", condition) - .orderBy("create_time", false); - List<Cstmr> cstmrs = cstmrService.selectPage(new Page<>(1, 30), wrapper).getRecords(); - List<KeyValueVo> valueVos = new ArrayList<>(); - for (Cstmr cstmr : cstmrs) { - KeyValueVo vo = new KeyValueVo(); - vo.setName(cstmr.getName() + "(" + cstmr.getUuid() + ")"); - vo.setValue(cstmr.getId()); - valueVos.add(vo); + 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<>(); + 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); + // 瀹㈡埛浠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().add(valueVos); + return R.ok(); } } -- Gitblit v1.9.1