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

Support sqlserver table schema edit

Browse Source
This commit is contained in:
SwallowGG
2023-10-09 21:58:00 +08:00
parent afbef2d7ac
commit 35f15835f1
9 changed files with 162 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteMetaData.java
View File

@ -6,10 +6,12 @@ import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import ai.chat2db.plugin.sqlite.builder.SqliteBuilder;
import ai.chat2db.plugin.sqlite.type.SqliteCollationEnum;
import ai.chat2db.plugin.sqlite.type.SqliteColumnTypeEnum;
import ai.chat2db.plugin.sqlite.type.SqliteIndexTypeEnum;
import ai.chat2db.spi.MetaData;
import ai.chat2db.spi.SqlBuilder;
import ai.chat2db.spi.jdbc.DefaultMetaService;
import ai.chat2db.spi.model.Database;
import ai.chat2db.spi.model.Schema;
@ -43,6 +45,10 @@ public class SqliteMetaData extends DefaultMetaService implements MetaData {
return Lists.newArrayList();
}
@Override
public SqlBuilder getSqlBuilder() {
return new SqliteBuilder();
}
@Override
public TableMeta getTableMeta(String databaseName, String schemaName, String tableName) {
return TableMeta.builder()
@ -56,6 +62,6 @@ public class SqliteMetaData extends DefaultMetaService implements MetaData {
@Override
public String getMetaDataName(String... names) {
return Arrays.stream(names).filter(name -> StringUtils.isBlank(name)).map(name -> "\"" + name + "\"").collect(Collectors.joining("."));
return Arrays.stream(names).filter(name -> StringUtils.isNotBlank(name)).map(name -> "\"" + name + "\"").collect(Collectors.joining("."));
}
}

43
chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/builder/SqliteBuilder.java
View File

@ -14,7 +14,7 @@ public class SqliteBuilder implements SqlBuilder {
public String buildCreateTableSql(Table table) {
StringBuilder script = new StringBuilder();
script.append("CREATE TABLE ");
script.append("\"").append(table.getName()).append("\"").append(" (").append("\n");
script.append("\"").append(table.getDatabaseName()).append("\".\"").append(table.getName()).append("\"").append(" (").append("\n");
// append column
for (TableColumn column : table.getColumnList()) {
@ -37,30 +37,6 @@ public class SqliteBuilder implements SqlBuilder {
script = new StringBuilder(script.substring(0, script.length() - 2));
script.append("\n)");
if (StringUtils.isNotBlank(table.getEngine())) {
script.append(" ENGINE=").append(table.getEngine());
}
if (StringUtils.isNotBlank(table.getCharset())) {
script.append(" DEFAULT CHARACTER SET=").append(table.getCharset());
}
if (StringUtils.isNotBlank(table.getCollate())) {
script.append(" COLLATE=").append(table.getCollate());
}
if (table.getIncrementValue() != null) {
script.append(" AUTO_INCREMENT=").append(table.getIncrementValue());
}
if (StringUtils.isNotBlank(table.getComment())) {
script.append(" COMMENT='").append(table.getComment()).append("'");
}
if (StringUtils.isNotBlank(table.getPartition())) {
script.append(" \n").append(table.getPartition());
}
script.append(";");
return script.toString();
@ -69,30 +45,27 @@ public class SqliteBuilder implements SqlBuilder {
@Override
public String buildModifyTaleSql(Table oldTable, Table newTable) {
StringBuilder script = new StringBuilder();
script.append("ALTER TABLE ").append("\"").append(oldTable.getName()).append("\"").append("\n");
if (!StringUtils.equalsIgnoreCase(oldTable.getName(), newTable.getName())) {
script.append("\t").append("RENAME TO ").append("\"").append(newTable.getName()).append("\"").append(",\n");
}
if (!StringUtils.equalsIgnoreCase(oldTable.getComment(), newTable.getComment())) {
script.append("\t").append("COMMENT=").append("'").append(newTable.getComment()).append("'").append(",\n");
}
if (oldTable.getIncrementValue() != newTable.getIncrementValue()) {
script.append("\t").append("AUTO_INCREMENT=").append(newTable.getIncrementValue()).append(",\n");
script.append("ALTER TABLE ").append("\"").append(oldTable.getDatabaseName()).append("\".\"").append(oldTable.getName()).append("\"").append("\n");
script.append("\t").append("RENAME TO ").append("\"").append(newTable.getName()).append("\"").append(";\n");
}
// append modify column
for (TableColumn tableColumn : newTable.getColumnList()) {
if (StringUtils.isNotBlank(tableColumn.getEditStatus()) && StringUtils.isNotBlank(tableColumn.getColumnType())&& StringUtils.isNotBlank(tableColumn.getName())){
script.append("ALTER TABLE ").append("\"").append(newTable.getDatabaseName()).append("\".\"").append(newTable.getName()).append("\"").append("\n");
SqliteColumnTypeEnum typeEnum = SqliteColumnTypeEnum.getByType(tableColumn.getColumnType());
script.append("\t").append(typeEnum.buildModifyColumn(tableColumn)).append(",\n");
script.append("\t").append(typeEnum.buildModifyColumn(tableColumn)).append(";\n");
}
}
// append modify index
for (TableIndex tableIndex : newTable.getIndexList()) {
if (StringUtils.isNotBlank(tableIndex.getEditStatus()) && StringUtils.isNotBlank(tableIndex.getType())) {
// script.append("ALTER TABLE ").append("\"").append(newTable.getDatabaseName()).append("\".\"").append(newTable.getName()).append("\"").append("\n");
SqliteIndexTypeEnum sqliteIndexTypeEnum = SqliteIndexTypeEnum.getByType(tableIndex.getType());
script.append("\t").append(sqliteIndexTypeEnum.buildModifyIndex(tableIndex)).append(",\n");
script.append("\t").append(sqliteIndexTypeEnum.buildModifyIndex(tableIndex)).append(";\n");
}
}

40
chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/type/SqliteColumnTypeEnum.java
View File

@ -14,14 +14,14 @@ import java.util.Map;
public enum SqliteColumnTypeEnum implements ColumnBuilder {
INTEGER("INTEGER", true, false, true, false, false, true, true, false, false, false),
INTEGER("INTEGER", true, false, true, false, false, true, false, false, false, false),
REAL("REAL", true, false, true, false, false, true, true, false, false, false),
REAL("REAL", true, false, true, false, false, true, false, false, false, false),
BLOB("BLOB", true, false, true, false, false, true, true, false, false, false),
BLOB("BLOB", true, false, true, false, false, true, false, false, false, false),
TEXT("TEXT", true, false, true, false, false, true, true, false, false, false),
TEXT("TEXT", true, false, true, false, false, true, false, false, false, false),
;
private ColumnType columnType;
@ -56,7 +56,7 @@ public enum SqliteColumnTypeEnum implements ColumnBuilder {
}
StringBuilder script = new StringBuilder();
script.append("`").append(column.getName()).append("`").append(" ");
script.append("\"").append(column.getName()).append("\"").append(" ");
script.append(buildDataType(column, type)).append(" ");
@ -72,7 +72,7 @@ public enum SqliteColumnTypeEnum implements ColumnBuilder {
script.append(buildAutoIncrement(column, type)).append(" ");
script.append(buildComment(column, type)).append(" ");
// script.append(buildComment(column, type)).append(" ");
return script.toString();
}
@ -94,19 +94,19 @@ public enum SqliteColumnTypeEnum implements ColumnBuilder {
@Override
public String buildModifyColumn(TableColumn tableColumn) {
if (EditStatus.DELETE.name().equals(tableColumn.getEditStatus())) {
return StringUtils.join("DROP COLUMN `", tableColumn.getName() + "`");
}
// if (EditStatus.DELETE.name().equals(tableColumn.getEditStatus())) {
// return StringUtils.join("DROP COLUMN \"", tableColumn.getName() + "\"");
// }
if (EditStatus.ADD.name().equals(tableColumn.getEditStatus())) {
return StringUtils.join("ADD COLUMN ", buildCreateColumnSql(tableColumn));
}
if (EditStatus.MODIFY.name().equals(tableColumn.getEditStatus())) {
if (!StringUtils.equalsIgnoreCase(tableColumn.getOldName(), tableColumn.getName())) {
return StringUtils.join("CHANGE COLUMN `", tableColumn.getOldName(), "` ", buildCreateColumnSql(tableColumn));
} else {
return StringUtils.join("MODIFY COLUMN ", buildCreateColumnSql(tableColumn));
}
return StringUtils.join("ADD ", buildCreateColumnSql(tableColumn));
}
// if (EditStatus.MODIFY.name().equals(tableColumn.getEditStatus())) {
// if (!StringUtils.equalsIgnoreCase(tableColumn.getOldName(), tableColumn.getName())) {
// return StringUtils.join("CHANGE COLUMN \"", tableColumn.getOldName(), "\" ", buildCreateColumnSql(tableColumn));
// } else {
// return StringUtils.join("MODIFY COLUMN ", buildCreateColumnSql(tableColumn));
// }
// }
return "";
}
@ -120,12 +120,6 @@ public enum SqliteColumnTypeEnum implements ColumnBuilder {
return "";
}
private String buildComment(TableColumn column, SqliteColumnTypeEnum type) {
if (!type.columnType.isSupportComments() || StringUtils.isEmpty(column.getComment())) {
return "";
}
return StringUtils.join("COMMENT '", column.getComment(), "'");
}
private String buildExt(TableColumn column, SqliteColumnTypeEnum type) {
if (!type.columnType.isSupportExtent() || StringUtils.isEmpty(column.getExtent())) {

11
chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/type/SqliteIndexTypeEnum.java
View File

@ -63,11 +63,10 @@ public enum SqliteIndexTypeEnum {
script.append(keyword).append(" ");
script.append(buildIndexName(tableIndex)).append(" ");
script.append(buildIndexName(tableIndex)).append(" ON ").append(tableIndex.getTableName()).append(" ");
script.append(buildIndexColumn(tableIndex)).append(" ");
script.append(buildIndexComment(tableIndex)).append(" ");
return script.toString();
}
@ -86,7 +85,7 @@ public enum SqliteIndexTypeEnum {
script.append("(");
for (TableIndexColumn column : tableIndex.getColumnList()) {
if(StringUtils.isNotBlank(column.getColumnName())) {
script.append("`").append(column.getColumnName()).append("`").append(",");
script.append("\"").append(column.getColumnName()).append("\"").append(",");
}
}
script.deleteCharAt(script.length() - 1);
@ -98,7 +97,7 @@ public enum SqliteIndexTypeEnum {
if(this.equals(PRIMARY_KEY)){
return "";
}else {
return "`"+tableIndex.getName()+"`";
return "\""+tableIndex.getName()+"\"";
}
}
@ -110,7 +109,7 @@ public enum SqliteIndexTypeEnum {
return StringUtils.join(buildDropIndex(tableIndex),",\n", "ADD ", buildIndexScript(tableIndex));
}
if (EditStatus.ADD.name().equals(tableIndex.getEditStatus())) {
return StringUtils.join("ADD ", buildIndexScript(tableIndex));
return StringUtils.join("CREATE ", buildIndexScript(tableIndex));
}
return "";
}
@ -119,7 +118,7 @@ public enum SqliteIndexTypeEnum {
if (SqliteIndexTypeEnum.PRIMARY_KEY.getName().equals(tableIndex.getType())) {
return StringUtils.join("DROP PRIMARY KEY");
}
return StringUtils.join("DROP INDEX `", tableIndex.getOldName(),"`");
return StringUtils.join("DROP INDEX \"", tableIndex.getOldName(),"\"");
}
public static List<IndexType> getIndexTypes() {
return Arrays.asList(SqliteIndexTypeEnum.values()).stream().map(SqliteIndexTypeEnum::getIndexType).collect(java.util.stream.Collectors.toList());