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

fix sqlserver connection rest

Browse Source
This commit is contained in:
jipengfei-jpf
2023-08-07 23:51:15 +08:00
parent 9e8a7cae21
commit 155132275d
1 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/SqlServerMetaData.java
View File

@ -2,10 +2,13 @@ package ai.chat2db.plugin.sqlserver;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.List;
import ai.chat2db.spi.MetaData;
import ai.chat2db.spi.jdbc.DefaultMetaService;
import ai.chat2db.spi.model.Database;
import ai.chat2db.spi.sql.SQLExecutor;
import com.google.common.collect.Lists;
public class SqlServerMetaData extends DefaultMetaService implements MetaData {
private String functionSQL
@ -59,4 +62,24 @@ public class SqlServerMetaData extends DefaultMetaService implements MetaData {
return null;
});
}
@Override
public List<Database> databases(Connection connection) {
List<Database> databases = Lists.newArrayList();
SQLExecutor.getInstance().executeSql(connection, "SELECT name "
+ "FROM sys.databases",
resultSet -> {
try {
while (resultSet.next()) {
Database database = new Database();
String databaseName = resultSet.getString("name");
database.setName(databaseName);
databases.add(database);
}
} catch (SQLException e) {
throw new RuntimeException(e);
}
return null;
});
return databases;
}
}