From ad67e60bdc758f04b813f4a08b711504a5d0c996 Mon Sep 17 00:00:00 2001
From: Jay <csj123456>
Date: 星期三, 30 四月 2025 11:28:49 +0800
Subject: [PATCH] 新增最近90天指标时间限定

---
 iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/utils/IndSqlUtils.java |  373 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 373 insertions(+), 0 deletions(-)

diff --git a/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/utils/IndSqlUtils.java b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/utils/IndSqlUtils.java
new file mode 100644
index 0000000..30f6c76
--- /dev/null
+++ b/iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/utils/IndSqlUtils.java
@@ -0,0 +1,373 @@
+package com.iailab.module.data.ind.collection.utils;
+
+import com.iailab.framework.common.util.date.DateUtils;
+import com.iailab.module.data.common.enums.IndStatFuncEnum;
+import com.iailab.module.data.common.enums.IndTimeLimitEnum;
+import com.iailab.module.data.enums.TimeGranularityEnum;
+import com.iailab.module.data.ind.data.dto.IndDataSetDTO;
+import com.iailab.module.data.ind.data.service.IndDataSetService;
+import com.iailab.module.data.ind.item.entity.IndItemEntity;
+import com.iailab.module.data.ind.item.service.IndItemAtomService;
+import com.iailab.module.data.ind.item.service.IndItemDerService;
+import com.iailab.module.data.ind.item.service.IndItemService;
+import com.iailab.module.data.ind.item.vo.IndItemAtomVO;
+import com.iailab.module.data.ind.item.vo.IndItemDerVO;
+import com.iailab.module.data.ind.value.dto.QuerySourceValueDTO;
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.text.SimpleDateFormat;
+import java.util.Calendar;
+import java.util.Date;
+
+/**
+ * @author PanZhibao
+ * @Description
+ * @createTime 2024年10月04日
+ */
+@Component
+public class IndSqlUtils {
+
+    @Autowired
+    private IndItemAtomService indItemAtomService;
+
+    @Autowired
+    private IndDataSetService indDataSetService;
+
+    @Autowired
+    private IndItemService indItemService;
+
+    @Autowired
+    private IndItemDerService indItemDerService;
+
+    public final static String PATTERN_YEAR = "yyyy";
+
+    public final static String PATTERN_MON = "yyyy-MM";
+
+    public final static String PATTERN_DAY = "yyyy-MM-dd";
+
+    public final static String PATTERN_HOUR = "yyyy-MM-dd HH";
+
+    public static final String PATTERN_MIN = "yyyy-MM-dd HH:mm:ss";
+
+    public QuerySourceValueDTO getAtomSql(String itemId) {
+        QuerySourceValueDTO result = new QuerySourceValueDTO();
+        IndItemAtomVO indItem = indItemAtomService.getByItemId(itemId);
+        if (indItem == null) {
+            return result;
+        }
+        IndItemEntity itemEntity = indItemService.get(itemId);
+        result.setItemNo(itemEntity.getItemNo());
+        result.setIndItemAtom(indItem);
+        IndDataSetDTO dataSet = indDataSetService.getDet(indItem.getDataSet());
+        if (dataSet == null) {
+            return result;
+        }
+        result.setDataSource(indItem.getDataSource());
+        StringBuilder selectSql = new StringBuilder();
+        IndStatFuncEnum statFun = IndStatFuncEnum.getEumByCode(StringUtils.isBlank(indItem.getStatFunc()) ? IndStatFuncEnum.DEFAULT.getCode() : indItem.getStatFunc());
+        switch (statFun) {
+            case AVG:
+                selectSql.append("AVG(");
+                break;
+            case COUNT:
+                selectSql.append("COUNT(");
+                break;
+            case MAX:
+                selectSql.append("MAX(");
+                break;
+            case MIN:
+                selectSql.append("MIN(");
+                break;
+            case SUM:
+                selectSql.append("SUM(");
+                break;
+            case DEFAULT:
+                selectSql.append("(");
+                break;
+        }
+        selectSql.append(indItem.getUsingField());
+        selectSql.append(") data_value");
+        //selectSql.append(") \"data_value\"");
+        result.setSelectSql(selectSql.toString());
+        result.setViewSql(dataSet.getQuerySql());
+        return result;
+    }
+
+    public QuerySourceValueDTO getDerSql(String itemId) {
+        QuerySourceValueDTO result = getAtomSql(itemId);
+        IndItemDerVO indItem = indItemDerService.getByItemId(itemId);
+        if (indItem == null) {
+            return result;
+        }
+        IndItemEntity itemEntity = indItemService.get(itemId);
+        result.setItemNo(itemEntity.getItemNo());
+        result = getAtomSql(indItem.getAtomItemId());
+
+        // 拼接SELECT
+        StringBuilder selectSql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
+            selectSql.append(indItem.getDimension());
+            selectSql.append(", ");
+            selectSql.append(result.getSelectSql());
+        } else if (StringUtils.isBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
+            selectSql.append(result.getSelectSql());
+        } else {
+            selectSql.append(result.getSelectSql());
+            selectSql.append(", ");
+            selectSql.append(indItem.getTimeLabel());
+            selectSql.append(" data_time");
+        }
+        result.setSelectSql(selectSql.toString());
+
+        // 拼接WHERE
+        StringBuilder whereSql = new StringBuilder();
+        whereSql.append(" ");
+        Calendar calendar = Calendar.getInstance();
+        calendar.set(Calendar.MILLISECOND, 0);
+        calendar.set(Calendar.SECOND, 0);
+        IndTimeLimitEnum timeLimit = IndTimeLimitEnum.getEumByCode(indItem.getTimeLimit());
+        // 根据时间粒度确定 时间格式
+        String datePattern = PATTERN_MIN;
+        if (StringUtils.isNotBlank(itemEntity.getTimeGranularity())) {
+            switch (TimeGranularityEnum.getEumByCode(itemEntity.getTimeGranularity())){
+                case TG_MIN:
+                    datePattern = PATTERN_MIN;
+                    break;
+                case TG_HOUR:
+                    datePattern = PATTERN_HOUR;
+                    break;
+                case TG_DAY:
+                    datePattern = PATTERN_DAY;
+                    break;
+                case TG_MON:
+                    datePattern = PATTERN_MON;
+                    break;
+                default:
+                    datePattern = PATTERN_YEAR;
+            }
+        }
+        switch (timeLimit) {
+            case TODAY:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("='");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case YESTERDAY:
+                calendar.add(Calendar.DAY_OF_YEAR, -1);
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("='");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_DAY_7:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.DAY_OF_YEAR, -7);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_DAY_30:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.DAY_OF_YEAR, -30);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_DAY_90:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.DAY_OF_YEAR, -90);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case THIS_MONTH:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("='");
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart()));
+                whereSql.append("'");
+                break;
+            case LAST_MONTH:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("' < '");
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart()));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.MONTH, -1);
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart()));
+                whereSql.append("'");
+                break;
+            case LAST_MONTH_12:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart()));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("' >= '");
+                calendar.add(Calendar.MONTH, -12);
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart()));
+                whereSql.append("'");
+                break;
+            case THIS_YEAR_MONTH:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                calendar.set(Calendar.MONTH, 11);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.set(Calendar.MONTH, 0);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_YEAR:
+                calendar.add(Calendar.YEAR, -1);
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("='");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case THIS_YEAR:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append("='");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_HOUR_24:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.HOUR_OF_DAY, -24);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case CUSTOM:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(indItem.getTimeEnd(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                whereSql.append(DateUtils.format(indItem.getTimeStart(), datePattern));
+                whereSql.append("'");
+                break;
+            case ALL:
+                whereSql.append("1 = 1");
+                break;
+            default:
+                break;
+        }
+        result.setWhereSql(whereSql.toString());
+
+        // 拼接GROUP
+        StringBuilder groupSql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension())) {
+            groupSql.append(indItem.getDimension());
+        }
+        result.setGroupSql(groupSql.toString());
+
+        // 拼接ORDER
+        StringBuilder orderBySql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension()) && indItem.getDimension().equals(indItem.getTimeLabel())) {
+            orderBySql.append(indItem.getTimeLabel());
+            result.setOrderBySql(orderBySql.toString());
+        }
+        return result;
+    }
+
+    public static String getYearMonth(Calendar now,Integer monthStart) {
+        int day = now.get(Calendar.DAY_OF_MONTH);
+
+        // 当前月
+        if (monthStart > 0) {
+            if (day < monthStart) {
+                now.add(Calendar.MONTH, -1);
+            }
+        }
+
+        if (monthStart < 0) {
+            if (day >= Math.abs(monthStart)) {
+                now.add(Calendar.MONTH, 1);
+            }
+        }
+        // 定义日期格式化对象,指定格式为 yyyy-MM
+        SimpleDateFormat sdf = new SimpleDateFormat(PATTERN_MON);
+        // 格式化日期并返回结果
+        return sdf.format(now.getTime());
+    }
+
+    public QuerySourceValueDTO getDerSql(String itemId, Date startTime, Date endTime) {
+        QuerySourceValueDTO result = getAtomSql(itemId);
+        IndItemDerVO indItem = indItemDerService.getByItemId(itemId);
+        if (indItem == null) {
+            return result;
+        }
+        result = getAtomSql(indItem.getAtomItemId());
+
+        IndItemEntity itemEntity = indItemService.get(itemId);
+        result.setItemNo(itemEntity.getItemNo());
+
+        // 拼接SELECT
+        StringBuilder selectSql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
+            selectSql.append(indItem.getDimension());
+            selectSql.append(", ");
+            selectSql.append(result.getSelectSql());
+        } else if (StringUtils.isBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
+            selectSql.append(result.getSelectSql());
+        } else {
+            selectSql.append(result.getSelectSql());
+        }
+        selectSql.append(", ");
+        selectSql.append(indItem.getTimeLabel());
+        selectSql.append(" data_time");
+        result.setSelectSql(selectSql.toString());
+
+        // 拼接WHERE
+        StringBuilder whereSql = new StringBuilder();
+        whereSql.append(" ");
+        whereSql.append(indItem.getTimeLabel());
+        whereSql.append(" <= '");
+        whereSql.append(DateUtils.format(endTime, DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND));
+        whereSql.append("' AND ");
+        whereSql.append(indItem.getTimeLabel());
+        whereSql.append(" >= '");
+        whereSql.append(DateUtils.format(startTime, DateUtils.FORMAT_YEAR_MONTH_DAY_HOUR_MINUTE_SECOND));
+        whereSql.append("'");
+        result.setWhereSql(whereSql.toString());
+
+        // 拼接GROUP
+        StringBuilder groupSql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension())) {
+            groupSql.append(indItem.getDimension());
+        }
+        result.setGroupSql(groupSql.toString());
+
+        // 拼接ORDER
+        StringBuilder orderBySql = new StringBuilder();
+        if (StringUtils.isNotBlank(indItem.getDimension()) && indItem.getDimension().equals(indItem.getTimeLabel())) {
+            orderBySql.append(indItem.getTimeLabel());
+            result.setOrderBySql(orderBySql.toString());
+        }
+
+        return result;
+    }
+}
\ No newline at end of file

--
Gitblit v1.9.3