From e517585cc08d1d53b5a280193540f69301b63543 Mon Sep 17 00:00:00 2001 From: whycq <10027870+whycq@user.noreply.gitee.com> Date: 星期五, 23 十二月 2022 09:23:46 +0800 Subject: [PATCH] # --- src/main/java/com/zy/crm/manager/controller/CstmrController.java | 109 ++++++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 92 insertions(+), 17 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 92337c6..7bf253d 100644 --- a/src/main/java/com/zy/crm/manager/controller/CstmrController.java +++ b/src/main/java/com/zy/crm/manager/controller/CstmrController.java @@ -1,5 +1,7 @@ package com.zy.crm.manager.controller; +import com.alibaba.fastjson.JSON; +import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.baomidou.mybatisplus.mapper.EntityWrapper; import com.baomidou.mybatisplus.mapper.Wrapper; @@ -8,21 +10,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 +39,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 @@ -107,8 +111,21 @@ @RequestMapping(value = "/cstmr/delete/auth") @ManagerAuth public R delete(@RequestParam(value="ids[]") Long[] ids){ - for (Long id : ids){ + for (Long id : ids){ cstmrService.deleteById(id); + } + return R.ok(); + } + + @RequestMapping(value = "/cstmr/delete/json") + @ManagerAuth + public R deleteJson(@RequestBody() String o){ + JSONObject jsonObject = JSON.parseObject(o); + JSONArray ids = jsonObject.getJSONArray("ids"); + for (int i = 0; i < ids.size(); i++) { + Long id = Long.valueOf((String) ids.get(i)); + cstmrService.deleteById(id); + } return R.ok(); } @@ -299,6 +316,25 @@ return R.ok("娣诲姞鎴愬姛"); } + + @RequestMapping(value = "/cstmr/followers/add/json") + @ManagerAuth + @Transactional + public R cstmrFollowersAddJson(@RequestBody() Map map) { + long cstmrId = Long.valueOf(String.valueOf(map.get("cstmrId"))); + long followerIds = Long.valueOf(String.valueOf(map.get("followerIds"))); + if (Cools.isEmpty(cstmrId, followerIds)) { + return R.parse(BaseRes.PARAM); + } + CstmrFoll cstmrFoll = new CstmrFoll(); + cstmrFoll.setCstmrId(cstmrId); + cstmrFoll.setUserId(followerIds); + 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 +348,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()); - 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