(improvement)(chat) Fix the SQL fields generated by the large model that contain (#1715)

This commit is contained in:
lexluo09
2024-09-25 22:49:46 +08:00
committed by GitHub
parent e410611a7c
commit 066b9cc1f7
3 changed files with 5 additions and 2 deletions

View File

@@ -1,5 +1,6 @@
package com.tencent.supersonic.common.jsqlparser;
import com.tencent.supersonic.common.util.StringUtil;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.schema.Column;
import org.apache.commons.lang3.StringUtils;
@@ -14,7 +15,7 @@ public class ParseVisitorHelper {
public void replaceColumn(
Column column, Map<String, String> fieldNameMap, boolean exactReplace) {
String columnName = column.getColumnName();
String columnName = StringUtil.replaceBackticks(column.getColumnName());
String replaceColumn = getReplaceValue(columnName, fieldNameMap, exactReplace);
if (StringUtils.isNotBlank(replaceColumn)) {
column.setColumnName(replaceColumn);

View File

@@ -46,6 +46,9 @@ public class StringUtil {
}
public static String replaceBackticks(String sql) {
if (StringUtils.isBlank(sql)) {
return sql;
}
return sql.replaceAll("`", "");
}
}

View File

@@ -66,6 +66,5 @@ public class HavingCorrector extends BaseSemanticCorrector {
SqlAddHelper.addFunctionToSelect(correctS2SQL, havingExpressionList);
semanticParseInfo.getSqlInfo().setCorrectedS2SQL(replaceSql);
}
return;
}
}