opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-31 11:42:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(Oraclesql-builder): wrap objectName

Browse Source
This commit is contained in:
zgq
2024-07-09 11:58:08 +08:00
parent ab96987d92
commit 0f23e12e50
1 changed files with 27 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/builder/OracleSqlBuilder.java
View File

@ -6,8 +6,13 @@ import ai.chat2db.spi.jdbc.DefaultSqlBuilder;
import ai.chat2db.spi.model.Table; import ai.chat2db.spi.model.Table;
import ai.chat2db.spi.model.TableColumn; import ai.chat2db.spi.model.TableColumn;
import ai.chat2db.spi.model.TableIndex; import ai.chat2db.spi.model.TableIndex;
import ai.chat2db.spi.util.SqlUtils;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.StringUtils;
import java.util.List;
import java.util.stream.Collectors;
public class OracleSqlBuilder extends DefaultSqlBuilder { public class OracleSqlBuilder extends DefaultSqlBuilder {
@Override @Override
public String buildCreateTableSql(Table table) { public String buildCreateTableSql(Table table) {
@ -149,4 +154,26 @@ public class OracleSqlBuilder extends DefaultSqlBuilder {
// } // }
// return sqlBuilder.toString(); // return sqlBuilder.toString();
// } // }
@Override
protected void buildTableName(String databaseName, String schemaName, String tableName, StringBuilder script) {
if (StringUtils.isNotBlank(databaseName)) {
script.append(SqlUtils.quoteObjectName(databaseName)).append('.');
}
script.append(SqlUtils.quoteObjectName(tableName));
}
/**
* @param columnList
* @param script
*/
@Override
protected void buildColumns(List<String> columnList, StringBuilder script) {
if (CollectionUtils.isNotEmpty(columnList)) {
script.append(" (")
.append(columnList.stream().map(SqlUtils::quoteObjectName).collect(Collectors.joining(",")))
.append(") ");
}
}
} }