package com.linkwechat.controller; import cn.hutool.core.collection.CollectionUtil; import cn.hutool.core.collection.ListUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.linkwechat.common.annotation.PhoneEncryptMethod; import com.linkwechat.common.core.controller.BaseController; import com.linkwechat.common.core.domain.AjaxResult; import com.linkwechat.common.core.domain.entity.SysUser; import com.linkwechat.common.core.page.TableDataInfo; import com.linkwechat.common.exception.wecom.WeComException; import com.linkwechat.common.utils.*; import com.linkwechat.common.utils.poi.ExcelUtil; import com.linkwechat.common.utils.poi.LwExcelUtil; import com.linkwechat.domain.WeCustomerSeas; import com.linkwechat.domain.live.WeLive; import com.linkwechat.domain.wecom.vo.customer.seas.CustomerSeasRecordVo; import com.linkwechat.service.IWeCustomerSeasService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; import org.springframework.web.multipart.MultipartFile; import java.text.MessageFormat; import java.util.*; import java.util.stream.Collectors; import java.util.stream.IntStream; /** * 客户公海 */ @RestController @RequestMapping(value = "/seas") public class WeCustomerSeasController extends BaseController { @Autowired private IWeCustomerSeasService iWeCustomerSeasService; /** * 客户公海模版下载 * @return */ @GetMapping("/importTemplate") public void importTemplate() { LwExcelUtil.exprotForWeb( ServletUtils.getResponse(), WeCustomerSeas.class, ListUtil.toList(new WeCustomerSeas()),"客户公海" ); } /** * 导入公海数据 * @param file * @param weCustomerSea * @return */ @PostMapping("/importData") public AjaxResult importData(MultipartFile file, WeCustomerSeas weCustomerSea) throws Exception { ExcelUtil util = new ExcelUtil(WeCustomerSeas.class); List weCustomerSeas = util.importExcel(file.getInputStream()); String tip=new String("成功导入{0}条，去重复{1}条"); if(CollectionUtil.isNotEmpty(weCustomerSeas)){ if(StringUtils.isEmpty(weCustomerSea.getAddUserName())&&StringUtils.isEmpty(weCustomerSea.getAddUserId())){ return AjaxResult.error("分配人不可为空！"); } //过滤不符合规范的手机号 List noRuleWeCustomerSeas = weCustomerSeas.stream().filter(s -> StringUtils.isMobile(s.getPhone())).collect(Collectors.toList()); if(CollectionUtil.isEmpty(noRuleWeCustomerSeas)){ return AjaxResult.error("请传入合法手机号！"); } //过滤字段为空的数据 List deduplicationSeas = noRuleWeCustomerSeas.stream().filter(s -> StringUtils.isNotEmpty(s.getCustomerName()) || StringUtils.isNotEmpty(s.getPhone())).collect(Collectors.toList()); if(CollectionUtil.isEmpty(deduplicationSeas)){ return AjaxResult.error("导入用户数据不能为空！"); } //根据手机号去重(去除excel中重复的号码) List deduplicationSeasNoRepeat=deduplicationSeas.stream().collect(Collectors.collectingAndThen(Collectors.toCollection(() -> new TreeSet<>(Comparator.comparing(WeCustomerSeas :: getPhone))), ArrayList::new)); //根据手机号去除数据库与excel中重复号码 List dbExist = iWeCustomerSeasService.list(new LambdaQueryWrapper() .in(WeCustomerSeas::getPhone, deduplicationSeasNoRepeat.stream() .map(WeCustomerSeas::getPhone).collect(Collectors.toList()))); if(CollectionUtil.isNotEmpty(dbExist)){ List noRepetWeCustomerSeas = deduplicationSeasNoRepeat.stream().filter(item -> !dbExist.stream().map(e->e.getPhone()).collect(Collectors.toList()) .contains(item.getPhone())).collect(Collectors.toList()); deduplicationSeasNoRepeat.clear(); deduplicationSeasNoRepeat.addAll( noRepetWeCustomerSeas ); } if(CollectionUtil.isNotEmpty(deduplicationSeasNoRepeat)){ Long tabaleExcelId= SnowFlakeUtil.nextId(); List userIds = ListUtil.toList(weCustomerSea.getAddUserId().split(",")); List userNames = ListUtil.toList(weCustomerSea.getAddUserName().split(",")); IntStream.range(0,deduplicationSeasNoRepeat.size()).forEach(i->{ WeCustomerSeas k = deduplicationSeasNoRepeat.get(i); if(Objects.nonNull(k)){ k.setTagIds(weCustomerSea.getTagIds()); k.setTagNames(weCustomerSea.getTagNames()); k.setAddUserId( userIds.get(Math.floorMod(i, userIds.size()))); k.setAddUserName(userNames.get(Math.floorMod(i, userNames.size()))); k.setTableExcelName(file.getOriginalFilename()); k.setTableExcelId(tabaleExcelId); } }); if(iWeCustomerSeasService.saveBatch(deduplicationSeasNoRepeat)){ //发送提醒 deduplicationSeasNoRepeat.stream().collect(Collectors.groupingBy(WeCustomerSeas::getAddUserId)).forEach((k,v)->{ iWeCustomerSeasService.remidUser(ListUtil.toList(k),v.size()); }); } tip = MessageFormat.format(tip, new Object[]{new Integer(deduplicationSeasNoRepeat.size()).toString(), new Integer( weCustomerSeas.size()-deduplicationSeasNoRepeat.size()).toString()}); }else{ tip = MessageFormat.format(tip, new Object[] { "0",weCustomerSeas.size() }); } ; }else{ return AjaxResult.error("导入用户数据不能为空！"); } return AjaxResult.success(tip); } /** * 获取公海客户列表 * @param weCustomerSea * @return */ @GetMapping("list") public TableDataInfo list(WeCustomerSeas weCustomerSea){ startPage(); return getDataTable( iWeCustomerSeasService.findWeCustomerSeas(weCustomerSea) ); } /** * 删除公海客户 * @param ids * @return */ @DeleteMapping("/{ids}") public AjaxResult remove(@PathVariable Long[] ids) { iWeCustomerSeasService.removeByIds( ListUtil.toList(ids) ); return AjaxResult.success(); } /** * 提醒员工 * @param ids * @return */ @PostMapping("/remidUser/{ids}") public AjaxResult remidUser(@PathVariable Long[] ids){ List weCustomerSeas = iWeCustomerSeasService.listByIds(ListUtil.toList(ids)); if(CollectionUtil.isNotEmpty(weCustomerSeas)){ weCustomerSeas.stream().collect(Collectors.groupingBy(WeCustomerSeas::getAddUserId)).forEach((k,v)->{ iWeCustomerSeasService.remidUser(ListUtil.toList(k),v.size()); }); } return AjaxResult.success(); } /** * 详情头部统计 * @return */ @GetMapping("/countCustomerSeas") public AjaxResult countCustomerSeas(){ return AjaxResult.success( iWeCustomerSeasService.countCustomerSeas() ); } /** * 导入记录或员工添加统计 * @param groupByType 1:获取导入记录数据 2:员工添加统计 * @return */ @GetMapping("/findSeasRecord") public TableDataInfo findSeasRecord(@RequestParam(defaultValue = "1") Integer groupByType){ startPage(); List seasRecord = iWeCustomerSeasService.findSeasRecord(groupByType); return getDataTable(seasRecord); } /** * 设置公海客户状态 * @return */ @PostMapping("/setState") public AjaxResult setState(@RequestBody WeCustomerSeas weCustomerSea){ if(StringUtils.isEmpty(weCustomerSea.getPhone())){ return AjaxResult.error("手机号不可为空"); } iWeCustomerSeasService.update(WeCustomerSeas.builder().addState(new Integer(3)).build() ,new LambdaQueryWrapper().eq(WeCustomerSeas::getPhone,weCustomerSea.getPhone())); return AjaxResult.success(); } /** * 获取指定员工所拥有的状态(移动端) * @param addState * @return */ @GetMapping("/findEmployeeCustomer") public TableDataInfo findEmployeeCustomer(Integer addState){ SysUser sysUser = SecurityUtils.getLoginUser().getSysUser(); String userId = sysUser.getWeUserId(); if (StringUtils.isBlank(userId)) { throw new WeComException("token中获取的we_user_id为空！"); } startPage(); List weCustomerSeas = iWeCustomerSeasService.list(new LambdaQueryWrapper() .eq(StringUtils.isNotEmpty(userId),WeCustomerSeas::getAddUserId, userId) .eq(addState != null,WeCustomerSeas::getAddState, addState)); return getDataTable(weCustomerSeas); } }