opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 21:50:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(sql-builder): escape line

Browse Source
This commit is contained in:
zgq
2024-07-08 19:17:23 +08:00
parent ec9121bf35
commit 2d8f9baa77
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultSqlBuilder.java
View File

@ -258,7 +258,7 @@ public class DefaultSqlBuilder implements SqlBuilder<Table> {
*/
public String buildSingleInsertSql(String databaseName, String schemaName, String tableName, List<String> columnList, List<String> valueList) {
String baseSql = buildBaseInsertSql(databaseName, schemaName, tableName, columnList);
List<String> list = valueList.stream().map(EasyStringUtils::escapeString).toList();
List<String> list = valueList.stream().map(EasyStringUtils::escapeLineString).toList();
return baseSql + "(" + String.join(",", list) + ");";
}
@ -274,7 +274,7 @@ public class DefaultSqlBuilder implements SqlBuilder<Table> {
public String buildMultiInsertSql(String databaseName, String schemaName, String tableName, List<String> columnList, List<List<String>> valueLists) {
String baseSql = buildBaseInsertSql(databaseName, schemaName, tableName, columnList);
String valuesPart = valueLists.stream()
.map(values -> "(" + String.join(",", values.stream().map(EasyStringUtils::escapeString).toList()) + ")")
.map(values -> "(" + String.join(",", values.stream().map(EasyStringUtils::escapeLineString).toList()) + ")")
.collect(Collectors.joining(",\n"));
return baseSql + valuesPart + ";";
}