增加树查询

energy-manage-service/src/main/java/com/energy/manage/service/controller/company/WindCompanyController.java

@@ -39,6 +39,10 @@ public class WindCompanyController {
     @PostMapping(value = "/delCompany")
     @ApiOperation(value = "删除企业")
     public ResultResp delCompany(@RequestBody WindCompanyDelDto dto) {
+
+
+
+
         boolean flg = windCompanyService.delCompany(dto);
         return flg ? ResultResp.SUCCESS() : ResultResp.FAIL();
     }

energy-manage-service/src/main/java/com/energy/manage/service/controller/windfield/WindFieldController.java

@@ -109,6 +109,15 @@ public class WindFieldController {
     }
 
     @UserLoginToken
+    @PostMapping(value = "/getWindFieldVos")
+    @ApiOperation(value = "通过企业编号查询关联风场")
+    public ResultResp<List<WindFieldVo>> getWindFieldVos(@RequestParam(required = true) String companyCode) {
+
+        List<WindFieldVo>  list = windFieldService.getWindFieldVos(companyCode);
+        return ResultResp.SUCCESS(list);
+    }
+
+    @UserLoginToken
     @PostMapping(value = "/updateWindFieldStateById")
     @ApiOperation(value = "修改状态")
     public ResultResp updateWindFieldStateById(@RequestBody WindFieldUpdateStateDto windFieldUpdateStateDto) {

energy-manage-service/src/main/java/com/energy/manage/service/service/windfield/WindFieldService.java

@@ -94,6 +94,14 @@ public interface WindFieldService {
 
 
     /**
+     * 通过企业编号查询关联风场
+     * @param companyCode
+     * @return
+     */
+    List<WindFieldVo> getWindFieldVos(String companyCode);
+
+
+    /**
      * 整体树结构
      * @return
      */

energy-manage-service/src/main/java/com/energy/manage/service/service/windfield/impl/WindFieldServiceImpl.java

@@ -302,6 +302,22 @@ public class WindFieldServiceImpl extends BaseServiceImpl<WindFieldPO> implement
     }
 
     @Override
+    public List<WindFieldVo> getWindFieldVos(String companyCode) {
+        WindFieldPO windFieldPO = new WindFieldPO();
+        windFieldPO.setCompanyCode(companyCode);
+        List<WindFieldPO> list = windFieldMapper.select(windFieldPO);
+
+        WindFieldVo windFieldVo = null;
+        List<WindFieldVo> windFieldVos = Lists.newArrayList();
+        for(WindFieldPO windFieldPO1 : list){
+            windFieldVo = new WindFieldVo();
+            BeanUtil.copyProperties(windFieldPO1,windFieldVo);
+            windFieldVos.add(windFieldVo);
+        }
+        return windFieldVos;
+    }
+
+    @Override
     public List<Map<String, Object>> getWindFieldTree() {
         return windRelationServic.getWindRelationTree();
     }