| | |
| | | import com.ruoyi.domain.ZInfoUser; |
| | | import com.ruoyi.domain.ZfEvent; |
| | | import com.ruoyi.domain.ZfEvent; |
| | | import com.ruoyi.domain.ZfProperty; |
| | | import com.ruoyi.domain.ZfEvent; |
| | | import com.ruoyi.mapper.ZfEventMapper; |
| | | import com.ruoyi.service.ZInfoUserService; |
| | | import com.ruoyi.service.ZfEventService; |
| | | import com.ruoyi.util.ArraysUtil; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | import org.springframework.stereotype.Service; |
| | | import org.springframework.transaction.annotation.Transactional; |
| | | import org.springframework.web.multipart.MultipartFile; |
| | | |
| | | import javax.annotation.Resource; |
| | | import java.util.HashMap; |
| | | import java.util.List; |
| | | import java.util.stream.Collectors; |
| | | |
| | | import static com.ruoyi.constant.MenuAuthority.EVENT_LIST; |
| | | import static com.ruoyi.constant.MenuAuthority.PROPERTY_LIST; |
| | | |
| | | /** |
| | | * <p> |
| | |
| | | @Resource |
| | | ZfEventService zfEventService; |
| | | |
| | | private String listFamilyIds() { |
| | | private LambdaQueryWrapper<ZfEvent> buildCondition(ZfEvent zfEvent) { |
| | | LambdaQueryWrapper<ZfEvent> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getAddress()), ZfEvent::getAddress, zfEvent.getAddress()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getPeople()), ZfEvent::getPeople, zfEvent.getPeople()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getTitle()), ZfEvent::getTitle, zfEvent.getTitle()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getRemark()), ZfEvent::getRemark, zfEvent.getRemark()); |
| | | lqw.eq(zfEvent.getHappenTime() != null, ZfEvent::getHappenTime, zfEvent.getHappenTime()); |
| | | lqw.between(zfEvent.getHappenStartTime() != null && zfEvent.getHappenEndTime() != null, ZfEvent::getHappenTime, zfEvent.getHappenStartTime(), zfEvent.getHappenEndTime()); |
| | | return lqw; |
| | | } |
| | | |
| | | private LambdaQueryWrapper<ZfEvent> buildCondition(ZfEvent zfEvent, String familyIds, String secondFamilyAuthority) { |
| | | //模糊查询的条件 |
| | | LambdaQueryWrapper<ZfEvent> lqw = buildCondition(zfEvent); |
| | | |
| | | //定义一个布尔标记,标记第一家庭号是否有逗号 |
| | | boolean flag = false; |
| | | |
| | | StringBuilder secondFamilyIdsBuffer = new StringBuilder(); |
| | | |
| | | //处理传入的secondFamilyAuthority 3{2007 2018 2015},4{2007 2019} |
| | | |
| | | if (StringUtils.isNotEmpty(secondFamilyAuthority)) { |
| | | |
| | | if (secondFamilyAuthority.contains(",")) { |
| | | //逗号隔开得到如下串:3{2007 2018 2015} |
| | | String[] authorityList = secondFamilyAuthority.split(","); |
| | | for (String authorityAndId : authorityList) { |
| | | setSecondFamilyAuthority(secondFamilyIdsBuffer, authorityAndId); |
| | | } |
| | | |
| | | //把后面的逗号去掉 |
| | | if (secondFamilyIdsBuffer.length() > 1) { |
| | | secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1); |
| | | } |
| | | |
| | | } else { |
| | | setSecondFamilyAuthority(secondFamilyIdsBuffer, secondFamilyAuthority); |
| | | //把后面的逗号去掉 |
| | | if (secondFamilyIdsBuffer.length() > 1) { |
| | | secondFamilyIdsBuffer.deleteCharAt(secondFamilyIdsBuffer.length() - 1); |
| | | } |
| | | } |
| | | } |
| | | //将StringBuffer转成String |
| | | String secondFamilyIds = secondFamilyIdsBuffer.toString(); |
| | | |
| | | //第一家庭号的条件 |
| | | if (familyIds.contains(",")) { |
| | | flag = true; |
| | | } |
| | | |
| | | //第二家庭号的条件 |
| | | if (StringUtils.isNotEmpty(secondFamilyIds)) { |
| | | if (secondFamilyIds.contains(",")) { |
| | | String[] secondFamilyIdList = secondFamilyIds.split(","); |
| | | if (flag) { |
| | | //如果第一家庭号和第二家庭号都有逗号 |
| | | String[] familyList = familyIds.split(","); |
| | | String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIdList); |
| | | lqw.in(ZfEvent::getFamilyId, totalFamilyIdList); |
| | | } else { |
| | | //如果第一家庭号没有逗号但是第二家庭号有逗号 |
| | | String[] totalFamilyIdList = ArraysUtil.appendToArray(secondFamilyIdList, familyIds); |
| | | lqw.in(ZfEvent::getFamilyId, totalFamilyIdList); |
| | | } |
| | | |
| | | } else { |
| | | if (flag) { |
| | | //如果第一家庭号有逗号但第二家庭号没有逗号 |
| | | String[] familyList = familyIds.split(","); |
| | | String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIds); |
| | | lqw.in(ZfEvent::getFamilyId, totalFamilyIdList); |
| | | } else { |
| | | //如果第一家庭号和第二家庭号都没有逗号 |
| | | String[] totalFamilyIdList = ArraysUtil.appendToArray(familyIds, secondFamilyIds); |
| | | lqw.in(ZfEvent::getFamilyId, totalFamilyIdList); |
| | | } |
| | | } |
| | | } else { |
| | | //如果没有第二家庭号 |
| | | if (flag) { |
| | | String[] familyList = familyIds.split(","); |
| | | lqw.in(ZfEvent::getFamilyId, familyList); |
| | | } else { |
| | | lqw.eq(ZfEvent::getFamilyId, familyIds); |
| | | } |
| | | |
| | | } |
| | | |
| | | lqw.orderByDesc(ZfEvent::getCreateTime); |
| | | return lqw; |
| | | } |
| | | |
| | | private void setSecondFamilyAuthority(StringBuilder secondFamilyIdsBuffer, String authorityAndId) { |
| | | //得到权限串 |
| | | String authorities = authorityAndId.substring(2, authorityAndId.length() - 1); |
| | | |
| | | if (authorities.contains(" ")) { |
| | | //空格隔开得到菜单id,看是否有对应的权限 |
| | | String[] authorityIdList = authorities.split(" "); |
| | | for (String s : authorityIdList) { |
| | | if (s.equals(EVENT_LIST)) { |
| | | secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(","); |
| | | break; |
| | | } |
| | | } |
| | | } else { |
| | | if (authorities.equals(EVENT_LIST)) { |
| | | secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(","); |
| | | } |
| | | } |
| | | } |
| | | |
| | | private ZInfoUser getMySelf() { |
| | | SysUser user = SecurityUtils.getLoginUser().getUser(); |
| | | Long userId = user.getUserId(); |
| | | LambdaQueryWrapper<ZInfoUser> zInfoUserLambdaQueryWrapper = new LambdaQueryWrapper<>(); |
| | | zInfoUserLambdaQueryWrapper.eq(ZInfoUser::getUserId, userId); |
| | | ZInfoUser zInfoUser = zInfoUserService.getOne(zInfoUserLambdaQueryWrapper); |
| | | return zInfoUser.getFamilyId(); |
| | | return zInfoUserService.getOne(zInfoUserLambdaQueryWrapper); |
| | | } |
| | | |
| | | |
| | | @Override |
| | | public AjaxResult selectEventList(ZfEvent zfEvent, Integer pageNum, Integer pageSize) { |
| | | String familyIds = listFamilyIds(); |
| | | LambdaQueryWrapper<ZfEvent> lqw = buildCondition(zfEvent, familyIds); |
| | | |
| | | Page<ZfEvent> ZfEventPage = new Page<>(pageNum, pageSize); |
| | | Page<ZfEvent> pageResult = page(ZfEventPage, lqw); |
| | | |
| | | HashMap<String, Object> data = MapUtils.getResult(pageResult); |
| | | return AjaxResult.success(data); |
| | | private String listFamilyIds() { |
| | | ZInfoUser mySelf = getMySelf(); |
| | | return mySelf.getFamilyId(); |
| | | } |
| | | |
| | | private String listSecondFamilyIds() { |
| | | ZInfoUser mySelf = getMySelf(); |
| | | return mySelf.getSecondFamilyId(); |
| | | } |
| | | |
| | | @Override |
| | | public List<ZfEvent> selectByCondition(ZfEvent zfEvent) { |
| | | String familyIds = listFamilyIds(); |
| | | LambdaQueryWrapper<ZfEvent> lqw = buildCondition(zfEvent, familyIds); |
| | | List<ZfEvent> list = list(lqw); |
| | | log.info("返回的数据为:{}", list); |
| | | return list; |
| | | String secondFamilyIds = listSecondFamilyIds(); |
| | | LambdaQueryWrapper<ZfEvent> lambdaQueryWrapper = buildCondition(zfEvent, familyIds, secondFamilyIds); |
| | | List<ZfEvent> beanRecords = list(lambdaQueryWrapper); |
| | | log.info("从数据库中查到的为:{}", beanRecords); |
| | | return markOwnData(familyIds, beanRecords); |
| | | } |
| | | |
| | | private static List<ZfEvent> markOwnData(String familyIds, List<ZfEvent> beanRecords) { |
| | | return beanRecords.stream().peek(record -> { |
| | | String recordFamilyId = record.getFamilyId(); |
| | | if (familyIds.contains(",")) { |
| | | String[] familyIdList = familyIds.split(","); |
| | | for (String familyId : familyIdList) { |
| | | if (familyId.equals(recordFamilyId)) { |
| | | record.setOwnData(1); |
| | | } |
| | | } |
| | | } else { |
| | | if (familyIds.equals(recordFamilyId)) { |
| | | record.setOwnData(1); |
| | | } |
| | | } |
| | | }).collect(Collectors.toList()); |
| | | } |
| | | |
| | | @Override |
| | | public int addEvent(ZfEvent zfEvent) { |
| | | public AjaxResult selectDataList(ZfEvent zfEvent, Integer pageNum, Integer pageSize) { |
| | | String familyIds = listFamilyIds(); |
| | | String secondFamilyAuthority = listSecondFamilyIds(); |
| | | LambdaQueryWrapper<ZfEvent> lqw = buildCondition(zfEvent, familyIds, secondFamilyAuthority); |
| | | |
| | | Page<ZfEvent> zfEventPage = new Page<>(pageNum, pageSize); |
| | | Page<ZfEvent> pageResult = page(zfEventPage, lqw); |
| | | |
| | | List<ZfEvent> beanRecords = pageResult.getRecords();//得到查询出来的数据 |
| | | |
| | | List<ZfEvent> dtoResult = markOwnData(familyIds, beanRecords); |
| | | |
| | | HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult); |
| | | return AjaxResult.success(data); |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult addData(ZfEvent zfEvent) { |
| | | |
| | | String familyIds = listFamilyIds(); |
| | | boolean flag = false;//判断当前用户的id是否有权加入当前家庭id的对象 |
| | | if (familyIds.contains(",")) { |
| | |
| | | } |
| | | } |
| | | if (flag) { |
| | | boolean save = save(zfEvent); |
| | | return save ? 1 : 0; |
| | | if (save(zfEvent)) { |
| | | return AjaxResult.success(); |
| | | } else { |
| | | return AjaxResult.error(); |
| | | } |
| | | |
| | | } else { |
| | | throw new RuntimeException("你没有操作该家庭号为" + zfEvent.getFamilyId() + "数据的权限"); |
| | | } |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult addData2(ZfEvent zfEvent) { |
| | | ZInfoUser mySelf = getMySelf(); |
| | | String myFamilyId = mySelf.getFamilyId(); |
| | | |
| | | if (StringUtils.isNotEmpty(zfEvent.getFamilyId())) { |
| | | throw new RuntimeException("导入数据时您不能指定家庭号"); |
| | | } |
| | | |
| | | if (StringUtils.isEmpty(myFamilyId)) { |
| | | throw new RuntimeException("您还未加入任何家庭"); |
| | | } |
| | | |
| | | if (myFamilyId.contains(",")) { |
| | | String[] myFamilyIds = myFamilyId.split(","); |
| | | //默认添加第一个家庭的数据 |
| | | zfEvent.setFamilyId(myFamilyIds[0]); |
| | | } else { |
| | | zfEvent.setFamilyId(myFamilyId); |
| | | } |
| | | |
| | | if (save(zfEvent)) { |
| | | return AjaxResult.success(); |
| | | } else { |
| | | return AjaxResult.error(); |
| | | } |
| | | } |
| | | |
| | | |
| | | @Override |
| | | @Transactional |
| | | public AjaxResult importExcel(MultipartFile file) { |
| | | ExcelUtil<ZfEvent> util = new ExcelUtil<>(ZfEvent.class); |
| | | List<ZfEvent> eventList = null; |
| | | List<ZfEvent> propertyList = null; |
| | | try { |
| | | eventList = util.importExcel(file.getInputStream()); |
| | | propertyList = util.importExcel(file.getInputStream()); |
| | | } catch (Exception e) { |
| | | throw new RuntimeException(e); |
| | | } |
| | | log.info("资产列表为:{}", eventList); |
| | | log.info("资产列表为:{}", propertyList); |
| | | |
| | | for (ZfEvent zfEvent : eventList) { |
| | | zfEventService.addEvent(zfEvent); |
| | | for (ZfEvent zfEvent : propertyList) { |
| | | zfEventService.addData2(zfEvent); |
| | | } |
| | | |
| | | return AjaxResult.success("导入数据成功"); |
| | | |
| | | } |
| | | |
| | | private LambdaQueryWrapper<ZfEvent> buildCondition(ZfEvent zfEvent) { |
| | | LambdaQueryWrapper<ZfEvent> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getAddress()), ZfEvent::getAddress, zfEvent.getAddress()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getPeople()), ZfEvent::getPeople, zfEvent.getPeople()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getTitle()), ZfEvent::getTitle, zfEvent.getTitle()); |
| | | lqw.like(!StringUtils.isEmpty(zfEvent.getRemark()), ZfEvent::getRemark, zfEvent.getRemark()); |
| | | lqw.eq(zfEvent.getHappenTime() != null, ZfEvent::getHappenTime, zfEvent.getHappenTime()); |
| | | lqw.between(zfEvent.getHappenStartTime() != null && zfEvent.getHappenEndTime() != null, ZfEvent::getHappenTime, zfEvent.getHappenStartTime(), zfEvent.getHappenEndTime()); |
| | | return lqw; |
| | | } |
| | | |
| | | private LambdaQueryWrapper<ZfEvent> buildCondition(ZfEvent zfEvent, String familyIds) { |
| | | LambdaQueryWrapper<ZfEvent> lqw = buildCondition(zfEvent); |
| | | if (familyIds.contains(",")) { |
| | | String[] familyList = familyIds.split(","); |
| | | lqw.in(ZfEvent::getFamilyId, familyList); |
| | | } else { |
| | | lqw.eq(ZfEvent::getFamilyId, familyIds); |
| | | } |
| | | return lqw; |
| | | } |
| | | |
| | | |
| | | } |