package com.vci.ubcs.code.service.impl;
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
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.starter.web.util.VciBaseUtil;
|
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.secure.utils.AuthUtil;
|
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.*;
|
import java.util.concurrent.atomic.AtomicInteger;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
import static com.vci.ubcs.starter.web.util.VciBaseUtil.getTableName;
|
|
/**
|
* 主数据统计服务
|
* @author ludc
|
* @date 2023/10/16 15:47
|
*/
|
@Slf4j
|
@Service
|
public class MdmCountServiceImpl implements IMdmCountService {
|
|
/**
|
* 系统服务
|
*/
|
@Resource
|
private ISysClient sysClient;
|
|
@Resource
|
private ICodeClassifyService codeClassifyService;
|
|
@Resource
|
private CommonsMapper commonsMapper;
|
|
/**
|
* 查询主数据总数
|
* @param userId
|
* @return
|
* @throws ServiceException
|
*/
|
@Override
|
public List<MdmCount> getMdmCounts(String userId) throws ServiceException{
|
if(Func.isBlank(userId) && Func.isBlank(AuthUtil.getUserId().toString())){
|
throw new ServiceException("必传参数用户id不能为空");
|
}
|
/** 1、先查询出各分类顶层节点有哪些主数据,拿到绑定的业务类型名称，相当于拿到主数据表名
|
顶层节点的btmtypeid（2024年1月31日的时候改成了用分类的id）去查询对应的菜单表的code字段
|
，查询出主数据管理下面有哪些主数据的
|
但是为了减少查询次数，就需在查询是使用id字段进行排序
|
*/
|
LambdaQueryWrapper<CodeClassify> wrapper = Wrappers.<CodeClassify>query()
|
.lambda()
|
.isNull(CodeClassify::getParentCodeClassifyOid)/*顶层节点*/
|
.orderByAsc(CodeClassify::getId);
|
// 加上租户查询条件
|
if(!VciBaseUtil.checkAdminTenant()){
|
wrapper.eq(CodeClassify::getTenantId,AuthUtil.getTenantId());
|
}
|
List<CodeClassify> codeClassifies = codeClassifyService.list(wrapper);
|
// List<String> btmTypeIds = codeClassifies.stream().map(CodeClassify::getBtmTypeId).collect(Collectors.toList());
|
List<String> classifyIds = codeClassifies.stream().map(CodeClassify::getId).collect(Collectors.toList());
|
// 2、然后feign调用需要查询出菜单路由和图标
|
R<List<Menu>> menuList = sysClient.getMenuByCodes(classifyIds,Func.toLong(userId));
|
if(!menuList.isSuccess()){
|
throw new ServiceException("获取主数据统计时，主数据菜单查询失败，原因："+menuList.getMsg());
|
}
|
List<MdmCount> mdmCountList = new ArrayList<>();
|
// 3、查询当前登录的用户配置了那些主数据统计项
|
R<MdmCountConfig> mdmCountConfig = sysClient.getMdmCountConfig(Func.isBlank(userId) ? AuthUtil.getUserId().toString():userId);
|
if(!mdmCountConfig.isSuccess()){
|
throw new ServiceException("获取主数据统计项配置时出现错误，原因："+mdmCountConfig.getMsg());
|
}
|
List<String> mdmNames = null;
|
//首页配置显示少于5个就默认显示前五个
|
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、获取表名，当前未获取到表名，菜单的code值对应了分类的id，
|
未对应业务类型的id，所以需要通过菜单的id筛选出配置并且在使用的主数据分类,
|
然后通过分类信息查询到业务类型id，在通过业务类型接口查询出表名(这一步可以先省略因为分类上关联的业务类型id也可以拼接出表名)
|
*/
|
// 获取分类上的业务类型的id
|
Map<String, String> btmTypeIdMaps = menuList.getData().stream()
|
.filter(menu -> codeClassifies.stream()
|
.anyMatch(codeClassify -> codeClassify.getId().equals(menu.getCode()))
|
).collect(Collectors.toMap(
|
Menu::getCode,
|
menu -> codeClassifies.stream()
|
.filter(codeClassify -> codeClassify.getId().equals(menu.getCode()))
|
.findFirst()
|
.map(CodeClassify::getBtmTypeId)
|
.orElse(null)
|
));
|
|
// 5、然后set进对象中
|
menuList.getData().stream().forEach(item->{
|
Integer isDefault = mdmNamesFinal.contains(item.getCode()) ? 1:0;
|
MdmCount mdmCount = new MdmCount(
|
item.getName(),
|
getCountByTableName(btmTypeIdMaps.get(item.getCode())), /*查询主数据总数*/
|
item.getPath(),
|
item.getSource(),
|
item.getCode(),
|
isDefault
|
);
|
mdmCountList.add(mdmCount);
|
});
|
return mdmCountList;
|
}
|
|
/**
|
* 拼接表名查询总数
|
* @param tableName
|
* @return
|
*/
|
private String getCountByTableName(String tableName){
|
String count = "0";
|
if(Func.isBlank(tableName)){
|
return count;
|
}
|
String sql = "SELECT COUNT(*) FROM " + getTableName(tableName,true) +";";
|
try {
|
count = commonsMapper.queryCountStringBySql(sql);
|
}catch (Exception e){
|
throw new ServiceException("主数据总数查询失败，原因："+e.getCause());
|
}
|
return count;
|
}
|
|
}
|
