From eb1c7808ccd8cf2aa2d0eae1ab20e560908841e3 Mon Sep 17 00:00:00 2001
From: dengzedong <dengzedong@email>
Date: 星期四, 19 六月 2025 18:38:39 +0800
Subject: [PATCH] setDataSource(dataSet.getDataSource())

---
 iailab-module-data/iailab-module-data-biz/src/main/java/com/iailab/module/data/ind/collection/utils/IndSqlUtils.java |  270 +++++++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 217 insertions(+), 53 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
index b682fdb..3f99482 100644
--- 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
@@ -3,18 +3,22 @@
 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.IndDataSetFieldService;
 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 lombok.extern.slf4j.Slf4j;
 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;
 
@@ -23,6 +27,7 @@
  * @Description
  * @createTime 2024年10月04日
  */
+@Slf4j
 @Component
 public class IndSqlUtils {
 
@@ -33,7 +38,7 @@
     private IndDataSetService indDataSetService;
 
     @Autowired
-    private IndDataSetFieldService indDataSetFieldService;
+    private IndItemService indItemService;
 
     @Autowired
     private IndItemDerService indItemDerService;
@@ -42,7 +47,11 @@
 
     public final static String PATTERN_MON = "yyyy-MM";
 
-    public final static String PATTERN_DATE = "yyyy-MM-dd";
+    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();
@@ -50,14 +59,16 @@
         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());
+        result.setDataSource(dataSet.getDataSource());
         StringBuilder selectSql = new StringBuilder();
-        IndStatFuncEnum statFun = IndStatFuncEnum.getEumByCode(StringUtils.isBlank(indItem.getStatFunc())? IndStatFuncEnum.DEFAULT.getCode() : indItem.getStatFunc());
+        IndStatFuncEnum statFun = IndStatFuncEnum.getEumByCode(StringUtils.isBlank(indItem.getStatFunc()) ? IndStatFuncEnum.DEFAULT.getCode() : indItem.getStatFunc());
         switch (statFun) {
             case AVG:
                 selectSql.append("AVG(");
@@ -80,6 +91,7 @@
         }
         selectSql.append(indItem.getUsingField());
         selectSql.append(") data_value");
+        //selectSql.append(") \"data_value\"");
         result.setSelectSql(selectSql.toString());
         result.setViewSql(dataSet.getQuerySql());
         return result;
@@ -91,19 +103,19 @@
         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())){
+        if (StringUtils.isNotBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
             selectSql.append(indItem.getDimension());
             selectSql.append(", ");
             selectSql.append(result.getSelectSql());
-        }
-        if(StringUtils.isBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())){
+        } else if (StringUtils.isBlank(indItem.getDimension()) && StringUtils.isNotBlank(result.getIndItemAtom().getStatFunc())) {
             selectSql.append(result.getSelectSql());
-        }
-        else{
+        } else {
             selectSql.append(result.getSelectSql());
             selectSql.append(", ");
             selectSql.append(indItem.getTimeLabel());
@@ -118,106 +130,222 @@
         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(), PATTERN_DATE));
+                whereSql.append(" >= '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" < '");
+                calendar.add(Calendar.DAY_OF_YEAR, 1);
+                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(), PATTERN_DATE));
+                whereSql.append(" < '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.DAY_OF_YEAR, -1);
+                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(), PATTERN_DATE));
-                whereSql.append("' AND >= '");
+                whereSql.append(" < '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                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(), PATTERN_DATE));
+                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(), PATTERN_DATE));
-                whereSql.append("' AND >= '");
+                whereSql.append(" < '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                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(), PATTERN_DATE));
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_DAY_90:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" < '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                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(DateUtils.format(calendar.getTime(), PATTERN_MON));
+                whereSql.append(" < '");
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.MONTH, -1);
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
+                whereSql.append("'");
+                break;
+            case LAST_MONTH:
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" < '");
+                calendar.add(Calendar.MONTH, -1);
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.MONTH, -1);
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
                 whereSql.append("'");
                 break;
             case LAST_MONTH_12:
                 whereSql.append(indItem.getTimeLabel());
-                whereSql.append(" <= '");
-                whereSql.append(DateUtils.format(calendar.getTime(), PATTERN_MON));
-                whereSql.append("' AND '");
+                whereSql.append(" < '");
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
+                whereSql.append("' AND ");
                 whereSql.append(indItem.getTimeLabel());
-                whereSql.append("' >= '");
+                whereSql.append(" >= '");
                 calendar.add(Calendar.MONTH, -12);
-                whereSql.append(DateUtils.format(calendar.getTime(), PATTERN_MON));
+                whereSql.append(getYearMonth(calendar,indItem.getMonthStart(),datePattern));
                 whereSql.append("'");
                 break;
             case THIS_YEAR_MONTH:
-                calendar.set(Calendar.MONTH, 11);
+                whereSql.append(indItem.getTimeLabel());
                 whereSql.append(" <= '");
-                whereSql.append(DateUtils.format(calendar.getTime(), PATTERN_MON));
-                whereSql.append("' AND >= '");
+                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(), PATTERN_MON));
+                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(), PATTERN_YEAR));
+                whereSql.append(" <= '");
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" >= '");
+                calendar.add(Calendar.YEAR, -1);
+                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(), PATTERN_YEAR));
+                whereSql.append(" >= '");
+                calendar.set(Calendar.MINUTE, 0);
+                calendar.set(Calendar.HOUR_OF_DAY, 0);
+                calendar.set(Calendar.DAY_OF_YEAR, 1);
+                whereSql.append(DateUtils.format(calendar.getTime(), datePattern));
+                whereSql.append("' AND ");
+                whereSql.append(indItem.getTimeLabel());
+                whereSql.append(" < '");
+                calendar.add(Calendar.YEAR, 1);
+                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(), PATTERN_MON));
+                whereSql.append(DateUtils.format(indItem.getTimeEnd(), datePattern));
                 whereSql.append("' AND ");
                 whereSql.append(indItem.getTimeLabel());
                 whereSql.append(" >= '");
-                whereSql.append(DateUtils.format(indItem.getTimeStart(), PATTERN_MON));
+                whereSql.append(DateUtils.format(indItem.getTimeStart(), datePattern));
                 whereSql.append("'");
+                break;
+            case ALL:
+                whereSql.append("1 = 1");
                 break;
             default:
                 break;
         }
+        log.info("whereSql:" + whereSql);
         result.setWhereSql(whereSql.toString());
 
         // 拼接GROUP
         StringBuilder groupSql = new StringBuilder();
-        if (StringUtils.isNotBlank(indItem.getDimension())){
+        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())){
+        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 calendar,Integer monthStart,String datePattern) {
+        Calendar now = (Calendar) calendar.clone();
+        int day = now.get(Calendar.DAY_OF_MONTH);
+
+        if (day >= Math.abs(monthStart)) {
+            now.add(Calendar.MONTH, 1);
+        }
+
+        now.set(Calendar.DAY_OF_MONTH,Math.abs(monthStart));
+        // 定义日期格式化对象,指定格式为 yyyy-MM
+        SimpleDateFormat sdf = new SimpleDateFormat(datePattern);
+        // 格式化日期并返回结果
+        return sdf.format(now.getTime());
     }
 
     public QuerySourceValueDTO getDerSql(String itemId, Date startTime, Date endTime) {
@@ -228,37 +356,73 @@
         }
         result = getAtomSql(indItem.getAtomItemId());
 
+        IndItemEntity itemEntity = indItemService.get(itemId);
+        result.setItemNo(itemEntity.getItemNo());
+
         // 拼接SELECT
         StringBuilder selectSql = new StringBuilder();
-        selectSql.append(indItem.getDimension());
-        selectSql.append(", ");
-        selectSql.append(result.getSelectSql());
+        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());
+
+        // 根据时间粒度确定 时间格式
+        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;
+            }
+        }
 
         // 拼接WHERE
         StringBuilder whereSql = new StringBuilder();
         whereSql.append(" ");
         whereSql.append(indItem.getTimeLabel());
         whereSql.append(" <= '");
-        whereSql.append(DateUtils.format(endTime, PATTERN_MON));
-        whereSql.append("' AND >= '");
-        whereSql.append(DateUtils.format(startTime, PATTERN_MON));
+        whereSql.append(DateUtils.format(endTime, datePattern));
+        whereSql.append("' AND ");
+        whereSql.append(indItem.getTimeLabel());
+        whereSql.append(" >= '");
+        whereSql.append(DateUtils.format(startTime, datePattern));
         whereSql.append("'");
         result.setWhereSql(whereSql.toString());
 
         // 拼接GROUP
         StringBuilder groupSql = new StringBuilder();
-        groupSql.append(" GROUP BY ");
-        groupSql.append(indItem.getDimension());
+        if (StringUtils.isNotBlank(indItem.getDimension())) {
+            groupSql.append(indItem.getDimension());
+        }
         result.setGroupSql(groupSql.toString());
 
         // 拼接ORDER
         StringBuilder orderBySql = new StringBuilder();
-        orderBySql.append(indItem.getTimeLabel());
-        result.setOrderBySql(orderBySql.toString());
+        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