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.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.concurrent.atomic.AtomicInteger;
|
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{
|
/** 1、先查询出各分类顶层节点有哪些主数据,拿到业务类型名称，相当于拿到主数据表名
|
顶层节点的btmtypeid还需要用作查询条件对应的菜单表的code字段
|
但是为了减少查询次数，就需在查询是使用btmtypeid字段进行排序
|
*/
|
LambdaQueryWrapper<CodeClassify> wrapper = Wrappers.<CodeClassify>query()
|
.lambda()
|
.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(),
|
isDefault
|
);
|
mdmCountList.add(mdmCount);
|
});
|
return mdmCountList;
|
}
|
|
/**
|
* 拼接表名查询总数
|
* @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;
|
}
|
|
}
|
