Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/StrategyServiceImpl.java

@@ -5,6 +5,7 @@
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.vci.ubcs.common.constant.CommonConstant;
import com.vci.ubcs.system.cache.NacosConfigCache;
import com.vci.ubcs.system.entity.Strategy;
import com.vci.ubcs.system.mapper.StrategyMapper;
import com.vci.ubcs.system.service.IStrategyService;
@@ -12,6 +13,7 @@
import org.springblade.core.cache.utils.CacheUtil;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.mp.support.Query;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.Func;
import org.springframework.data.domain.PageImpl;
import org.springframework.stereotype.Service;
@@ -40,6 +42,9 @@

   @Resource
   private IUserClient userClient;

   @Resource
   private NacosConfigCache nacosConfigCache;

   /**
    * 通过ID查询单条数据
@@ -70,7 +75,17 @@
   @Override
   public IPage<Strategy> queryAllByPage(Query query) {
      Page<Strategy> strategyPage = new Page<>(query.getCurrent(), query.getSize());
      return this.strategyMapper.queryAllByPage(strategyPage);
      // 添加租户查询条件
      IPage<Strategy> strategyIPage = this.strategyMapper.queryAllByPage(
         strategyPage,
         AuthUtil.getTenantId()
            .equals(
               nacosConfigCache
               .getAdminUserInfo()
               .getTenantId()
            ) ? null:AuthUtil.getTenantId()
      );
      return strategyIPage;
   }

   /**