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

Fix bug with invalid schema specification

Browse Source
This commit is contained in:
SwallowGG
2023-08-29 23:38:24 +08:00
parent 656cd3a338
commit 9b659016f6
15 changed files with 269 additions and 241 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -87,12 +87,42 @@ public class DatabaseServiceImpl implements DatabaseService {
(key) -> param.isRefresh(), (key) -> {
Connection connection = param.getConnection() == null ? Chat2DBContext.getConnection()
: param.getConnection();
MetaData metaData = Chat2DBContext.getMetaData();
return metaData.schemas(connection, param.getDataBaseName());
return getSchemaList(param.getDataBaseName(), connection);
});
return ListResult.of(schemas);
}
private List<Schema> getSchemaList(String databaseName, Connection connection) {
MetaData metaData = Chat2DBContext.getMetaData();
List<Schema> schemas = metaData.schemas(connection,databaseName);
sortSchema(schemas,connection);
return schemas;
}
private void sortSchema(List<Schema> schemas,Connection connection) {
if (CollectionUtils.isEmpty(schemas)) {
return;
}
String ulr = null;
try {
ulr = connection.getMetaData().getURL();
} catch (SQLException e) {
log.error("get url error", e);
}
// If the database name contains the name of the current database, the current database is placed in the first place
int num = -1;
for (int i = 0; i < schemas.size(); i++) {
if (StringUtils.isNotBlank(ulr) && ulr.contains(schemas.get(i).getName())) {
num = i;
break;
}
}
if (num != -1 && num != 0) {
Collections.swap(schemas, num, 0);
}
}
@Override
public DataResult<MetaSchema> queryDatabaseSchema(MetaDataQueryParam param) {
MetaSchema metaSchema = new MetaSchema();