|
|
@ -6,12 +6,14 @@ import com.qs.serve.common.model.dto.PageVo; |
|
|
|
import com.qs.serve.common.model.dto.R; |
|
|
|
import com.qs.serve.common.model.enums.BizType; |
|
|
|
import com.qs.serve.common.model.enums.SystemModule; |
|
|
|
import com.qs.serve.common.util.CopierUtil; |
|
|
|
import com.qs.serve.common.util.PageUtil; |
|
|
|
import com.qs.serve.common.util.TreeUtil; |
|
|
|
import com.qs.serve.common.util.*; |
|
|
|
import com.qs.serve.modules.bms.entity.BmsErpInfo; |
|
|
|
import com.qs.serve.modules.bms.entity.BmsRegion; |
|
|
|
import com.qs.serve.modules.bms.entity.bo.BmsSupplierBatchBo; |
|
|
|
import com.qs.serve.modules.bms.entity.bo.BmsSupplierBo; |
|
|
|
import com.qs.serve.modules.bms.entity.vo.BmsSupplierTreeVo; |
|
|
|
import com.qs.serve.modules.bms.service.BmsErpInfoService; |
|
|
|
import com.qs.serve.modules.bms.service.BmsRegionService; |
|
|
|
import lombok.AllArgsConstructor; |
|
|
|
import lombok.extern.slf4j.Slf4j; |
|
|
|
import org.springframework.security.access.prepost.PreAuthorize; |
|
|
@ -36,6 +38,8 @@ import java.util.stream.Collectors; |
|
|
|
public class BmsSupplierController { |
|
|
|
|
|
|
|
private BmsSupplierService bmsSupplierService; |
|
|
|
private BmsRegionService bmsRegionService; |
|
|
|
private BmsErpInfoService bmsErpInfoService; |
|
|
|
|
|
|
|
/** |
|
|
|
* 翻页查询 |
|
|
@ -45,12 +49,25 @@ public class BmsSupplierController { |
|
|
|
@GetMapping("/page") |
|
|
|
@PreAuthorize("hasRole('bms:supplier:query')") |
|
|
|
public R<PageVo<BmsSupplier>> getPage(BmsSupplier param){ |
|
|
|
BmsRegion bmsRegion = bmsRegionService.getById(param.getRegionId()); |
|
|
|
if(bmsRegion.getLevel()==1){ |
|
|
|
param.setRegionFirst(param.getRegionId()); |
|
|
|
} |
|
|
|
if(bmsRegion.getLevel()==2){ |
|
|
|
param.setRegionSecond(param.getRegionId()); |
|
|
|
} |
|
|
|
if(bmsRegion.getLevel()==3){ |
|
|
|
param.setRegionThird(param.getRegionId()); |
|
|
|
} |
|
|
|
PageUtil.startPage(); |
|
|
|
LambdaQueryWrapper<BmsSupplier> supplierWrapper = new LambdaQueryWrapper<>(param); |
|
|
|
List<BmsSupplier> list = bmsSupplierService.list(supplierWrapper); |
|
|
|
initSupplierList(list); |
|
|
|
return R.byPageHelperList(list); |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
* 树查询 |
|
|
|
* @param param |
|
|
@ -61,6 +78,23 @@ public class BmsSupplierController { |
|
|
|
public R<List<BmsSupplierTreeVo>> getTree(BmsSupplier param){ |
|
|
|
LambdaQueryWrapper<BmsSupplier> supplierWrapper = new LambdaQueryWrapper<>(param); |
|
|
|
List<BmsSupplier> list = bmsSupplierService.list(supplierWrapper); |
|
|
|
initSupplierList(list); |
|
|
|
List<BmsErpInfo> erpInfos = bmsErpInfoService.list(); |
|
|
|
for (BmsSupplier supplier : list) { |
|
|
|
String regionId = supplier.getRegionLast(); |
|
|
|
if(regionId!=null&&!regionId.equals("0")){ |
|
|
|
BmsRegion bmsRegion2 = bmsRegionService.getById(regionId); |
|
|
|
supplier.setRegionLabel(bmsRegion2.getName()); |
|
|
|
} |
|
|
|
if(StringUtils.hasText(supplier.getBelong())){ |
|
|
|
for (BmsErpInfo erpInfo : erpInfos) { |
|
|
|
if(supplier.getBelong().equals(erpInfo.getCode())){ |
|
|
|
supplier.setBelongLabel(erpInfo.getName()); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
List<BmsSupplierTreeVo> list2 = list.stream().map(supplier->{ |
|
|
|
BmsSupplierTreeVo treeNode = CopierUtil.copy(supplier,new BmsSupplierTreeVo()); |
|
|
|
treeNode.setId(supplier.getId()); |
|
|
@ -85,8 +119,6 @@ public class BmsSupplierController { |
|
|
|
return R.ok(bmsSupplier); |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/** |
|
|
|
* 根据ID更新 |
|
|
|
* @param param |
|
|
@ -145,7 +177,39 @@ public class BmsSupplierController { |
|
|
|
return R.ok(); |
|
|
|
} |
|
|
|
|
|
|
|
//todo 迁移
|
|
|
|
/** |
|
|
|
* 显示关联 |
|
|
|
* @param list |
|
|
|
*/ |
|
|
|
private void initSupplierList(List<BmsSupplier> list){ |
|
|
|
List<String> regionIds = list.stream() |
|
|
|
.filter(sup->sup.getRegionLast()!=null&&!sup.getRegionLast().equals("0")) |
|
|
|
.map(BmsSupplier::getRegionLast).collect(Collectors.toList()); |
|
|
|
List<BmsRegion> regionList = null; |
|
|
|
if(CollectionUtil.isNotEmpty(regionIds)){ |
|
|
|
regionList = bmsRegionService.listByIds(regionIds); |
|
|
|
} |
|
|
|
List<BmsErpInfo> erpInfos = bmsErpInfoService.list(); |
|
|
|
for (BmsSupplier supplier : list) { |
|
|
|
String regionId = supplier.getRegionLast(); |
|
|
|
if(regionId!=null&&CollectionUtil.isNotEmpty(regionList)){ |
|
|
|
for (BmsRegion region : regionList) { |
|
|
|
if(region.getId().equals(regionId)){ |
|
|
|
supplier.setRegionLabel(region.getName()); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
if(StringUtils.hasText(supplier.getBelong())){ |
|
|
|
for (BmsErpInfo erpInfo : erpInfos) { |
|
|
|
if(supplier.getBelong().equals(erpInfo.getCode())){ |
|
|
|
supplier.setBelongLabel(erpInfo.getName()); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|