|
@@ -40,9 +40,20 @@ public class ContractController {
|
|
public Response getList(@RequestBody Contract contract) {
|
|
public Response getList(@RequestBody Contract contract) {
|
|
List<Contract> list = contractService.getList(contract);
|
|
List<Contract> list = contractService.getList(contract);
|
|
tags(list);
|
|
tags(list);
|
|
|
|
+ rooms(list);
|
|
return Response.ok(list);
|
|
return Response.ok(list);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+ private void rooms(List<Contract> list) {
|
|
|
|
+ Map<Long, String> rooms = new HashMap();
|
|
|
|
+ for (Contract ct : list) {
|
|
|
|
+ rooms.clear();
|
|
|
|
+ if (StringUtil.notBlank(ct.getProjectItemTargetRoomIds())) {
|
|
|
|
+ ct.setRoomMap(projectItemTargetRoomService.getRoomLevel(rooms, ct.getProjectItemTargetRoomIds()));
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
private void tags(List<Contract> list) {
|
|
private void tags(List<Contract> list) {
|
|
List<Tag> tags = new ArrayList<>();
|
|
List<Tag> tags = new ArrayList<>();
|
|
for (Contract ct : list) {
|
|
for (Contract ct : list) {
|
|
@@ -62,6 +73,7 @@ public class ContractController {
|
|
int totalCount = contractService.getTotalCount(contract);
|
|
int totalCount = contractService.getTotalCount(contract);
|
|
List<Contract> dataList = contractService.getLimit(contract, currPage, pageSize);
|
|
List<Contract> dataList = contractService.getLimit(contract, currPage, pageSize);
|
|
tags(dataList);
|
|
tags(dataList);
|
|
|
|
+ rooms(dataList);
|
|
Map<String, Object> result = new HashMap<>();
|
|
Map<String, Object> result = new HashMap<>();
|
|
result.put("dataList", dataList);
|
|
result.put("dataList", dataList);
|
|
result.put("totalCount", totalCount);
|
|
result.put("totalCount", totalCount);
|
|
@@ -101,6 +113,8 @@ public class ContractController {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
contract.setTagList(tags);
|
|
contract.setTagList(tags);
|
|
|
|
+ Map<Long, String> rooms = new HashMap();
|
|
|
|
+ contract.setRoomMap(projectItemTargetRoomService.getRoomLevel(rooms, contract.getProjectItemTargetRoomIds()));
|
|
return Response.ok(contract);
|
|
return Response.ok(contract);
|
|
}
|
|
}
|
|
|
|
|