opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 02:32:33 +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

36
chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/builder/SqlServerSqlBuilder.java
View File

@ -3,13 +3,12 @@ package ai.chat2db.plugin.sqlserver.builder;
import ai.chat2db.plugin.sqlserver.type.SqlServerColumnTypeEnum;
import ai.chat2db.plugin.sqlserver.type.SqlServerIndexTypeEnum;
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 ai.chat2db.spi.sql.Chat2DBContext;
import org.apache.commons.lang3.StringUtils;
public class SqlServerSqlBuilder implements SqlBuilder {
public class SqlServerSqlBuilder extends DefaultSqlBuilder implements SqlBuilder {
@Override
public String buildCreateTableSql(Table table) {
StringBuilder script = new StringBuilder();
@ -142,4 +141,33 @@ public class SqlServerSqlBuilder implements SqlBuilder {
}
return "";
}
@Override
public String buildCreateDatabaseSql(Database database) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("CREATE DATABASE [" + database.getName() + "]");
if (StringUtils.isNotBlank(database.getCollation())) {
sqlBuilder.append(" COLLATE ").append(database.getCollation());
}
sqlBuilder.append("\ngo\n");
if (StringUtils.isNotBlank(database.getComment())) {
sqlBuilder.append("exec [" + database.getName() + "].sys. sp_addextendedproperty 'MS_Description','")
.append(database.getComment()).append("'").append("'\ngo\n");
}
return sqlBuilder.toString();
}
@Override
public String buildCreateSchemaSql(Schema schema) {
StringBuilder sqlBuilder = new StringBuilder();
sqlBuilder.append("CREATE SCHEMA [" + schema.getName() + "] \ngo\n");
if (StringUtils.isNotBlank(schema.getComment())) {
sqlBuilder.append("exec sp_addextendedproperty 'MS_Description','")
.append(schema.getComment()).append("'").append(",'SCHEMA'")
.append(",'").append(schema.getName()).append("'").append("'\ngo\n");
}
return sqlBuilder.toString();
}
}