DataEase 是开源的数据可视化分析工具（ BI 工具 ）

package io.dataease.chart.manage; import io.dataease.api.chart.dto.*; import io.dataease.api.chart.request.ChartDrillRequest; import io.dataease.api.chart.request.ChartExtRequest; import io.dataease.api.dataset.dto.SqlVariableDetails; import io.dataease.api.dataset.union.DatasetGroupInfoDTO; import io.dataease.api.dataset.union.model.SQLMeta; import io.dataease.api.permissions.auth.dto.BusiPerCheckDTO; import io.dataease.api.permissions.dataset.dto.DataSetRowPermissionsTreeDTO; import io.dataease.chart.constant.ChartConstants; import io.dataease.chart.utils.ChartDataBuild; import io.dataease.constant.AuthEnum; import io.dataease.dataset.dto.DatasourceSchemaDTO; import io.dataease.dataset.manage.DatasetGroupManage; import io.dataease.dataset.manage.DatasetSQLManage; import io.dataease.dataset.manage.DatasetTableFieldManage; import io.dataease.dataset.manage.PermissionManage; import io.dataease.dataset.utils.SqlUtils; import io.dataease.datasource.provider.CalciteProvider; import io.dataease.datasource.request.DatasourceRequest; import io.dataease.dto.dataset.DatasetTableFieldDTO; import io.dataease.engine.constant.SQLConstants; import io.dataease.engine.sql.SQLProvider; import io.dataease.engine.trans.*; import io.dataease.engine.utils.SQLUtils; import io.dataease.engine.utils.Utils; import io.dataease.exception.DEException; import io.dataease.i18n.Translator; import io.dataease.result.ResultCode; import io.dataease.system.manage.CorePermissionManage; import io.dataease.utils.BeanUtils; import io.dataease.utils.JsonUtil; import jakarta.annotation.Resource; import org.apache.commons.collections4.CollectionUtils; import org.apache.commons.collections4.MapUtils; import org.apache.commons.lang3.BooleanUtils; import org.apache.commons.lang3.ObjectUtils; import org.apache.commons.lang3.StringUtils; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.stereotype.Component; import java.math.BigDecimal; import java.math.RoundingMode; import java.text.SimpleDateFormat; import java.util.*; import java.util.stream.Collectors; import java.util.stream.Stream; /** * @Author Junjun */ @Component public class ChartDataManage { @Resource private DatasetTableFieldManage datasetTableFieldManage; @Resource private DatasetGroupManage datasetGroupManage; @Resource private DatasetSQLManage datasetSQLManage; @Resource private CalciteProvider calciteProvider; @Resource private ChartViewManege chartViewManege; @Resource private PermissionManage permissionManage; @Resource private CorePermissionManage corePermissionManage; public static final String START_END_SEPARATOR = "_START_END_SPLIT"; private static Logger logger = LoggerFactory.getLogger(ChartDataManage.class); public ChartViewDTO calcData(ChartViewDTO view) throws Exception { ChartExtRequest chartExtRequest = view.getChartExtRequest(); if (chartExtRequest == null) { chartExtRequest = new ChartExtRequest(); } ChartViewDTO chartViewDTO = new ChartViewDTO(); if (ObjectUtils.isEmpty(view)) { DEException.throwException(ResultCode.DATA_IS_WRONG.code(), Translator.get("i18n_chart_delete")); } //如果是excel导出 如果是从仪表板获取图表数据，则仪表板的查询模式，查询结果的数量，覆盖图表对应的属性 if (view.getIsExcelExport()) { view.setResultMode(ChartConstants.VIEW_RESULT_MODE.CUSTOM); } else if (ChartConstants.VIEW_RESULT_MODE.CUSTOM.equals(chartExtRequest.getResultMode())) { view.setResultMode(chartExtRequest.getResultMode()); view.setResultCount(chartExtRequest.getResultCount()); } List<ChartViewFieldDTO> viewFields = new ArrayList<>(view.getViewFields()); final Map<String, List<ChartViewFieldDTO>> extFieldsMap = new LinkedHashMap<>(); if (ObjectUtils.isNotEmpty(viewFields)) { viewFields.forEach(field -> { String busiType = field.getBusiType(); List<ChartViewFieldDTO> list = extFieldsMap.containsKey(busiType) ? extFieldsMap.get(busiType) : new ArrayList<>(); list.add(field); extFieldsMap.put(field.getBusiType(), list); }); } // get all fields List<ChartViewFieldDTO> allFields = getAllChartFields(view); List<ChartViewFieldDTO> xAxisBase = new ArrayList<>(view.getXAxis()); List<ChartViewFieldDTO> xAxis = new ArrayList<>(view.getXAxis()); List<ChartViewFieldDTO> xAxisExt = new ArrayList<>(view.getXAxisExt()); if (StringUtils.equalsIgnoreCase(view.getType(), "table-pivot") || StringUtils.containsIgnoreCase(view.getType(), "group") || ("antv".equalsIgnoreCase(view.getRender()) && "line".equalsIgnoreCase(view.getType())) || StringUtils.equalsIgnoreCase(view.getType(), "flow-map")) { xAxis.addAll(xAxisExt); } List<ChartViewFieldDTO> yAxis = new ArrayList<>(view.getYAxis()); if (StringUtils.equalsIgnoreCase(view.getType(), "chart-mix") || StringUtils.equalsIgnoreCase(view.getType(), "bidirectional-bar") || StringUtils.equalsIgnoreCase(view.getType(), "quadrant")) { List<ChartViewFieldDTO> yAxisExt = new ArrayList<>(view.getYAxisExt()); yAxis.addAll(yAxisExt); } if (StringUtils.equalsIgnoreCase(view.getRender(), "antv") && StringUtils.equalsAnyIgnoreCase(view.getType(), "gauge", "liquid")) { List<ChartViewFieldDTO> sizeField = getSizeField(view); yAxis.addAll(sizeField); } boolean skipBarRange = false; boolean barRangeDate = false; if (StringUtils.equalsIgnoreCase(view.getType(), "bar-range")) { //针对区间条形图进行处理 yAxis.clear(); if (CollectionUtils.isNotEmpty(view.getYAxis()) && CollectionUtils.isNotEmpty(view.getYAxisExt())) { ChartViewFieldDTO axis1 = view.getYAxis().get(0); ChartViewFieldDTO axis2 = view.getYAxisExt().get(0); if (StringUtils.equalsIgnoreCase(axis1.getGroupType(), "q") && StringUtils.equalsIgnoreCase(axis2.getGroupType(), "q")) { yAxis.add(axis1); yAxis.add(axis2); } else if (StringUtils.equalsIgnoreCase(axis1.getGroupType(), "d") && axis1.getDeType() == 1 && StringUtils.equalsIgnoreCase(axis2.getGroupType(), "d") && axis2.getDeType() == 1) { barRangeDate = true; if (BooleanUtils.isTrue(view.getAggregate())) { axis1.setSummary("min"); axis2.setSummary("max"); yAxis.add(axis1); yAxis.add(axis2); } else { xAxis.add(axis1); xAxis.add(axis2); } } else { skipBarRange = true; } } } List<ChartViewFieldDTO> extStack = new ArrayList<>(view.getExtStack()); List<ChartViewFieldDTO> extBubble = new ArrayList<>(view.getExtBubble()); if (ObjectUtils.isNotEmpty(view.getExtLabel()) && enableExtData(view.getType())) { List<ChartViewFieldDTO> extLabel = new ArrayList<>(view.getExtLabel()); yAxis.addAll(extLabel); } if (ObjectUtils.isNotEmpty(view.getExtTooltip()) && enableExtData(view.getType())) { List<ChartViewFieldDTO> extTooltip = new ArrayList<>(view.getExtTooltip()); yAxis.addAll(extTooltip); } List<ChartFieldCustomFilterDTO> fieldCustomFilter = new ArrayList<>(view.getCustomFilter()); List<ChartViewFieldDTO> drill = new ArrayList<>(view.getDrillFields()); DatasetGroupInfoDTO table = datasetGroupManage.get(view