数据隔离

wangqian
1 parent ab9a74b7
Showing 2 changed files with 5 additions and 1 deletions
urbanops-module-garden/src/main/java/com/zteits/urbanops/module/garden/dal/mysql/rentalmechanicalcost/RentalMechanicalCostMapper.java
urbanops-module-garden/src/main/java/com/zteits/urbanops/module/garden/service/personalcost/PersonalCostServiceImpl.java
@@ -32,7 +32,7 @@ public interface RentalMechanicalCostMapper extends BaseMapperX&lt;RentalMechanical
                 .eqIfPresent(RentalMechanicalCostDO::getUseDuration, reqVO.getUseDuration())
                 //.betweenIfPresent(RentalMechanicalCostDO::getEndUseTime, reqVO.getEndUseTime())
                 //.likeIfPresent(RentalMechanicalCostDO::getCompanyName, reqVO.getCompanyName())
-                //.eqIfPresent(RentalMechanicalCostDO::getCompanyId, reqVO.getCompanyId())
+                .eqIfPresent(RentalMechanicalCostDO::getCompanyId, reqVO.getCompanyId())
                 //.eqIfPresent(RentalMechanicalCostDO::getDeptId, reqVO.getDeptId())
                 //.eqIfPresent(RentalMechanicalCostDO::getRoadIds, reqVO.getRoadIds())
                 ///.//eqIfPresent(RentalMechanicalCostDO::getRemark, reqVO.getRemark())
@@ -22,6 +22,7 @@ import java.util.stream.Collectors;
 import static com.zteits.urbanops.framework.common.exception.enums.GlobalErrorCodeConstants.BAD_REQUEST;
 import static com.zteits.urbanops.framework.common.exception.util.ServiceExceptionUtil.exception;
 import static com.zteits.urbanops.framework.common.exception.util.ServiceExceptionUtil.exception0;
+import static com.zteits.urbanops.framework.security.core.util.SecurityFrameworkUtils.getCompanyId;
 import static com.zteits.urbanops.module.garden.enums.ErrorCodeConstants.PERSONAL_COST_NOT_EXISTS;
  
 /**
@@ -118,6 +119,9 @@ public class PersonalCostServiceImpl implements PersonalCostService {
  
     @Override
     public PageResult<PersonalCostDO> getPersonalCostPage(PersonalCostPageReqVO pageReqVO) {
+        if (getCompanyId() != 100) {
+            pageReqVO.setCompanyId(String.valueOf(getCompanyId()));
+        }
         return personalCostMapper.selectPage(pageReqVO);
     }