[improvement][headless]Add ANY agg operator and remove table name from bizName of field. (#2028)

common/src/main/java/com/tencent/supersonic/common/jsqlparser/SqlSelectHelper.java

@@ -666,8 +666,9 @@ public class SqlSelectHelper {
                 }
                 if (withSelect instanceof ParenthesedSelect) {
                     ParenthesedSelect parenthesedSelect = (ParenthesedSelect) withSelect;
-                    PlainSelect withPlainSelect = parenthesedSelect.getPlainSelect();
-                    plainSelectList.add(withPlainSelect);
+                    List<PlainSelect> plainSelects = new ArrayList<>();
+                    SqlReplaceHelper.getFromSelect(parenthesedSelect, plainSelects);
+                    plainSelectList.addAll(plainSelects);
                 }
             }
         }
@@ -893,7 +894,9 @@ public class SqlSelectHelper {
             collectSelects(withItem.getSelect(), selects);
         } else if (select instanceof ParenthesedSelect) {
             ParenthesedSelect parenthesedSelect = (ParenthesedSelect) select;
-            collectSelects(parenthesedSelect.getPlainSelect(), selects);
+            List<PlainSelect> plainSelects = new ArrayList<>();
+            SqlReplaceHelper.getFromSelect(parenthesedSelect, plainSelects);
+            plainSelects.forEach(s -> collectSelects(s, selects));
         }
     }
 

common/src/main/java/com/tencent/supersonic/common/pojo/enums/AggOperatorEnum.java

@@ -1,6 +1,8 @@
 package com.tencent.supersonic.common.pojo.enums;
 
 public enum AggOperatorEnum {
+    ANY("ANY"),
+
     MAX("MAX"),
 
     MIN("MIN"),

headless/api/src/main/java/com/tencent/supersonic/headless/api/pojo/ColumnSchema.java

@@ -15,7 +15,7 @@ public class ColumnSchema {
 
     private FieldType filedType;
 
-    private AggOperatorEnum agg = AggOperatorEnum.SUM;
+    private AggOperatorEnum agg = AggOperatorEnum.ANY;
 
     private String name;
 

headless/chat/src/main/java/com/tencent/supersonic/headless/chat/corrector/SchemaCorrector.java

@@ -55,12 +55,7 @@ public class SchemaCorrector extends BaseSemanticCorrector {
             SemanticParseInfo semanticParseInfo) {
         Map<String, String> fieldNameMap =
                 getFieldNameMap(chatQueryContext, semanticParseInfo.getDataSetId());
-        // add as fieldName
         SqlInfo sqlInfo = semanticParseInfo.getSqlInfo();
-        List<String> asFields = SqlAsHelper.getAsFields(sqlInfo.getCorrectedS2SQL());
-        for (String asField : asFields) {
-            fieldNameMap.put(asField, asField);
-        }
         String sql = SqlReplaceHelper.replaceFields(sqlInfo.getCorrectedS2SQL(), fieldNameMap);
         sqlInfo.setCorrectedS2SQL(sql);
     }

headless/chat/src/test/java/com/tencent/supersonic/headless/chat/corrector/SchemaCorrectorTest.java

@@ -42,13 +42,15 @@ class SchemaCorrectorTest {
         SchemaCorrector schemaCorrector = new SchemaCorrector();
         schemaCorrector.correct(chatQueryContext, parseInfo);
 
-        Assert.assertEquals("SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' ORDER BY SUM(播放量) DESC LIMIT 10",
+        Assert.assertEquals(
+                "SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' ORDER BY SUM(播放量) DESC LIMIT 10",
                 parseInfo.getSqlInfo().getCorrectedS2SQL());
     }
 
     @Test
     void testRemoveUnmappedFilterValue() throws JsonProcessingException {
-        String sql = "SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' AND 商务组 = 'xxx' ORDER BY 播放量 DESC LIMIT 10";
+        String sql =
+                "SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' AND 商务组 = 'xxx' ORDER BY 播放量 DESC LIMIT 10";
         ChatQueryContext chatQueryContext = buildQueryContext(sql);
         SemanticParseInfo parseInfo = chatQueryContext.getCandidateQueries().get(0).getParseInfo();
 
@@ -60,7 +62,8 @@ class SchemaCorrectorTest {
         SchemaCorrector schemaCorrector = new SchemaCorrector();
         schemaCorrector.removeUnmappedFilterValue(chatQueryContext, parseInfo);
 
-        Assert.assertEquals("SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' ORDER BY 播放量 DESC LIMIT 10",
+        Assert.assertEquals(
+                "SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' ORDER BY 播放量 DESC LIMIT 10",
                 parseInfo.getSqlInfo().getCorrectedS2SQL());
 
         List<LLMReq.ElementValue> linkingValues = new ArrayList<>();
@@ -74,7 +77,8 @@ class SchemaCorrectorTest {
         parseInfo.getSqlInfo().setCorrectedS2SQL(sql);
         parseInfo.getSqlInfo().setParsedS2SQL(sql);
         schemaCorrector.removeUnmappedFilterValue(chatQueryContext, parseInfo);
-        Assert.assertEquals("SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' "
+        Assert.assertEquals(
+                "SELECT 歌曲名 FROM 歌曲 WHERE 发行日期 >= '2024-01-01' "
                         + "AND 商务组 = 'xxx' ORDER BY 播放量 DESC LIMIT 10",
                 parseInfo.getSqlInfo().getCorrectedS2SQL());
     }

headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelConverter.java

@@ -163,13 +163,11 @@ public class ModelConverter {
                         getIdentifyType(fieldType).name(), columnSchema.getColumnName(), 1);
                 modelDetail.getIdentifiers().add(identify);
             } else if (FieldType.measure.equals(fieldType)) {
-                Measure measure = new Measure(columnSchema.getName(),
-                        modelReq.getBizName() + "_" + columnSchema.getColumnName(),
+                Measure measure = new Measure(columnSchema.getName(), columnSchema.getColumnName(),
                         columnSchema.getColumnName(), columnSchema.getAgg().getOperator(), 1);
                 modelDetail.getMeasures().add(measure);
             } else {
-                Dimension dim = new Dimension(columnSchema.getName(),
-                        modelReq.getBizName() + "_" + columnSchema.getColumnName(),
+                Dimension dim = new Dimension(columnSchema.getName(), columnSchema.getColumnName(),
                         columnSchema.getColumnName(),
                         DimensionType.valueOf(columnSchema.getFiledType().name()), 1);
                 modelDetail.getDimensions().add(dim);