package cn.iocoder.yudao.module.bi.service.decision;
|
|
import cn.iocoder.yudao.module.bi.dal.mysql.decision.BiKpiDefinitionMapper;
|
import cn.iocoder.yudao.module.bi.dal.dataobject.decision.BiKpiDefinitionDO;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.jdbc.core.JdbcTemplate;
|
import org.springframework.stereotype.Service;
|
|
import java.math.BigDecimal;
|
import java.util.*;
|
|
/**
|
* 决策支持 KPI 聚合 Service 实现
|
*
|
* 通过 JdbcTemplate 跨模块执行聚合 SQL
|
*
|
* @author 超级管理员
|
*/
|
@Slf4j
|
@Service
|
public class BiDecisionKpiServiceImpl implements BiDecisionKpiService {
|
|
@Resource
|
private BiKpiDefinitionMapper kpiMapper;
|
|
@Resource
|
private JdbcTemplate jdbcTemplate;
|
|
@Override
|
public Map<String, Object> getKpiOverview() {
|
List<BiKpiDefinitionDO> kpis = kpiMapper.selectList();
|
List<Map<String, Object>> items = new ArrayList<>();
|
|
for (BiKpiDefinitionDO kpi : kpis) {
|
Map<String, Object> item = new LinkedHashMap<>();
|
item.put("code", kpi.getCode());
|
item.put("name", kpi.getName());
|
item.put("category", kpi.getCategory());
|
item.put("unit", kpi.getUnit());
|
item.put("chartType", kpi.getChartType());
|
item.put("status", kpi.getStatus());
|
|
// 执行查询获取当前值
|
BigDecimal value = executeKpiSql(kpi.getQuerySql());
|
item.put("value", value);
|
|
// 判断预警状态
|
String alertStatus = evaluateAlertStatus(kpi, value);
|
item.put("alertStatus", alertStatus);
|
|
items.add(item);
|
}
|
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("kpis", items);
|
result.put("total", items.size());
|
return result;
|
}
|
|
@Override
|
public Map<String, Object> getKpiDetail(String kpiCode) {
|
BiKpiDefinitionDO kpi = kpiMapper.selectOne(BiKpiDefinitionDO::getCode, kpiCode);
|
if (kpi == null) {
|
return Collections.emptyMap();
|
}
|
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("code", kpi.getCode());
|
result.put("name", kpi.getName());
|
result.put("unit", kpi.getUnit());
|
result.put("category", kpi.getCategory());
|
result.put("value", executeKpiSql(kpi.getQuerySql()));
|
|
return result;
|
}
|
|
@Override
|
public Map<String, Object> getKpiByCategory(String category) {
|
List<BiKpiDefinitionDO> kpis = kpiMapper.selectList(
|
BiKpiDefinitionDO::getCategory, category);
|
List<Map<String, Object>> items = new ArrayList<>();
|
|
for (BiKpiDefinitionDO kpi : kpis) {
|
Map<String, Object> item = new LinkedHashMap<>();
|
item.put("code", kpi.getCode());
|
item.put("name", kpi.getName());
|
item.put("unit", kpi.getUnit());
|
item.put("value", executeKpiSql(kpi.getQuerySql()));
|
items.add(item);
|
}
|
|
Map<String, Object> result = new LinkedHashMap<>();
|
result.put("category", category);
|
result.put("kpis", items);
|
return result;
|
}
|
|
@Override
|
public BigDecimal refreshKpi(String kpiCode) {
|
BiKpiDefinitionDO kpi = kpiMapper.selectOne(BiKpiDefinitionDO::getCode, kpiCode);
|
if (kpi == null) {
|
return BigDecimal.ZERO;
|
}
|
return executeKpiSql(kpi.getQuerySql());
|
}
|
|
/**
|
* 执行 KPI 查询 SQL
|
*/
|
private BigDecimal executeKpiSql(String querySql) {
|
if (querySql == null || querySql.isBlank()) {
|
return BigDecimal.ZERO;
|
}
|
try {
|
List<Map<String, Object>> result = jdbcTemplate.queryForList(querySql);
|
if (result.isEmpty()) {
|
return BigDecimal.ZERO;
|
}
|
// 取第一行第一列的值
|
Map<String, Object> row = result.get(0);
|
Object value = row.values().iterator().next();
|
if (value instanceof Number) {
|
return BigDecimal.valueOf(((Number) value).doubleValue());
|
}
|
return BigDecimal.ZERO;
|
} catch (Exception e) {
|
log.error("[executeKpiSql][SQL执行失败] {}", querySql, e);
|
return BigDecimal.ZERO;
|
}
|
}
|
|
/**
|
* 评估预警状态
|
*/
|
private String evaluateAlertStatus(BiKpiDefinitionDO kpi, BigDecimal value) {
|
if (kpi.getThresholdCritical() != null) {
|
int cmp = value.compareTo(kpi.getThresholdCritical());
|
boolean triggered = ">".equals(kpi.getComparisonOperator()) ? cmp > 0
|
: "<".equals(kpi.getComparisonOperator()) ? cmp < 0
|
: ">=".equals(kpi.getComparisonOperator()) ? cmp >= 0
|
: "<=".equals(kpi.getComparisonOperator()) ? cmp <= 0 : false;
|
if (triggered) {
|
return "critical";
|
}
|
}
|
if (kpi.getThresholdWarn() != null) {
|
int cmp = value.compareTo(kpi.getThresholdWarn());
|
boolean triggered = ">".equals(kpi.getComparisonOperator()) ? cmp > 0
|
: "<".equals(kpi.getComparisonOperator()) ? cmp < 0
|
: ">=".equals(kpi.getComparisonOperator()) ? cmp >= 0
|
: "<=".equals(kpi.getComparisonOperator()) ? cmp <= 0 : false;
|
if (triggered) {
|
return "warn";
|
}
|
}
|
return "normal";
|
}
|
|
}
|
