zhangyuewww 3 år sedan
förälder
incheckning
26fb2ee61f

+ 4 - 5
winsea-haixin-plugin-yiliangyiyun/src/main/java/com/yh/saas/plugin/yiliangyiyun/service/impl/CostManagementInfoServiceImpl.java

@@ -3,13 +3,11 @@ package com.yh.saas.plugin.yiliangyiyun.service.impl;
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.baomidou.mybatisplus.mapper.Wrapper;
 import com.baomidou.mybatisplus.plugins.Page;
+import com.winsea.svc.base.security.util.AuthSecurityUtils;
 import com.yh.saas.common.support.util.IdGenerator;
 import com.yh.saas.common.support.util.StringUtils;
 import com.yh.saas.plugin.yiliangyiyun.constant.NumberConstant;
-import com.yh.saas.plugin.yiliangyiyun.entity.CostManagementInfo;
-import com.yh.saas.plugin.yiliangyiyun.entity.CustomerInfo;
-import com.yh.saas.plugin.yiliangyiyun.entity.WarehouseBaseInfo;
-import com.yh.saas.plugin.yiliangyiyun.entity.WeightedDetails;
+import com.yh.saas.plugin.yiliangyiyun.entity.*;
 import com.yh.saas.plugin.yiliangyiyun.mapper.CostManagementInfoMapper;
 import com.yh.saas.plugin.yiliangyiyun.service.IContractGoodsInfoService;
 import com.yh.saas.plugin.yiliangyiyun.service.ICostManagementInfoService;
@@ -45,7 +43,8 @@ public class CostManagementInfoServiceImpl extends ServiceImpl<CostManagementInf
     @Override
     public List<CostManagementInfo> selectCostManagementInfo(CostManagementInfo costManagementInfo) {
         Wrapper<CostManagementInfo> costManagementInfoWrapper = new EntityWrapper<>();
-        costManagementInfoWrapper.eq("warehouse_type", costManagementInfo.getWarehouseType());
+        costManagementInfoWrapper.eq("warehouse_type", costManagementInfo.getWarehouseType())
+                                 .eq("comp_id", AuthSecurityUtils.getCurrentUserInfo().getCompId());
         costManagementInfoWrapper.orderBy("warehouse_name");
         if (costManagementInfo.getWarehouseId()!=null&&!costManagementInfo.getWarehouseId().isEmpty()){
             costManagementInfoWrapper.eq("warehouse_id", costManagementInfo.getWarehouseId());