From 9de59eeae9728b9085c6e964a87ba54d2c85c2e7 Mon Sep 17 00:00:00 2001 From: xuhy <3313886187@qq.com> Date: 星期一, 16 十二月 2024 17:43:50 +0800 Subject: [PATCH] 用户端小程序 --- manage/src/main/java/com/jilongda/manage/service/impl/TModelServiceImpl.java | 55 ++++++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 42 insertions(+), 13 deletions(-) diff --git a/manage/src/main/java/com/jilongda/manage/service/impl/TModelServiceImpl.java b/manage/src/main/java/com/jilongda/manage/service/impl/TModelServiceImpl.java index d0e2ee2..5219d73 100644 --- a/manage/src/main/java/com/jilongda/manage/service/impl/TModelServiceImpl.java +++ b/manage/src/main/java/com/jilongda/manage/service/impl/TModelServiceImpl.java @@ -3,8 +3,10 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.jilongda.common.basic.PageInfo; import com.jilongda.manage.mapper.TBrandMapper; +import com.jilongda.manage.mapper.TMaterialMapper; import com.jilongda.manage.mapper.TSupplierMapper; import com.jilongda.manage.model.TBrand; +import com.jilongda.manage.model.TMaterial; import com.jilongda.manage.model.TModel; import com.jilongda.manage.mapper.TModelMapper; import com.jilongda.manage.model.TSupplier; @@ -15,7 +17,9 @@ import com.jilongda.manage.vo.TModelVO; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; +import org.springframework.util.CollectionUtils; +import java.util.ArrayList; import java.util.List; import java.util.Objects; import java.util.stream.Collectors; @@ -35,6 +39,8 @@ private TBrandMapper brandMapper; @Autowired private TSupplierMapper supplierMapper; + @Autowired + private TMaterialMapper materialMapper; @Override public Boolean upAndDown(String name, Integer status) { @@ -61,20 +67,43 @@ public PageInfo<TModelVO> pageList(TModelQuery query) { PageInfo<TModelVO> pageInfo = new PageInfo<>(query.getPageNum(), query.getPageSize()); List<TModelVO> list = this.baseMapper.pageList(query,pageInfo); - List<Integer> brandIds = list.stream().map(TModel::getBrandId).collect(Collectors.toList()); - List<Integer> supplierIds= list.stream().map(TModel::getSupplierId).collect(Collectors.toList()); - List<TBrand> tBrands = brandMapper.selectList(Wrappers.lambdaQuery(TBrand.class) - .in(TBrand::getId, brandIds)); - List<TSupplier> tSuppliers = supplierMapper.selectList(Wrappers.lambdaQuery(TSupplier.class) - .in(TSupplier::getId, supplierIds)); - for (TModelVO modelVO : list) { - TBrand tBrand = tBrands.stream().filter(brand -> brand.getId().equals(modelVO.getBrandId())).findFirst().orElse(null); - if(Objects.nonNull(tBrand)){ - modelVO.setBrandName(tBrand.getName()); + if(!CollectionUtils.isEmpty(list)){ + List<Integer> brandIds = list.stream().map(TModel::getBrandId).collect(Collectors.toList()); + List<Integer> supplierIds= list.stream().map(TModel::getSupplierId).collect(Collectors.toList()); + List<Integer> materialIds= list.stream().map(TModel::getMaterialId).collect(Collectors.toList()); + List<TBrand> tBrands = new ArrayList<>(); + if(!CollectionUtils.isEmpty(brandIds)){ + tBrands = brandMapper.selectList(Wrappers.lambdaQuery(TBrand.class) + .in(TBrand::getId, brandIds)); } - TSupplier tSupplier = tSuppliers.stream().filter(supplier -> supplier.getId().equals(modelVO.getSupplierId())).findFirst().orElse(null); - if(Objects.nonNull(tSupplier)){ - modelVO.setSupplierName(tSupplier.getName()); + List<TSupplier> tSuppliers = new ArrayList<>(); + if(!CollectionUtils.isEmpty(supplierIds)){ + tSuppliers = supplierMapper.selectList(Wrappers.lambdaQuery(TSupplier.class) + .in(TSupplier::getId, supplierIds)); + } + List<TMaterial> materials = new ArrayList<>(); + if(!CollectionUtils.isEmpty(materialIds)){ + materials = materialMapper.selectList(Wrappers.lambdaQuery(TMaterial.class) + .in(TMaterial::getId, materialIds)); + } + for (TModelVO modelVO : list) { + TBrand tBrand = tBrands.stream().filter(brand -> brand.getId().equals(modelVO.getBrandId())).findFirst().orElse(null); + if(Objects.nonNull(tBrand)){ + modelVO.setBrandName(tBrand.getName()); + } + TSupplier tSupplier = tSuppliers.stream().filter(supplier -> supplier.getId().equals(modelVO.getSupplierId())).findFirst().orElse(null); + if(Objects.nonNull(tSupplier)){ + modelVO.setSupplierName(tSupplier.getName()); + } + TMaterial tMaterial = materials.stream().filter(material -> material.getId().equals(modelVO.getMaterialId())).findFirst().orElse(null); + if(Objects.nonNull(tMaterial)){ + modelVO.setMaterialName(tMaterial.getName()); + } + // 查询型号的色号 + List<TModel> colorList = this.list(Wrappers.lambdaQuery(TModel.class) + .eq(TModel::getName,modelVO.getName())); + String color = colorList.stream().map(TModel::getColor).collect(Collectors.joining(",")); + modelVO.setColor(color); } } pageInfo.setRecords(list); -- Gitblit v1.7.1