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,8 +13,8 @@
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;
import org.springframework.transaction.annotation.Transactional;

@@ -42,6 +43,12 @@
   private IUserClient userClient;

   /**
    * nacos上拉取的配置
    */
   @Resource
   private NacosConfigCache nacosConfigCache;

   /**
    * 通过ID查询单条数据
    *
    * @param id 主键
@@ -58,7 +65,13 @@
    */
   @Override
   public Strategy queryByIsDefault() {
      return this.strategyMapper.queryByIsDefault();
      return this.strategyMapper.queryByIsDefault(AuthUtil.getTenantId()
         .equals(
            nacosConfigCache
               .getAdminUserInfo()
               .getTenantId()
         ) ? null:AuthUtil.getTenantId()
      );
   }

   /**
@@ -70,7 +83,18 @@
   @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;
   }

   /**