怎加队伍单位查询接口

amos-boot-module/amos-boot-module-biz/amos-boot-module-common-biz/src/main/java/com/yeejoin/amos/boot/module/common/biz/controller/FireTeamController.java

@@ -197,7 +197,29 @@ public class FireTeamController extends BaseController {
         return ResponseHelper.buildResponse(menus);
     }
 
+    @TycloudOperation(ApiLevel = UserType.AGENCY)
+    @RequestMapping(value = "/{companyId}/listTreetypeLikeCode", method = RequestMethod.GET)
+    @ApiOperation(httpMethod = "GET", value = "查询所属单位下消防队伍树", notes = "查询所属单位下消防队伍树")
+    public ResponseModel<List<Menu>> getlistTreeByCompanyIdLike(
+            @PathVariable(value = "companyId") String companyId) throws Exception {
+        QueryWrapper<FireTeam> fireTeamQueryWrapper = new QueryWrapper<>();
+        fireTeamQueryWrapper.eq("is_delete",0);
+        OrgUsr our=iOrgUsrService.getById(companyId);
+        //车辆装备需要判断
+        if(our.getParentId()!=null){
+            //判断是不是部门
 
+            if(our.getBizOrgType().equals("COMPANY")){
+                fireTeamQueryWrapper.likeRight("biz_org_code",our.getBizOrgCode());
+            }else{
+                String id= orgUsrMapper.getParentList(companyId);
+                OrgUsr ourf=iOrgUsrService.getById(id);
+                fireTeamQueryWrapper.likeRight("biz_org_code",ourf.getBizOrgCode());
+            }
+        }
+        List<Menu> menus = iFireTeamService.getTeamTreeLikeCode(fireTeamQueryWrapper);
+        return ResponseHelper.buildResponse(menus);
+    }
 
     @TycloudOperation(ApiLevel = UserType.AGENCY)
     @RequestMapping(value = "/{companyId}/getBizCode", method = RequestMethod.GET)

amos-boot-module/amos-boot-module-biz/amos-boot-module-common-biz/src/main/java/com/yeejoin/amos/boot/module/common/biz/service/impl/FireTeamServiceImpl.java

@@ -183,6 +183,11 @@ public class FireTeamServiceImpl extends BaseService<FireTeamDto, FireTeam, Fire
                 null);
     }
 
+    public List<Menu> getTeamTreeLikeCode(QueryWrapper<FireTeam> columnMap) throws Exception {
+        Collection<FireTeam> list = this.list(columnMap);
+        return TreeParser.getTree(null, list, FireTeam.class.getName(), "getSequenceNbr", 2, "getName", "getParent",
+                null);
+    }
     /**
      * 获取机场单位-队伍树
      *