隐患修改

amos-boot-module/amos-boot-module-biz/amos-boot-module-latentdanger-biz/src/main/java/com/yeejoin/amos/latentdanger/business/controller/LatentDangerController.java

@@ -241,7 +241,8 @@ public class LatentDangerController extends BaseController {
     @TycloudOperation(ApiLevel = UserType.AGENCY)
     public ResponseModel listDanger(@ApiParam(value = "查询条件", required = true) @RequestBody PageParam pageParam) throws Exception {
         ReginParams reginParams = getSelectedOrgInfo();
-        pageParam.put("orgCode", reginParams.getCompany().getOrgCode());
+        String loginOrgCode = getOrgCode(reginParams);
+        pageParam.put("orgCode", loginOrgCode);
         IPage<LatentDanger> result = iLatentDangerService.pageListDanger(pageParam);
         return ResponseHelper.buildResponse(result);
     }
@@ -297,7 +298,8 @@ public class LatentDangerController extends BaseController {
     @TycloudOperation(ApiLevel = UserType.AGENCY)
     public ResponseModel<List<LatentDanger>> listAllDanger(@RequestBody LatentDangerListParam searchParam) {
         ReginParams reginParams = getSelectedOrgInfo();
-        searchParam.setOrgCode(reginParams.getCompany().getOrgCode());
+        String loginOrgCode = getOrgCode(reginParams);
+        searchParam.setOrgCode(loginOrgCode);
         return ResponseHelper.buildResponse(iLatentDangerService.listAllDanger(searchParam));
     }
 
@@ -341,7 +343,8 @@ public class LatentDangerController extends BaseController {
     @TycloudOperation(ApiLevel = UserType.AGENCY)
     public ResponseModel reviewListDanger(@ApiParam(value = "查询条件", required = true) @RequestBody PageParam pageParam) throws Exception {
         ReginParams reginParams = getSelectedOrgInfo();
-        pageParam.put("orgCode", reginParams.getCompany().getOrgCode());
+        String loginOrgCode = getOrgCode(reginParams);
+        pageParam.put("orgCode", loginOrgCode);
         IPage<LatentDanger> result = iLatentDangerService.reviewListDanger(pageParam);
         return ResponseHelper.buildResponse(result);
     }