package com.ruoyi.service.impl; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; 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.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.*; import com.ruoyi.domain.ZfEquipment; import com.ruoyi.mapper.ZfEquipmentMapper; import com.ruoyi.service.ZInfoUserService; import com.ruoyi.service.ZfEquipmentService; 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.ArrayList; import java.util.Arrays; import java.util.HashMap; import java.util.List; import java.util.stream.Collectors; import static com.ruoyi.constant.MenuAuthority.EQUIPMENT_LIST; import static com.ruoyi.constant.MenuAuthority.EVENT_LIST; /** *

* 服务实现类 *

* * @author ojq * @since 2023-03-12 */ @Service @Slf4j public class ZfEquipmentServiceImpl extends ServiceImpl implements ZfEquipmentService { @Resource ZInfoUserService zInfoUserService; @Resource ZfEquipmentService zfEquipmentService; private LambdaQueryWrapper buildCondition(ZfEquipment zfEquipment) { LambdaQueryWrapper 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.eq(zfEquipment.getHappenTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenTime()); lqw.between(zfEquipment.getHappenStartTime() != null && zfEquipment.getHappenEndTime() != null, ZfEquipment::getHappenTime, zfEquipment.getHappenStartTime(), zfEquipment.getHappenEndTime()); return lqw; } private LambdaQueryWrapper uniqueCondition(ZfEquipment zfEquipment) { LambdaQueryWrapper 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 buildCondition(ZfEquipment zfEquipment, String familyIds, String secondFamilyAuthority) { //模糊查询的条件 LambdaQueryWrapper 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 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 selectByCondition(ZfEquipment zfEquipment) { String familyIds = listFamilyIds(); String secondFamilyIds = listSecondFamilyIds(); LambdaQueryWrapper lambdaQueryWrapper = buildCondition(zfEquipment, familyIds, secondFamilyIds); List beanRecords = list(lambdaQueryWrapper); log.info("从数据库中查到的为:{}", beanRecords); return markOwnData(familyIds, beanRecords); } @Override public List selectByIds(Long[] ids) { List list = new ArrayList<>(); if(ids.length!=0) list = listByIds(Arrays.asList(ids)); else list = list(); return list; } private static List markOwnData(String familyIds, List 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 AjaxResult selectDataList(ZfEquipment zfEquipment, Integer pageNum, Integer pageSize) { String familyIds = listFamilyIds(); String secondFamilyAuthority = listSecondFamilyIds(); LambdaQueryWrapper lqw = buildCondition(zfEquipment, familyIds, secondFamilyAuthority); Page zfEquipmentPage = new Page<>(pageNum, pageSize); Page pageResult = page(zfEquipmentPage, lqw); List beanRecords = pageResult.getRecords();//得到查询出来的数据 List dtoResult = markOwnData(familyIds, beanRecords); HashMap data = MapUtils.getResult(pageResult, dtoResult); return AjaxResult.success(data); } @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 lqw = uniqueCondition(zfEquipment); List 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 util = new ExcelUtil<>(ZfEquipment.class); List propertyList = null; try { propertyList = util.importExcel(file.getInputStream()); } catch (Exception e) { throw new RuntimeException(e); } log.info("资产列表为：{}", propertyList); for (ZfEquipment zfEquipment : propertyList) { zfEquipmentService.addData2(zfEquipment); } return AjaxResult.success("导入数据成功"); } }