[improvement][chat] Fix the issue with the DatabaseMatchStrategy vari…

…able under multi-threading (#1963)

common/src/main/java/com/tencent/supersonic/common/calcite/Configuration.java

@@ -81,7 +81,7 @@ public static SqlParser.Config getParserConfig(EngineType engineType) {
                 .setUnquotedCasing(Casing.TO_UPPER).setConformance(sqlDialect.getConformance())
                 .setLex(Lex.BIG_QUERY);
         if (EngineType.HANADB.equals(engineType)) {
-                parserConfig = parserConfig.setQuoting(Quoting.DOUBLE_QUOTE);
+            parserConfig = parserConfig.setQuoting(Quoting.DOUBLE_QUOTE);
         }
         parserConfig = parserConfig.setQuotedCasing(Casing.UNCHANGED);
         parserConfig = parserConfig.setUnquotedCasing(Casing.UNCHANGED);

common/src/main/java/com/tencent/supersonic/common/calcite/SqlDialectFactory.java

@@ -21,10 +21,11 @@ public class SqlDialectFactory {
             .withDatabaseProduct(DatabaseProduct.BIG_QUERY).withLiteralQuoteString("'")
             .withLiteralEscapedQuoteString("''").withUnquotedCasing(Casing.UNCHANGED)
             .withQuotedCasing(Casing.UNCHANGED).withCaseSensitive(false);
-    public static final Context HANADB_CONTEXT = SqlDialect.EMPTY_CONTEXT
-            .withDatabaseProduct(DatabaseProduct.BIG_QUERY).withLiteralQuoteString("'")
-            .withIdentifierQuoteString("\"").withLiteralEscapedQuoteString("''").withUnquotedCasing(Casing.UNCHANGED)
-            .withQuotedCasing(Casing.UNCHANGED).withCaseSensitive(true);
+    public static final Context HANADB_CONTEXT =
+            SqlDialect.EMPTY_CONTEXT.withDatabaseProduct(DatabaseProduct.BIG_QUERY)
+                    .withLiteralQuoteString("'").withIdentifierQuoteString("\"")
+                    .withLiteralEscapedQuoteString("''").withUnquotedCasing(Casing.UNCHANGED)
+                    .withQuotedCasing(Casing.UNCHANGED).withCaseSensitive(true);
     private static Map<EngineType, SemanticSqlDialect> sqlDialectMap;
 
     static {

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

@@ -1,15 +1,13 @@
 package com.tencent.supersonic.common.jsqlparser;
 
+import net.sf.jsqlparser.expression.Alias;
 import net.sf.jsqlparser.statement.select.SelectItem;
 import net.sf.jsqlparser.statement.select.SelectItemVisitorAdapter;
+import org.apache.commons.lang3.StringUtils;
 
 import java.util.HashMap;
 import java.util.Map;
 
-import org.apache.commons.lang3.StringUtils;
-
-import net.sf.jsqlparser.expression.Alias;
-
 public class FieldAliasReplaceNameVisitor extends SelectItemVisitorAdapter {
     private Map<String, String> fieldNameMap;
 

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

@@ -465,6 +465,7 @@ public static String replaceAliasFieldName(String sql, Map<String, String> field
         }
         return selectStatement.toString();
     }
+
     public static String replaceAlias(String sql) {
         Select selectStatement = SqlSelectHelper.getSelect(sql);
         if (!(selectStatement instanceof PlainSelect)) {

common/src/main/java/dev/langchain4j/provider/ZhipuModelFactory.java

@@ -9,6 +9,7 @@
 import dev.langchain4j.model.zhipu.ZhipuAiEmbeddingModel;
 import org.springframework.beans.factory.InitializingBean;
 import org.springframework.stereotype.Service;
+
 import static java.time.Duration.ofSeconds;
 
 @Service
@@ -32,8 +33,8 @@ public EmbeddingModel createEmbeddingModel(EmbeddingModelConfig embeddingModelCo
         return ZhipuAiEmbeddingModel.builder().baseUrl(embeddingModelConfig.getBaseUrl())
                 .apiKey(embeddingModelConfig.getApiKey()).model(embeddingModelConfig.getModelName())
                 .maxRetries(embeddingModelConfig.getMaxRetries()).callTimeout(ofSeconds(60))
-                .connectTimeout(ofSeconds(60)).writeTimeout(ofSeconds(60)).readTimeout(ofSeconds(60))
-                .logRequests(embeddingModelConfig.getLogRequests())
+                .connectTimeout(ofSeconds(60)).writeTimeout(ofSeconds(60))
+                .readTimeout(ofSeconds(60)).logRequests(embeddingModelConfig.getLogRequests())
                 .logResponses(embeddingModelConfig.getLogResponses()).build();
     }
 

common/src/test/java/com/tencent/supersonic/common/jsqlparser/SqlReplaceHelperTest.java

@@ -326,34 +326,35 @@ void testReplaceAlias() {
 
     @Test
     void testReplaceAliasFieldName() {
-            Map<String, String> map = new HashMap<>();
-            map.put("总访问次数", "\"总访问次数\"");
-            map.put("访问次数", "\"访问次数\"");
-            String sql = "select 部门, sum(访问次数) as 总访问次数 from 超音数 where "
-                            + "datediff('day', 数据日期, '2023-09-05') <= 3 group by 部门 order by 总访问次数 desc limit 10";
-            String replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
-            System.out.println(replaceSql);
-            Assert.assertEquals("SELECT 部门, sum(访问次数) AS \"总访问次数\" FROM 超音数 WHERE "
-                            + "datediff('day', 数据日期, '2023-09-05') <= 3 GROUP BY 部门 ORDER BY \"总访问次数\" DESC LIMIT 10",
-                            replaceSql);
-
-            sql = "select 部门, sum(访问次数) as 总访问次数 from 超音数 where "
-                            + "(datediff('day', 数据日期, '2023-09-05') <= 3) and 数据日期 = '2023-10-10' "
-                            + "group by 部门 order by 总访问次数 desc limit 10";
-            replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
-            System.out.println(replaceSql);
-            Assert.assertEquals("SELECT 部门, sum(访问次数) AS \"总访问次数\" FROM 超音数 WHERE "
-                            + "(datediff('day', 数据日期, '2023-09-05') <= 3) AND 数据日期 = '2023-10-10' "
-                            + "GROUP BY 部门 ORDER BY \"总访问次数\" DESC LIMIT 10", replaceSql);
-
-            sql = "select 部门, sum(访问次数) as 访问次数 from 超音数 where "
-                            + "(datediff('day', 数据日期, '2023-09-05') <= 3) and 数据日期 = '2023-10-10' "
-                            + "group by 部门 order by 访问次数 desc limit 10";
-            replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
-            System.out.println(replaceSql);
-            Assert.assertEquals("SELECT 部门, sum(\"访问次数\") AS \"访问次数\" FROM 超音数 WHERE (datediff('day', 数据日期, "
-                            + "'2023-09-05') <= 3) AND 数据日期 = '2023-10-10' GROUP BY 部门 ORDER BY \"访问次数\" DESC LIMIT 10",
-                            replaceSql);
+        Map<String, String> map = new HashMap<>();
+        map.put("总访问次数", "\"总访问次数\"");
+        map.put("访问次数", "\"访问次数\"");
+        String sql = "select 部门, sum(访问次数) as 总访问次数 from 超音数 where "
+                + "datediff('day', 数据日期, '2023-09-05') <= 3 group by 部门 order by 总访问次数 desc limit 10";
+        String replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
+        System.out.println(replaceSql);
+        Assert.assertEquals("SELECT 部门, sum(访问次数) AS \"总访问次数\" FROM 超音数 WHERE "
+                + "datediff('day', 数据日期, '2023-09-05') <= 3 GROUP BY 部门 ORDER BY \"总访问次数\" DESC LIMIT 10",
+                replaceSql);
+
+        sql = "select 部门, sum(访问次数) as 总访问次数 from 超音数 where "
+                + "(datediff('day', 数据日期, '2023-09-05') <= 3) and 数据日期 = '2023-10-10' "
+                + "group by 部门 order by 总访问次数 desc limit 10";
+        replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
+        System.out.println(replaceSql);
+        Assert.assertEquals("SELECT 部门, sum(访问次数) AS \"总访问次数\" FROM 超音数 WHERE "
+                + "(datediff('day', 数据日期, '2023-09-05') <= 3) AND 数据日期 = '2023-10-10' "
+                + "GROUP BY 部门 ORDER BY \"总访问次数\" DESC LIMIT 10", replaceSql);
+
+        sql = "select 部门, sum(访问次数) as 访问次数 from 超音数 where "
+                + "(datediff('day', 数据日期, '2023-09-05') <= 3) and 数据日期 = '2023-10-10' "
+                + "group by 部门 order by 访问次数 desc limit 10";
+        replaceSql = SqlReplaceHelper.replaceAliasFieldName(sql, map);
+        System.out.println(replaceSql);
+        Assert.assertEquals(
+                "SELECT 部门, sum(\"访问次数\") AS \"访问次数\" FROM 超音数 WHERE (datediff('day', 数据日期, "
+                        + "'2023-09-05') <= 3) AND 数据日期 = '2023-10-10' GROUP BY 部门 ORDER BY \"访问次数\" DESC LIMIT 10",
+                replaceSql);
     }
 
     @Test

headless/chat/src/main/java/com/tencent/supersonic/headless/chat/mapper/DatabaseMatchStrategy.java

@@ -27,12 +27,12 @@
 @Slf4j
 public class DatabaseMatchStrategy extends SingleMatchStrategy<DatabaseMapResult> {
 
-    private List<SchemaElement> allElements;
+    private ThreadLocal<List<SchemaElement>> allElements = ThreadLocal.withInitial(ArrayList::new);
 
     @Override
     public Map<MatchText, List<DatabaseMapResult>> match(ChatQueryContext chatQueryContext,
             List<S2Term> terms, Set<Long> detectDataSetIds) {
-        this.allElements = getSchemaElements(chatQueryContext);
+        allElements.set(getSchemaElements(chatQueryContext));
         return super.match(chatQueryContext, terms, detectDataSetIds);
     }
 
@@ -43,7 +43,7 @@ public List<DatabaseMapResult> detectByStep(ChatQueryContext chatQueryContext,
         }
 
         Double metricDimensionThresholdConfig = getThreshold(chatQueryContext);
-        Map<String, Set<SchemaElement>> nameToItems = getNameToItems(allElements);
+        Map<String, Set<SchemaElement>> nameToItems = getNameToItems(allElements.get());
         List<DatabaseMapResult> results = new ArrayList<>();
         for (Entry<String, Set<SchemaElement>> entry : nameToItems.entrySet()) {
             String name = entry.getKey();