Skip to content

Commit 1a06332

Browse files
committed
Fix kingbase8r6 error
1 parent 06e779d commit 1a06332

3 files changed

Lines changed: 13 additions & 4 deletions

File tree

chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/DatabaseServiceImpl.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,8 @@ private void sortSchema(List<Schema> schemas, Connection connection) {
8989
// If the database name contains the name of the current database, the current database is placed in the first place
9090
int num = -1;
9191
for (int i = 0; i < schemas.size(); i++) {
92-
if (StringUtils.isNotBlank(ulr) && ulr.contains(schemas.get(i).getName())) {
92+
String schema = schemas.get(i).getName();
93+
if (StringUtils.isNotBlank(ulr) && schema!=null && ulr.contains(schema)) {
9394
num = i;
9495
break;
9596
}

chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/jdbc/DefaultMetaService.java

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,15 @@ public List<Database> databases(Connection connection) {
2626

2727
@Override
2828
public List<Schema> schemas(Connection connection, String databaseName) {
29-
return SQLExecutor.getInstance().schemas(connection, databaseName, null);
29+
List<Schema> schemas = SQLExecutor.getInstance().schemas(connection, databaseName, null);
30+
if(StringUtils.isNotBlank(databaseName) && CollectionUtils.isNotEmpty(schemas)){
31+
for ( Schema schema : schemas) {
32+
if(StringUtils.isBlank(schema.getDatabaseName())){
33+
schema.setDatabaseName(databaseName);
34+
}
35+
}
36+
}
37+
return schemas;
3038
}
3139

3240
@Override

chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/Schema.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@ public class Schema implements Serializable {
2424
/**
2525
* databaseName
2626
*/
27-
@JsonAlias({"TABLE_CATALOG"})
27+
@JsonAlias({"TABLE_CATALOG","table_catalog"})
2828
private String databaseName;
2929
/**
3030
* 数据名字
3131
*/
32-
@JsonAlias({"TABLE_SCHEM"})
32+
@JsonAlias({"TABLE_SCHEM","table_schem"})
3333
private String name;
3434

3535

0 commit comments

Comments
 (0)