Source/UBCS/ubcs-service/ubcs-code/src/main/java/com/vci/ubcs/code/service/impl/MdmCountServiceImpl.java

@@ -4,15 +4,26 @@
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.vci.ubcs.code.entity.CodeClassify;
import com.vci.ubcs.code.entity.MdmCount;
import com.vci.ubcs.code.mapper.CommonsMapper;
import com.vci.ubcs.code.service.ICodeClassifyService;
import com.vci.ubcs.code.service.IMdmCountService;
import com.vci.ubcs.system.entity.MdmCountConfig;
import com.vci.ubcs.system.entity.Menu;
import com.vci.ubcs.system.feign.ISysClient;
import lombok.extern.slf4j.Slf4j;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.Func;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;

import static com.vci.ubcs.starter.web.util.VciBaseUtil.getTableName;

/**
 * 主数据统计服务
@@ -32,18 +43,84 @@
   @Resource
   private ICodeClassifyService codeClassifyService;

   @Resource
   private CommonsMapper commonsMapper;

   /**
    * 查询主数据总数
    * @param userId
    * @return
    * @throws ServiceException
    */
   @Override
   public List<MdmCount> getMdmCounts() {
      // 1、需要先查询出分类下顶级节点有哪些主数据,拿到业务类型名称
   public List<MdmCount> getMdmCounts(String userId) throws ServiceException{
      if(Func.isBlank(userId)){
         throw new ServiceException("必传参数用户id不能为空");
      }
      /** 1、先查询出各分类顶层节点有哪些主数据,拿到业务类型名称，相当于拿到主数据表名
         顶层节点的btmtypeid还需要用作查询条件对应的菜单表的code字段
         但是为了减少查询次数，就需在查询是使用btmtypeid字段进行排序
       */
      LambdaQueryWrapper<CodeClassify> wrapper = Wrappers.<CodeClassify>query()
         .lambda()
         .isNull(CodeClassify::getParentCodeClassifyOid).select(CodeClassify::getBtmTypeId);
      Map<String, Object> map = codeClassifyService.getMap(wrapper);
      // 2、然后需要查询出菜单路由和图标
         .isNull(CodeClassify::getParentCodeClassifyOid).select(CodeClassify::getBtmTypeId)
         .orderByAsc(CodeClassify::getBtmTypeId);
      List<CodeClassify> codeClassifies = codeClassifyService.list(wrapper);
      List<String> btyTypeIds = codeClassifies.stream().map(CodeClassify::getBtmTypeId).collect(Collectors.toList());
      // 2、然后feign调用需要查询出菜单路由和图标
      R<List<Menu>> menuList = sysClient.getMenuByCodes(btyTypeIds);
      if(!menuList.isSuccess() && menuList.getData().isEmpty()){
         throw new ServiceException("获取主数据统计时，主数据菜单查询为空，原因："+menuList.getMsg());
      }
      List<MdmCount> mdmCountList = new ArrayList<>();
      // 3、查询当前登录的用户配置了那些主数据统计项
      R<MdmCountConfig> mdmCountConfig = sysClient.getMdmCountConfig(userId);
      if(!mdmCountConfig.isSuccess()){
         throw new ServiceException("获取主数据统计项配置时出现错误，原因："+mdmCountConfig.getMsg());
      }
      List<String> mdmNames = null;
      if(Func.isBlank(mdmCountConfig.getData().getMdmName())){
         AtomicInteger count = new AtomicInteger();
         mdmNames = menuList.getData().stream().map(item->{
            if(count.getAndIncrement() <5){
               return item.getCode();
            }
            return null;
         }).collect(Collectors.toList());
      }else {
         mdmNames = Arrays.stream(mdmCountConfig.getData().getMdmName().split(",")).collect(Collectors.toList());
      }
      final List<String> mdmNamesFinal = mdmNames;
      // 4、然后set进对象中
      menuList.getData().stream().forEach(item->{
         Integer isDefault = mdmNamesFinal.contains(item.getCode()) ? 1:0;
         MdmCount mdmCount = new MdmCount(
            item.getName(),
            getCountByTableName(item.getCode()), /*查询主数据总数*/
            item.getPath(),
            item.getSource(),
            item.getCode(),
            isDefault
         );
         mdmCountList.add(mdmCount);
      });
      return mdmCountList;
   }

      // 3、然后设置进

      return null;
   /**
    * 拼接表名查询总数
    * @param tableName
    * @return
    */
   private String getCountByTableName(String tableName){
      String sql = "SELECT COUNT(*) FROM " + getTableName(tableName,true) +";";
      String count = "0";
      try {
         count = commonsMapper.queryCountStringBySql(sql);
      }catch (Exception e){
         throw new ServiceException("主数据总数查询失败，原因："+e.getCause());
      }
      return count;
   }

}