From 8e3f58c38fd3d552125ada6afdf88e7fc2b380a0 Mon Sep 17 00:00:00 2001 From: Jinquan_Ou <Jinquan@gdut.com> Date: 星期一, 17 四月 2023 12:47:32 +0800 Subject: [PATCH] 888 --- zhang-content/src/main/java/com/ruoyi/service/impl/IZfPropertyServiceImpl.java | 109 +++++++++++++++++++++++++++++++++++++----------------- 1 files changed, 74 insertions(+), 35 deletions(-) diff --git a/zhang-content/src/main/java/com/ruoyi/service/impl/IZfPropertyServiceImpl.java b/zhang-content/src/main/java/com/ruoyi/service/impl/IZfPropertyServiceImpl.java index 92428e1..d1c5af7 100644 --- a/zhang-content/src/main/java/com/ruoyi/service/impl/IZfPropertyServiceImpl.java +++ b/zhang-content/src/main/java/com/ruoyi/service/impl/IZfPropertyServiceImpl.java @@ -12,13 +12,13 @@ import com.ruoyi.common.utils.poi.ExcelUtil; import com.ruoyi.domain.ZInfoUser; import com.ruoyi.domain.ZfProperty; -import com.ruoyi.domain.dto.ZfPropertyDto; import com.ruoyi.mapper.ZfPropertyMapper; import com.ruoyi.service.IZfPropertyService; import com.ruoyi.service.ZInfoUserService; 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; @@ -38,21 +38,15 @@ @Service public class IZfPropertyServiceImpl extends ServiceImpl<ZfPropertyMapper, ZfProperty> implements IZfPropertyService { - @Resource - ZInfoUserService zInfoUserService; + @Resource IZfPropertyService zfPropertyService; - @Override - public List<ZfProperty> selectByCondition(ZfProperty zfProperty) { - String familyIds = listFamilyIds(); - String secondFamilyIds = listSecondFamilyIds(); - LambdaQueryWrapper<ZfProperty> lambdaQueryWrapper = buildCondition(zfProperty, familyIds, secondFamilyIds); - List<ZfProperty> list = list(lambdaQueryWrapper); - log.info("杩斿洖鐨勬暟鎹负:{}", list); - return list; - } + @Resource + ZInfoUserService zInfoUserService; + + private LambdaQueryWrapper<ZfProperty> buildCondition(ZfProperty zfProperty) { LambdaQueryWrapper<ZfProperty> lqw = new LambdaQueryWrapper<>(); @@ -189,6 +183,34 @@ } @Override + public List<ZfProperty> selectByCondition(ZfProperty zfProperty) { + String familyIds = listFamilyIds(); + String secondFamilyIds = listSecondFamilyIds(); + LambdaQueryWrapper<ZfProperty> lambdaQueryWrapper = buildCondition(zfProperty, familyIds, secondFamilyIds); + List<ZfProperty> beanRecords = list(lambdaQueryWrapper); + log.info("浠庢暟鎹簱涓煡鍒扮殑涓�:{}", beanRecords); + return markOwnData(familyIds, beanRecords); + } + + private static List<ZfProperty> markOwnData(String familyIds, List<ZfProperty> 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 selectPropertyList(ZfProperty zfProperty, Integer pageNum, Integer pageSize) { String familyIds = listFamilyIds(); String secondFamilyAuthority = listSecondFamilyIds(); @@ -199,32 +221,14 @@ List<ZfProperty> beanRecords = pageResult.getRecords();//寰楀埌鏌ヨ鍑烘潵鐨勬暟鎹� - List<ZfPropertyDto> dtoResult = beanRecords.stream().map(record -> { - String recordFamilyId = record.getFamilyId(); - ZfPropertyDto zfPropertyDto = new ZfPropertyDto(); - BeanUtils.copyProperties(record, zfPropertyDto); - - if (familyIds.contains(",")) { - String[] familyIdList = familyIds.split(","); - for (String familyId : familyIdList) { - if (familyId.equals(recordFamilyId)) { - zfPropertyDto.setOwnData(1); - } - } - } else { - if (familyIds.equals(recordFamilyId)) { - zfPropertyDto.setOwnData(1); - } - } - return zfPropertyDto; - }).collect(Collectors.toList()); + List<ZfProperty> dtoResult = markOwnData(familyIds, beanRecords); HashMap<String, Object> data = MapUtils.getResult(pageResult, dtoResult); return AjaxResult.success(data); } @Override - public int addProperty(ZfProperty zfProperty) { + public AjaxResult addData(ZfProperty zfProperty) { String familyIds = listFamilyIds(); boolean flag = false;//鍒ゆ柇褰撳墠鐢ㄦ埛鐨刬d鏄惁鏈夋潈鍔犲叆褰撳墠瀹跺涵id鐨勫璞� @@ -241,14 +245,49 @@ } } if (flag) { - boolean save = save(zfProperty); - return save ? 1 : 0; + if (save(zfProperty)) { + return AjaxResult.success(); + }else { + return AjaxResult.error(); + } + } else { throw new RuntimeException("浣犳病鏈夋搷浣滆瀹跺涵鍙蜂负" + zfProperty.getFamilyId() + "鏁版嵁鐨勬潈闄�"); } } @Override + public AjaxResult addData2(ZfProperty zfProperty){ + ZInfoUser mySelf = getMySelf(); + String myFamilyId = mySelf.getFamilyId(); + + if (StringUtils.isNotEmpty(zfProperty.getFamilyId())){ + throw new RuntimeException("瀵煎叆鏁版嵁鏃舵偍涓嶈兘鎸囧畾瀹跺涵鍙�"); + } + + if(StringUtils.isEmpty(myFamilyId)){ + throw new RuntimeException("鎮ㄨ繕鏈姞鍏ヤ换浣曞搴�"); + } + + if (myFamilyId.contains(",")){ + String[] myFamilyIds = myFamilyId.split(","); + //榛樿娣诲姞绗竴涓搴殑鏁版嵁 + zfProperty.setFamilyId(myFamilyIds[0]); + }else { + zfProperty.setFamilyId(myFamilyId); + } + + if (save(zfProperty)) { + return AjaxResult.success(); + }else { + return AjaxResult.error(); + } + } + + + + @Override + @Transactional public AjaxResult importExcel(MultipartFile file) { ExcelUtil<ZfProperty> util = new ExcelUtil<>(ZfProperty.class); List<ZfProperty> propertyList = null; @@ -260,7 +299,7 @@ log.info("璧勪骇鍒楄〃涓猴細{}", propertyList); for (ZfProperty zfProperty : propertyList) { - zfPropertyService.addProperty(zfProperty); + zfPropertyService.addData2(zfProperty); } return AjaxResult.success("瀵煎叆鏁版嵁鎴愬姛"); -- Gitblit v1.9.1