| | |
| | | import com.baomidou.mybatisplus.extension.plugins.pagination.Page; |
| | | import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; |
| | | import com.ruoyi.common.core.domain.AjaxResult; |
| | | import com.ruoyi.common.core.domain.entity.EsModel; |
| | | import com.ruoyi.common.core.domain.entity.SysUser; |
| | | import com.ruoyi.common.utils.MapUtils; |
| | | import com.ruoyi.common.utils.SecurityUtils; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | import com.ruoyi.common.utils.poi.ExcelUtil; |
| | | import com.ruoyi.domain.ZInfoUser; |
| | | import com.ruoyi.domain.*; |
| | | import com.ruoyi.domain.ZfEquipment; |
| | | import com.ruoyi.domain.ZfEquipment; |
| | | import com.ruoyi.domain.ZfProperty; |
| | | import com.ruoyi.mapper.ZfEquipmentMapper; |
| | | import com.ruoyi.service.ZInfoUserService; |
| | | import com.ruoyi.service.ZfEquipmentService; |
| | | import com.ruoyi.service.*; |
| | | import com.ruoyi.util.ArraysUtil; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | import org.elasticsearch.action.delete.DeleteRequest; |
| | | import org.elasticsearch.action.update.UpdateRequest; |
| | | import org.elasticsearch.client.RequestOptions; |
| | | import org.elasticsearch.client.RestHighLevelClient; |
| | | 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.io.IOException; |
| | | import java.time.LocalDateTime; |
| | | import java.util.*; |
| | | import java.util.stream.Collectors; |
| | | |
| | | import static com.ruoyi.constant.MenuAuthority.*; |
| | | |
| | | /** |
| | | * <p> |
| | |
| | | |
| | | @Resource |
| | | ZfEquipmentService zfEquipmentService; |
| | | |
| | | @Resource |
| | | ZAuthorityService zAuthorityService; |
| | | |
| | | @Resource |
| | | EsService esService; |
| | | |
| | | @Resource |
| | | private RestHighLevelClient restHighLevelClient; |
| | | |
| | | private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment,List<Long> familyIdList) { |
| | | LambdaQueryWrapper<ZfEquipment> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.orderByDesc(ZfEquipment::getCreateTime); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getName()), ZfEquipment::getName, zfEquipment.getName()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getBuyer()), ZfEquipment::getBuyer, zfEquipment.getBuyer()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getContent()), ZfEquipment::getContent, zfEquipment.getContent()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getLocation()), ZfEquipment::getLocation, zfEquipment.getLocation()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getRemark()), ZfEquipment::getRemark, zfEquipment.getRemark()); |
| | | lqw.eq(zfEquipment.getHappenTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenTime()); |
| | | lqw.in(ZfEquipment::getFamilyId,familyIdList); |
| | | lqw.between(zfEquipment.getHappenStartTime() != null && zfEquipment.getHappenEndTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenStartTime(), zfEquipment.getHappenEndTime()); |
| | | return lqw; |
| | | } |
| | | |
| | | private LambdaQueryWrapper<ZfEquipment> uniqueCondition(ZfEquipment zfEquipment) { |
| | | LambdaQueryWrapper<ZfEquipment> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.eq(StringUtils.isNotEmpty(zfEquipment.getName()), ZfEquipment::getName, zfEquipment.getName()); |
| | | lqw.eq(StringUtils.isNotEmpty(zfEquipment.getBuyer()), ZfEquipment::getBuyer, zfEquipment.getBuyer()); |
| | | lqw.eq(StringUtils.isNotEmpty(zfEquipment.getContent()), ZfEquipment::getContent, zfEquipment.getContent()); |
| | | lqw.eq(StringUtils.isNotEmpty(zfEquipment.getLocation()), ZfEquipment::getLocation, zfEquipment.getLocation()); |
| | | lqw.eq(StringUtils.isNotEmpty(zfEquipment.getRemark()), ZfEquipment::getRemark, zfEquipment.getRemark()); |
| | | lqw.eq(zfEquipment.getHappenTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenTime()); |
| | | return lqw; |
| | | } |
| | | |
| | | |
| | | |
| | | // //------------------------------------------------------------------------------------------ |
| | | // private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment, String familyIds, String secondFamilyAuthority) { |
| | | // //模糊查询的条件 |
| | | // LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment); |
| | | // |
| | | // //定义一个布尔标记,标记第一家庭号是否有逗号 |
| | | // 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(ZfEquipment::getFamilyId, totalFamilyIdList); |
| | | // } else { |
| | | // //如果第一家庭号没有逗号但是第二家庭号有逗号 |
| | | // String[] totalFamilyIdList = ArraysUtil.appendToArray(secondFamilyIdList, familyIds); |
| | | // lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList); |
| | | // } |
| | | // |
| | | // } else { |
| | | // if (flag) { |
| | | // //如果第一家庭号有逗号但第二家庭号没有逗号 |
| | | // String[] familyList = familyIds.split(","); |
| | | // String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIds); |
| | | // lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList); |
| | | // } else { |
| | | // //如果第一家庭号和第二家庭号都没有逗号 |
| | | // String[] totalFamilyIdList = ArraysUtil.appendToArray(familyIds, secondFamilyIds); |
| | | // lqw.in(ZfEquipment::getFamilyId, totalFamilyIdList); |
| | | // } |
| | | // } |
| | | // } else { |
| | | // //如果没有第二家庭号 |
| | | // if (flag) { |
| | | // String[] familyList = familyIds.split(","); |
| | | // lqw.in(ZfEquipment::getFamilyId, familyList); |
| | | // } else { |
| | | // lqw.eq(ZfEquipment::getFamilyId, familyIds); |
| | | // } |
| | | // |
| | | // } |
| | | // |
| | | // lqw.orderByDesc(ZfEquipment::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(EQUIPMENT_LIST)) { |
| | | // secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(","); |
| | | // break; |
| | | // } |
| | | // } |
| | | // } else { |
| | | // if (authorities.equals(EQUIPMENT_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); |
| | | // return zInfoUserService.getOne(zInfoUserLambdaQueryWrapper); |
| | | // } |
| | | // |
| | | // private String listFamilyIds() { |
| | | // ZInfoUser mySelf = getMySelf(); |
| | | // return mySelf.getFamilyId(); |
| | | // } |
| | | // |
| | | // private String listSecondFamilyIds() { |
| | | // ZInfoUser mySelf = getMySelf(); |
| | | // return mySelf.getSecondFamilyId(); |
| | | // } |
| | | |
| | | @Override |
| | | public List<ZfEquipment> selectByCondition(ZfEquipment zfEquipment) { |
| | | String familyIds = listFamilyIds(); |
| | | LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, familyIds); |
| | | List<ZfEquipment> list = list(lambdaQueryWrapper); |
| | | log.info("返回的数据为:{}", list); |
| | | return list; |
| | | // String familyIds = listFamilyIds(); |
| | | // String secondFamilyIds = listSecondFamilyIds(); |
| | | // LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, familyIds, secondFamilyIds); |
| | | // List<ZfEquipment> beanRecords = list(lambdaQueryWrapper); |
| | | // log.info("从数据库中查到的为:{}", beanRecords); |
| | | // return markOwnData(familyIds, beanRecords); |
| | | |
| | | //要查自己家庭的 |
| | | ZInfoUser myself = zInfoUserService.getMyself(); |
| | | Long familyId = myself.getFamilyId(); |
| | | //也要查别人授权的 |
| | | List<ZAuthority> authority = zAuthorityService.getAuthority(); |
| | | List<Long> idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST)).map(ZAuthority::getFid).collect(Collectors.toList()); |
| | | //加上自己家庭的id |
| | | idList.add(familyId); |
| | | |
| | | LambdaQueryWrapper<ZfEquipment> lambdaQueryWrapper = buildCondition(zfEquipment, idList); |
| | | List<ZfEquipment> beanRecords = list(lambdaQueryWrapper); |
| | | log.info("从数据库中查到的为:{}", beanRecords); |
| | | return markOwnData(familyId, beanRecords); |
| | | } |
| | | |
| | | @Override |
| | | public int addEquipment(ZfEquipment zfEquipment) { |
| | | String familyIds = listFamilyIds(); |
| | | boolean flag = false;//判断当前用户的id是否有权加入当前家庭id的对象 |
| | | if (familyIds.contains(",")) { |
| | | String[] familyList = familyIds.split(","); |
| | | for (String familyId : familyList) { |
| | | if (familyId.equals(zfEquipment.getFamilyId())) { |
| | | flag = true; |
| | | public List<ZfEquipment> selectByIds(Long[] ids) { |
| | | List<ZfEquipment> list = new ArrayList<>(); |
| | | if(ids.length!=0) |
| | | list = listByIds(Arrays.asList(ids)); |
| | | else |
| | | list = list(); |
| | | return list; } |
| | | |
| | | // private static List<ZfEquipment> markOwnData(String familyIds, List<ZfEquipment> 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()); |
| | | // } |
| | | |
| | | public List<ZfEquipment> markOwnData(Long familyId,List<ZfEquipment> beanRecords){ |
| | | return beanRecords.stream().peek( |
| | | bean -> { |
| | | if (bean.getFamilyId() == familyId) { |
| | | bean.setOwnData(1); |
| | | } else { |
| | | bean.setOwnData(0); |
| | | } |
| | | } |
| | | } |
| | | } else { |
| | | if (zfEquipment.getFamilyId().equals(familyIds)) { |
| | | flag = true; |
| | | } |
| | | } |
| | | if (flag) { |
| | | boolean save = save(zfEquipment); |
| | | return save ? 1 : 0; |
| | | } else { |
| | | throw new RuntimeException("你没有操作该家庭号为" + zfEquipment.getFamilyId() + "数据的权限"); |
| | | } |
| | | ).sorted((a,b)-> b.getOwnData()-a.getOwnData()).collect(Collectors.toList()); |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult selectDataList(ZfEquipment zfEquipment, Integer pageNum, Integer pageSize) { |
| | | // String familyIds = listFamilyIds(); |
| | | // String secondFamilyAuthority = listSecondFamilyIds(); |
| | | // LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, familyIds, secondFamilyAuthority); |
| | | // |
| | | // Page<ZfEquipment> zfEquipmentPage = new Page<>(pageNum, pageSize); |
| | | // Page<ZfEquipment> pageResult = page(zfEquipmentPage, lqw); |
| | | // |
| | | // List<ZfEquipment> beanRecords = pageResult.getRecords();//得到查询出来的数据 |
| | | // |
| | | // List<ZfEquipment> dtoResult = markOwnData(familyIds, beanRecords); |
| | | // |
| | | // HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult); |
| | | // return AjaxResult.success(data); |
| | | |
| | | //要查自己家庭的 |
| | | ZInfoUser myself = zInfoUserService.getMyself(); |
| | | if(myself==null) |
| | | { |
| | | // System.out.println("ssssss"); |
| | | return AjaxResult.success("您没加入到对应的家庭,请联系管理员"); |
| | | } |
| | | Long familyId = myself.getFamilyId(); |
| | | //也要查别人授权的 |
| | | List<ZAuthority> authority = zAuthorityService.getAuthority(); |
| | | List<Long> idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST)).map(ZAuthority::getFid).collect(Collectors.toList()); |
| | | //加上自己家庭的id |
| | | idList.add(familyId); |
| | | // String familyIds = listFamilyIds(); |
| | | // String secondFamilyAuthority = listSecondFamilyIds(); |
| | | LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, idList); |
| | | |
| | | |
| | | Page<ZfEquipment> zfEquipmentPage = new Page<>(pageNum, pageSize); |
| | | Page<ZfEquipment> pageResult = page(zfEquipmentPage, lqw); |
| | | |
| | | List<ZfEquipment> beanRecords = pageResult.getRecords();//得到查询出来的数据 |
| | | |
| | | List<ZfEquipment> dtoResult = markOwnData(familyId, beanRecords); |
| | | |
| | | HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult); |
| | | return AjaxResult.success(data); |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult addData(ZfEquipment zfEquipment){ |
| | | ZInfoUser myself = zInfoUserService.getMyself(); |
| | | Long familyId = myself.getFamilyId(); |
| | | |
| | | if(familyId == null){ |
| | | throw new RuntimeException("您还未加入任何家庭"); |
| | | } |
| | | |
| | | List<ZAuthority> authority = zAuthorityService.getAuthority(); |
| | | List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_ADD)).map(ZAuthority::getFid).collect(Collectors.toList()); |
| | | familyIdList.add(familyId); |
| | | |
| | | if (zfEquipment.getFamilyId()!=null && !familyIdList.contains(zfEquipment.getFamilyId())) { |
| | | throw new RuntimeException("你没有权限操作此家庭的数据"); |
| | | } |
| | | |
| | | |
| | | if(zfEquipment.getFamilyId() == null){ |
| | | //默认添加自己家庭的数据 |
| | | zfEquipment.setFamilyId(familyId); |
| | | } |
| | | |
| | | //判断是否有重复数据 |
| | | LambdaQueryWrapper<ZfEquipment> lqw = uniqueCondition(zfEquipment); |
| | | List<ZfEquipment> list = list(lqw); |
| | | |
| | | if(list.size()>0){ |
| | | throw new RuntimeException("请勿新增重复数据"); |
| | | } |
| | | |
| | | if (save(zfEquipment)) { |
| | | |
| | | EsModel esModel = new EsModel(); |
| | | Integer inte = zfEquipment.getId().intValue(); |
| | | String uuid = UUID.randomUUID().toString().replace("-",""); |
| | | esModel.setId(uuid); |
| | | esModel.setCtId(Long.valueOf(inte)); |
| | | esModel.setCtTableName("家庭设备"); |
| | | esModel.setBy1(zfEquipment.getName()); |
| | | esModel.setBy2(zfEquipment.getBuyer()); |
| | | esModel.setBy3(zfEquipment.getContent()); |
| | | esModel.setBy4(zfEquipment.getLocation()); |
| | | esModel.setBy5("/family/zfEquipment"); |
| | | esModel.setRemark(zfEquipment.getRemark()); |
| | | esModel.setFid(familyId); |
| | | //这里存储查询详情的路径 |
| | | esService.insertTable(esModel); |
| | | |
| | | return AjaxResult.success(); |
| | | } else { |
| | | return AjaxResult.error(); |
| | | } |
| | | |
| | | } |
| | | |
| | | // @Override |
| | | // public AjaxResult addData(ZfEquipment zfEquipment) { |
| | | // |
| | | // String familyIds = listFamilyIds(); |
| | | // boolean flag = false;//判断当前用户的id是否有权加入当前家庭id的对象 |
| | | // if (familyIds.contains(",")) { |
| | | // String[] familyList = familyIds.split(","); |
| | | // for (String familyId : familyList) { |
| | | // if (familyId.equals(zfEquipment.getFamilyId())) { |
| | | // flag = true; |
| | | // } |
| | | // } |
| | | // } else { |
| | | // if (zfEquipment.getFamilyId().equals(familyIds)) { |
| | | // flag = true; |
| | | // } |
| | | // } |
| | | // if (flag) { |
| | | // if (save(zfEquipment)) { |
| | | // return AjaxResult.success(); |
| | | // } else { |
| | | // return AjaxResult.error(); |
| | | // } |
| | | // |
| | | // } else { |
| | | // throw new RuntimeException("你没有操作该家庭号为" + zfEquipment.getFamilyId() + "数据的权限"); |
| | | // } |
| | | // } |
| | | |
| | | // @Override |
| | | // public AjaxResult addData2(ZfEquipment zfEquipment) { |
| | | // ZInfoUser mySelf = getMySelf(); |
| | | // String myFamilyId = mySelf.getFamilyId(); |
| | | // |
| | | // if (StringUtils.isNotEmpty(zfEquipment.getFamilyId())) { |
| | | // throw new RuntimeException("导入数据时您不能指定家庭号"); |
| | | // } |
| | | // |
| | | // if (StringUtils.isEmpty(myFamilyId)) { |
| | | // throw new RuntimeException("您还未加入任何家庭"); |
| | | // } |
| | | // |
| | | // if (myFamilyId.contains(",")) { |
| | | // String[] myFamilyIds = myFamilyId.split(","); |
| | | // //默认添加第一个家庭的数据 |
| | | // zfEquipment.setFamilyId(myFamilyIds[0]); |
| | | // } else { |
| | | // zfEquipment.setFamilyId(myFamilyId); |
| | | // } |
| | | // |
| | | // //判断是否有重复数据 |
| | | // LambdaQueryWrapper<ZfEquipment> lqw = uniqueCondition(zfEquipment); |
| | | // List<ZfEquipment> list = list(lqw); |
| | | // if(list.size()>0){ |
| | | // throw new RuntimeException("请勿新增重复数据"); |
| | | // } |
| | | // |
| | | // if (save(zfEquipment)) { |
| | | // return AjaxResult.success(); |
| | | // } else { |
| | | // return AjaxResult.error(); |
| | | // } |
| | | // } |
| | | |
| | | |
| | | @Override |
| | | @Transactional |
| | | public AjaxResult importExcel(MultipartFile file) { |
| | | |
| | | ExcelUtil<ZfEquipment> util = new ExcelUtil<>(ZfEquipment.class); |
| | | List<ZfEquipment> equipmentList = null; |
| | | List<ZfEquipment> propertyList = null; |
| | | try { |
| | | equipmentList = util.importExcel(file.getInputStream()); |
| | | propertyList = util.importExcel(file.getInputStream()); |
| | | } catch (Exception e) { |
| | | throw new RuntimeException(e); |
| | | } |
| | | log.info("资产列表为:{}", equipmentList); |
| | | log.info("资产列表为:{}", propertyList); |
| | | |
| | | for (ZfEquipment zfEquipment : equipmentList) { |
| | | zfEquipmentService.addEquipment(zfEquipment); |
| | | for (ZfEquipment zfEquipment : propertyList) { |
| | | zfEquipmentService.addData(zfEquipment); |
| | | } |
| | | |
| | | return AjaxResult.success("导入数据成功"); |
| | | |
| | | } |
| | | |
| | | private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment) { |
| | | LambdaQueryWrapper<ZfEquipment> lqw = new LambdaQueryWrapper<>(); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getName()), ZfEquipment::getName, zfEquipment.getName()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getBuyer()), ZfEquipment::getBuyer, zfEquipment.getBuyer()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getContent()), ZfEquipment::getContent, zfEquipment.getContent()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getLocation()), ZfEquipment::getLocation, zfEquipment.getLocation()); |
| | | lqw.like(StringUtils.isNotEmpty(zfEquipment.getRemark()), ZfEquipment::getRemark, zfEquipment.getRemark()); |
| | | lqw.like(zfEquipment.getCreateDate() != null, ZfEquipment::getCreateDate, zfEquipment.getCreateDate()); |
| | | return lqw; |
| | | } |
| | | @Resource |
| | | ZfLogService zfLogService; |
| | | |
| | | private LambdaQueryWrapper<ZfEquipment> buildCondition(ZfEquipment zfEquipment, String familyIds) { |
| | | LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment); |
| | | if (familyIds.contains(",")) { |
| | | String[] familyList = familyIds.split(","); |
| | | lqw.in(ZfEquipment::getFamilyId,familyList); |
| | | @Override |
| | | public AjaxResult updateData(ZfEquipment zfEquipment) { |
| | | |
| | | } else { |
| | | lqw.eq(ZfEquipment::getFamilyId, familyIds); |
| | | ZInfoUser myself = zInfoUserService.getMyself(); |
| | | Long familyId = myself.getFamilyId(); |
| | | |
| | | //先根据id查询出数据的familyId,看看有没有权限操作 |
| | | Long dataFamilyId = getById(zfEquipment.getId()).getFamilyId(); |
| | | |
| | | List<ZAuthority> authority = zAuthorityService.getAuthority(); |
| | | List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_UPDATE)).map(ZAuthority::getFid).collect(Collectors.toList()); |
| | | familyIdList.add(familyId); |
| | | |
| | | if (dataFamilyId!=null && !familyIdList.contains(dataFamilyId)) { |
| | | throw new RuntimeException("你没有权限操作此家庭的数据"); |
| | | } |
| | | return lqw; |
| | | } |
| | | |
| | | private String listFamilyIds() { |
| | | 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(); |
| | | //操作后加入日志 |
| | | ZfLog zfLog = new ZfLog(); |
| | | zfLog.setUpdateTime(LocalDateTime.now()); |
| | | zfLog.setModule("家庭设备"); |
| | | zfLog.setUpdater(zInfoUserService.getMyself().getNickName()); |
| | | zfLogService.save(zfLog); |
| | | |
| | | if(updateById(zfEquipment)){ |
| | | //到数据库中查询对应的数据 |
| | | ZfEquipment dataById = getById(zfEquipment.getId()); |
| | | |
| | | //先到es中查询到对应那条数据在es的id |
| | | EsModel esResult = esService.findByCtId(dataById.getId().intValue(), "家庭设备"); |
| | | |
| | | if (esResult == null){ |
| | | return AjaxResult.success(); |
| | | } |
| | | |
| | | //操作es修改数据 |
| | | EsModel newModel = new EsModel(); |
| | | if(zfEquipment.getName()!=null){ |
| | | newModel.setBy1(zfEquipment.getName()); |
| | | }else { |
| | | newModel.setBy1(dataById.getName()); |
| | | } |
| | | |
| | | if(zfEquipment.getBuyer()!=null){ |
| | | newModel.setBy2(zfEquipment.getBuyer()); |
| | | }else { |
| | | newModel.setBy2(dataById.getBuyer()); |
| | | } |
| | | |
| | | if(zfEquipment.getContent()!=null){ |
| | | newModel.setBy3(zfEquipment.getContent()); |
| | | }else { |
| | | newModel.setBy3(dataById.getContent()); |
| | | } |
| | | |
| | | if(zfEquipment.getLocation()!=null){ |
| | | newModel.setBy4(zfEquipment.getLocation()); |
| | | }else { |
| | | newModel.setBy4(dataById.getLocation()); |
| | | } |
| | | |
| | | if(zfEquipment.getRemark()!=null){ |
| | | newModel.setRemark(zfEquipment.getRemark()); |
| | | }else { |
| | | newModel.setRemark(dataById.getRemark()); |
| | | } |
| | | |
| | | |
| | | UpdateRequest updateRequest = new UpdateRequest("allsearchdata", esResult.getId()); |
| | | updateRequest.doc( |
| | | "by1",newModel.getBy1(), |
| | | "by2",newModel.getBy2(), |
| | | "by3",newModel.getBy3(), |
| | | "by4",newModel.getBy4(), |
| | | "remark",newModel.getRemark() |
| | | ); |
| | | |
| | | try { |
| | | restHighLevelClient.update(updateRequest, RequestOptions.DEFAULT); |
| | | } catch (IOException e) { |
| | | throw new RuntimeException(e); |
| | | } |
| | | return AjaxResult.success(); |
| | | }else { |
| | | return AjaxResult.error(); |
| | | } |
| | | |
| | | |
| | | } |
| | | |
| | | @Override |
| | | public AjaxResult selectEquipmentList(ZfEquipment zfEquipment, Integer pageNum, Integer pageSize) { |
| | | String familyIds = listFamilyIds(); |
| | | LambdaQueryWrapper<ZfEquipment> lqw = buildCondition(zfEquipment, familyIds); |
| | | Page<ZfEquipment> zfEquipmentPage = new Page<>(pageNum, pageSize); |
| | | Page<ZfEquipment> pageResult = page(zfEquipmentPage, lqw); |
| | | HashMap<String, Object> data = MapUtils.getResult(pageResult); |
| | | return AjaxResult.success(data); |
| | | public AjaxResult deleteData(Long[] ids) { |
| | | List<ZfEquipment> dataList = zfEquipmentService.listByIds(Arrays.asList(ids)); |
| | | |
| | | ZInfoUser myself = zInfoUserService.getMyself(); |
| | | Long familyId = myself.getFamilyId(); |
| | | |
| | | List<ZAuthority> authority = zAuthorityService.getAuthority(); |
| | | List<Long> familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(EQUIPMENT_LIST_REMOVE)).map(ZAuthority::getFid).collect(Collectors.toList()); |
| | | familyIdList.add(familyId); |
| | | |
| | | for (ZfEquipment data : dataList) { |
| | | if (!familyIdList.contains(data.getFamilyId())){ |
| | | throw new RuntimeException("你没有权限操作此家庭的数据"); |
| | | } |
| | | } |
| | | |
| | | List<ZfEquipment> zfEquipments = listByIds(Arrays.asList(ids)); |
| | | |
| | | if (zfEquipmentService.removeByIds(Arrays.asList(ids))) { |
| | | |
| | | //删除es中的数据 |
| | | zfEquipments.stream().forEach(zfEquipment -> { |
| | | EsModel esModel = esService.findByCtId(zfEquipment.getId().intValue(), "家庭设备"); |
| | | if (esModel != null){ |
| | | DeleteRequest deleteRequest = new DeleteRequest("allsearchdata", esModel.getId()); |
| | | try { |
| | | restHighLevelClient.delete(deleteRequest,RequestOptions.DEFAULT); |
| | | } catch (IOException e) { |
| | | throw new RuntimeException(e); |
| | | } |
| | | } |
| | | }); |
| | | return AjaxResult.success(); |
| | | }else { |
| | | return AjaxResult.error(); |
| | | } |
| | | |
| | | } |
| | | |
| | | |
| | | } |