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

Support for creating database and schema

Browse Source
This commit is contained in:
SwallowGG
2023-10-23 15:33:26 +08:00
parent 0df93d771c
commit 61b2ea14a1
23 changed files with 340 additions and 178 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/builder/PostgreSQLSqlBuilder.java
View File

@ -3,9 +3,8 @@ package ai.chat2db.plugin.postgresql.builder;
import ai.chat2db.plugin.postgresql.type.PostgreSQLColumnTypeEnum;
import ai.chat2db.plugin.postgresql.type.PostgreSQLIndexTypeEnum;
import ai.chat2db.spi.SqlBuilder;
import ai.chat2db.spi.model.Table;
import ai.chat2db.spi.model.TableColumn;
import ai.chat2db.spi.model.TableIndex;
import ai.chat2db.spi.jdbc.DefaultSqlBuilder;
import ai.chat2db.spi.model.*;
import com.google.common.collect.Lists;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.BooleanUtils;
@ -16,7 +15,7 @@ import java.util.Map;
import java.util.stream.Collectors;
public class PostgreSQLSqlBuilder implements SqlBuilder {
public class PostgreSQLSqlBuilder extends DefaultSqlBuilder implements SqlBuilder {
@Override
public String buildCreateTableSql(Table table) {
StringBuilder script = new StringBuilder();
@ -172,4 +171,35 @@ public class PostgreSQLSqlBuilder implements SqlBuilder {
return sqlStr.toString();
}
@Override
public String buildCreateDatabaseSql(Database database) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("CREATE DATABASE \""+database.getName()+"\"");
sqlBuilder.append("\nWITH ");
if(StringUtils.isNotBlank(database.getCharset())){
sqlBuilder.append("\n LC_CTYPE = '").append(database.getCharset()).append("' ");
}
if(StringUtils.isNotBlank(database.getCollation())){
sqlBuilder.append("\n LC_COLLATE = '").append(database.getCollation()).append("' ");
}
if(StringUtils.isNotBlank(database.getComment())){
sqlBuilder.append("; COMMENT ON DATABASE \"").append(database.getName()).append("\" IS '").append(database.getComment()).append("';");
}
return sqlBuilder.toString();
}
@Override
public String buildCreateSchemaSql(Schema schema){
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("CREATE SCHEMA \""+schema.getName()+"\"");
if(StringUtils.isNotBlank(schema.getOwner())){
sqlBuilder.append(" AUTHORIZATION ").append(schema.getOwner());
}
if(StringUtils.isNotBlank(schema.getComment())){
sqlBuilder.append("; COMMENT ON SCHEMA \"").append(schema.getName()).append("\" IS '").append(schema.getComment()).append("';");
}
return sqlBuilder.toString();
}
}