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

@@ -1,9 +1,11 @@
package com.vci.ubcs.system.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
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;
@@ -11,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 +43,9 @@
   @Resource
   private IUserClient userClient;

   @Resource
   private NacosConfigCache nacosConfigCache;

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

   /**