package com.ruoyi.service.impl; import com.alibaba.fastjson2.JSONObject; 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.fasterxml.jackson.core.JsonParser; 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.ZfCollection; import com.ruoyi.domain.ZfCollection; import com.ruoyi.mapper.ZfCollectionMapper; import com.ruoyi.service.ZAuthorityService; import com.ruoyi.service.ZInfoUserService; import com.ruoyi.service.ZfCollectionService; 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.*; import java.util.stream.Collectors; import static com.ruoyi.constant.MenuAuthority.*; /** *

* 服务实现类 *

* * @author ojq * @since 2023-03-12 */ @Service @Slf4j public class ZfCollectionServiceImpl extends ServiceImpl implements ZfCollectionService { @Resource ZfCollectionService zfCollectionService; @Resource ZInfoUserService zInfoUserService; @Resource ZAuthorityService zAuthorityService; @Resource ZfCollectionMapper zfCollectionMapper; private LambdaQueryWrapper buildCondition(ZfCollection zfCollection,List familyIdList) { LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.orderByDesc(ZfCollection::getCreateTime); lqw.like(StringUtils.isNotEmpty(zfCollection.getType()),ZfCollection::getType,zfCollection.getType()); lqw.like(StringUtils.isNotEmpty(zfCollection.getName()),ZfCollection::getName,zfCollection.getName()); lqw.like(StringUtils.isNotEmpty(zfCollection.getSource()),ZfCollection::getSource,zfCollection.getSource()); lqw.like(StringUtils.isNotEmpty(zfCollection.getOwner()),ZfCollection::getOwner,zfCollection.getOwner()); lqw.like(StringUtils.isNotEmpty(zfCollection.getPrice()),ZfCollection::getPrice,zfCollection.getPrice()); lqw.like(StringUtils.isNotEmpty(zfCollection.getLocation()),ZfCollection::getLocation,zfCollection.getLocation()); lqw.like(StringUtils.isNotEmpty(zfCollection.getRemark()),ZfCollection::getRemark,zfCollection.getRemark()); lqw.in(ZfCollection::getFamilyId,familyIdList); lqw.eq(zfCollection.getHappenTime() != null, ZfCollection::getHappenTime, zfCollection.getHappenTime()); lqw.between(zfCollection.getHappenStartTime() != null && zfCollection.getHappenEndTime() != null, ZfCollection::getHappenTime, zfCollection.getHappenStartTime(), zfCollection.getHappenEndTime()); return lqw; } private LambdaQueryWrapper uniqueCondition(ZfCollection zfCollection) { LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(StringUtils.isNotEmpty(zfCollection.getType()),ZfCollection::getType,zfCollection.getType()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getName()),ZfCollection::getName,zfCollection.getName()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getSource()),ZfCollection::getSource,zfCollection.getSource()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getOwner()),ZfCollection::getOwner,zfCollection.getOwner()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getPrice()),ZfCollection::getPrice,zfCollection.getPrice()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getLocation()),ZfCollection::getLocation,zfCollection.getLocation()); lqw.eq(StringUtils.isNotEmpty(zfCollection.getRemark()),ZfCollection::getRemark,zfCollection.getRemark()); lqw.eq(zfCollection.getHappenTime() != null, ZfCollection::getHappenTime, zfCollection.getHappenTime()); lqw.eq(zfCollection.getFamilyId()!=null,ZfCollection::getFamilyId,zfCollection.getFamilyId()); return lqw; } // private LambdaQueryWrapper buildCondition(ZfCollection zfCollection, String familyIds, String secondFamilyAuthority) { // //模糊查询的条件 // LambdaQueryWrapper lqw = buildCondition(zfCollection); // // //定义一个布尔标记,标记第一家庭号是否有逗号 // 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(ZfCollection::getFamilyId, totalFamilyIdList); // } else { // //如果第一家庭号没有逗号但是第二家庭号有逗号 // String[] totalFamilyIdList = ArraysUtil.appendToArray(secondFamilyIdList, familyIds); // lqw.in(ZfCollection::getFamilyId, totalFamilyIdList); // } // // } else { // if (flag) { // //如果第一家庭号有逗号但第二家庭号没有逗号 // String[] familyList = familyIds.split(","); // String[] totalFamilyIdList = ArraysUtil.appendToArray(familyList, secondFamilyIds); // lqw.in(ZfCollection::getFamilyId, totalFamilyIdList); // } else { // //如果第一家庭号和第二家庭号都没有逗号 // String[] totalFamilyIdList = ArraysUtil.appendToArray(familyIds, secondFamilyIds); // lqw.in(ZfCollection::getFamilyId, totalFamilyIdList); // } // } // } else { // //如果没有第二家庭号 // if (flag) { // String[] familyList = familyIds.split(","); // lqw.in(ZfCollection::getFamilyId, familyList); // } else { // lqw.eq(ZfCollection::getFamilyId, familyIds); // } // // } // // lqw.orderByDesc(ZfCollection::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(COLLECTION_LIST)) { // secondFamilyIdsBuffer.append(authorityAndId.charAt(0)).append(","); // break; // } // } // } else { // if (authorities.equals(COLLECTION_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(ZfCollection zfCollection) { // String familyIds = listFamilyIds(); // String secondFamilyIds = listSecondFamilyIds(); // LambdaQueryWrapper lambdaQueryWrapper = buildCondition(zfCollection, familyIds, secondFamilyIds); // List beanRecords = list(lambdaQueryWrapper); // log.info("从数据库中查到的为:{}", beanRecords); // return markOwnData(familyIds, beanRecords); //要查自己家庭的 ZInfoUser myself = zInfoUserService.getMyself(); Long familyId = myself.getFamilyId(); //也要查别人授权的 List authority = zAuthorityService.getAuthority(); List idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(COLLECTION_LIST)).map(ZAuthority::getFid).collect(Collectors.toList()); //加上自己家庭的id idList.add(familyId); LambdaQueryWrapper lambdaQueryWrapper = buildCondition(zfCollection, idList); List beanRecords = list(lambdaQueryWrapper); log.info("从数据库中查到的为:{}", beanRecords); return markOwnData(familyId, 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 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()); // } public List markOwnData(Long familyId,List beanRecords){ return beanRecords.stream().peek( bean -> { if (bean.getFamilyId() == familyId) { bean.setOwnData(1); } else { bean.setOwnData(0); } } ).sorted((a,b)-> b.getOwnData()-a.getOwnData()).collect(Collectors.toList()); } @Override public AjaxResult selectDataList(ZfCollection zfCollection, Integer pageNum, Integer pageSize) { // String familyIds = listFamilyIds(); // String secondFamilyAuthority = listSecondFamilyIds(); // LambdaQueryWrapper lqw = buildCondition(zfCollection, familyIds, secondFamilyAuthority); // // Page zfCollectionPage = new Page<>(pageNum, pageSize); // Page pageResult = page(zfCollectionPage, lqw); // // List beanRecords = pageResult.getRecords();//得到查询出来的数据 // // List dtoResult = markOwnData(familyIds, beanRecords); // // HashMap data = MapUtils.getResult(pageResult, dtoResult); // return AjaxResult.success(data); //要查自己家庭的 ZInfoUser myself = zInfoUserService.getMyself(); Long familyId = myself.getFamilyId(); //也要查别人授权的 List authority = zAuthorityService.getAuthority(); List idList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(COLLECTION_LIST)).map(ZAuthority::getFid).collect(Collectors.toList()); //加上自己家庭的id idList.add(familyId); // String familyIds = listFamilyIds(); // String secondFamilyAuthority = listSecondFamilyIds(); LambdaQueryWrapper lqw = buildCondition(zfCollection, idList); Page zfCollectionPage = new Page<>(pageNum, pageSize); Page pageResult = page(zfCollectionPage, lqw); List beanRecords = pageResult.getRecords();//得到查询出来的数据 List dtoResult = markOwnData(familyId, beanRecords); HashMap data = MapUtils.getResult(pageResult, dtoResult); return AjaxResult.success(data); } // @Override // public AjaxResult addData2(ZfCollection zfCollection){ // ZInfoUser mySelf = getMySelf(); // String myFamilyId = mySelf.getFamilyId(); // // if (StringUtils.isNotEmpty(zfCollection.getFamilyId())){ // throw new RuntimeException("导入数据时您不能指定家庭号"); // } // // if(StringUtils.isEmpty(myFamilyId)){ // throw new RuntimeException("您还未加入任何家庭"); // } // // if (myFamilyId.contains(",")){ // String[] myFamilyIds = myFamilyId.split(","); // //默认添加第一个家庭的数据 // zfCollection.setFamilyId(myFamilyIds[0]); // }else { // zfCollection.setFamilyId(myFamilyId); // } // // //判断是否有重复数据 // LambdaQueryWrapper lqw = uniqueCondition(zfCollection); // List list = list(lqw); // if(list.size()>0){ // throw new RuntimeException("请勿新增重复数据"); // } // // if (save(zfCollection)) { // return AjaxResult.success(); // }else { // return AjaxResult.error(); // } // } @Override public AjaxResult addData(ZfCollection zfCollection) { ZInfoUser myself = zInfoUserService.getMyself(); Long familyId = myself.getFamilyId(); if(familyId == null){ throw new RuntimeException("您还未加入任何家庭"); } List authority = zAuthorityService.getAuthority(); List familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(COLLECTION_LIST_ADD)).map(ZAuthority::getFid).collect(Collectors.toList()); familyIdList.add(familyId); if (zfCollection.getFamilyId()!=null && !familyIdList.contains(zfCollection.getFamilyId())) { throw new RuntimeException("你没有权限操作此家庭的数据"); } if(zfCollection.getFamilyId() == null){ //默认添加自己家庭的数据 zfCollection.setFamilyId(familyId); } //判断是否有重复数据 LambdaQueryWrapper lqw = uniqueCondition(zfCollection); List list = list(lqw); if(list.size()>0){ throw new RuntimeException("请勿新增重复数据"); } if (save(zfCollection)) { return AjaxResult.success(); } else { return AjaxResult.error(); } } @Override public AjaxResult updateData(ZfCollection zfCollection) { ZInfoUser myself = zInfoUserService.getMyself(); Long familyId = myself.getFamilyId(); //先根据id查询出数据的familyId,看看有没有权限操作 Long dataFamilyId = getById(zfCollection.getId()).getFamilyId(); List authority = zAuthorityService.getAuthority(); List familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(COLLECTION_LIST_UPDATE)).map(ZAuthority::getFid).collect(Collectors.toList()); familyIdList.add(familyId); if (dataFamilyId!=null && !familyIdList.contains(dataFamilyId)) { throw new RuntimeException("你没有权限操作此家庭的数据"); } //先记录下谁更新、何时更新、更新了哪里 LambdaQueryWrapper lqw = new LambdaQueryWrapper<>(); lqw.eq(ZfCollection::getId,zfCollection.getId()); zfCollectionService.getOne(lqw); ZfCollection oldData = getOne(lqw); String oldJson = JSONObject.toJSONString(oldData); String newJson = JSONObject.toJSONString(zfCollection); Map oldMap = JSONObject.parseObject(oldJson, Map.class); Map newMap = JSONObject.parseObject(newJson, Map.class); for (Object key : oldMap.keySet()) { if(newMap.get(key)!=null && !newMap.get(key).equals(oldMap.get(key))){ System.out.println(newMap.get(key)+" "+oldMap.get(key)); Date date = new Date(); System.out.println("用户"+myself.getNickName()+"在"+date+"这个时间"+"修改了<收藏荣誉>模块的"+key+"字段"); } } if(updateById(zfCollection)){ return AjaxResult.success(); }else { return AjaxResult.error(); } } @Override public AjaxResult deleteData(Long[] ids) { List dataList = zfCollectionService.listByIds(Arrays.asList(ids)); ZInfoUser myself = zInfoUserService.getMyself(); Long familyId = myself.getFamilyId(); List authority = zAuthorityService.getAuthority(); List familyIdList = authority.stream().filter(auth -> auth.getAuthority().toString().equals(COLLECTION_LIST_REMOVE)).map(ZAuthority::getFid).collect(Collectors.toList()); familyIdList.add(familyId); for (ZfCollection data : dataList) { if (!familyIdList.contains(data.getFamilyId())){ throw new RuntimeException("你没有权限操作此家庭的数据"); } } if (zfCollectionService.removeByIds(Arrays.asList(ids))) { return AjaxResult.success(); }else { return AjaxResult.error(); } } @Override public AjaxResult listType() { List result = zfCollectionMapper.listType(); return AjaxResult.success(result); } @Override @Transactional public AjaxResult importExcel(MultipartFile file) { ExcelUtil util = new ExcelUtil<>(ZfCollection.class); List dataList = null; try { dataList = util.importExcel(file.getInputStream()); } catch (Exception e) { throw new RuntimeException(e); } log.info("准备要导入的数据列表为:{}", dataList); for (ZfCollection zfCollection : dataList) { zfCollectionService.addData(zfCollection); } return AjaxResult.success("导入数据成功"); } }