施工管理列表添加电站安装规模

amos-boot-system-jxiop/amos-boot-module-hygf-biz/src/main/java/com/yeejoin/amos/boot/module/hygf/biz/controller/WorkOrderController.java

@@ -251,6 +251,16 @@ public class WorkOrderController extends BaseController {
         workOrder.setRegionCompanyId(userUnitInformationDto.getRegionalCompaniesSeq());
         workOrder.setAmosDealerId(userUnitInformationDto.getAmosDealerId());
         List<WorkOrderPage>  list=workOrderMapper.queryForwgpage(workOrder);
+
+        if(list != null && list.size() > 0) {
+            list.forEach(e -> {
+                if(e.getPeasantHouseholdNo() != null) {
+                    e.setScale(householdContractMapper.getHygfCommercialScale(e.getPeasantHouseholdNo()));
+                }
+            });
+
+        }
+
         return ResponseHelper.buildResponse(list);
     }