|
|
@ -68,12 +68,12 @@ public class BmsCostCenterController { |
|
|
|
*/ |
|
|
|
@GetMapping("/tree") |
|
|
|
@PreAuthorize("hasRole('bms:costCenter:query')") |
|
|
|
public R<List<BmsCostCenterTreeVo>> getTree(Integer loadByCurrent,Integer type, String tartgetId){ |
|
|
|
public R<List<BmsCostCenterTreeVo>> getTree(Integer loadByCurrent,Integer type, String targetId){ |
|
|
|
if(loadByCurrent!=null&&loadByCurrent.equals(1)){ |
|
|
|
return getTreeByCurrentUser(); |
|
|
|
} |
|
|
|
if(loadByCurrent!=null&&loadByCurrent.equals(2)){ |
|
|
|
return getListByCustomerId(type,tartgetId); |
|
|
|
return getListByCustomerId(type,targetId); |
|
|
|
} |
|
|
|
List<BmsCostCenter> list = bmsCostCenterService.list(new LambdaQueryWrapper<BmsCostCenter>().orderByDesc(BmsCostCenter::getId)); |
|
|
|
List<BmsCostCenterTreeVo> treeVoList = list.stream().map(obj->{ |
|
|
@ -86,12 +86,15 @@ public class BmsCostCenterController { |
|
|
|
return R.ok(TreeUtil.buildByRecursive(treeVoList,TreeUtil.DEFAULT_PID_STRING)); |
|
|
|
} |
|
|
|
|
|
|
|
public R<List<BmsCostCenterTreeVo>> getListByCustomerId(Integer type, String tartgetId) { |
|
|
|
public R<List<BmsCostCenterTreeVo>> getListByCustomerId(Integer type, String targetId) { |
|
|
|
BirBudgetTargetSo query = new BirBudgetTargetSo(); |
|
|
|
query.setLoadType(type); |
|
|
|
query.setSelectCenterFlag(1); |
|
|
|
query.setTargetId(tartgetId); |
|
|
|
query.setTargetId(targetId); |
|
|
|
List<String> centerIds = birBudgetTargetService.getCenterByTargetId(query); |
|
|
|
if(centerIds.size()==0){ |
|
|
|
return R.ok(new ArrayList<BmsCostCenterTreeVo>()); |
|
|
|
} |
|
|
|
LambdaQueryWrapper<BmsCostCenter> centerLambdaQueryWrapper = new LambdaQueryWrapper<>(); |
|
|
|
centerLambdaQueryWrapper.in(BmsCostCenter::getId,centerIds); |
|
|
|
List<BmsCostCenter> costCenters = bmsCostCenterService.list(centerLambdaQueryWrapper); |
|
|
|