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

Merge pull request #1181 from openai0229/issue#1135

Browse Source 
实现MySQL存储过程修改
This commit is contained in:
ji
2024-02-26 10:09:28 +08:00
committed by GitHub
parent fe3ad6673e 80634b5107
commit 083600603f
11 changed files with 149 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java
View File

@ -1,14 +1,24 @@
package ai.chat2db.plugin.mysql;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.model.Procedure;
import ai.chat2db.spi.sql.SQLExecutor;
import org.springframework.util.StringUtils;
import java.sql.Connection;
import java.sql.SQLException;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.sql.SQLExecutor;
import org.springframework.util.StringUtils;
public class MysqlDBManage extends DefaultDBManage implements DBManage {
@Override
public void updateProcedure(Connection connection, String databaseName, String schemaName, Procedure procedure) {
String sql = "DROP PROCEDURE " + procedure.getProcedureName();
SQLExecutor.getInstance().execute(connection, sql, resultSet -> {});
String procedureBody = procedure.getProcedureBody();
SQLExecutor.getInstance().execute(connection, procedureBody, resultSet -> {});
}
@Override
public void connectDatabase(Connection connection, String database) {
if (StringUtils.isEmpty(database)) {
@ -22,7 +32,6 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
}
@Override
public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
String sql = "DROP TABLE "+ format(tableName);

42
chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlMetaData.java
View File

@ -1,12 +1,5 @@
package ai.chat2db.plugin.mysql;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.*;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import ai.chat2db.plugin.mysql.builder.MysqlSqlBuilder;
import ai.chat2db.plugin.mysql.type.*;
import ai.chat2db.spi.MetaData;
@ -18,15 +11,22 @@ import ai.chat2db.spi.sql.SQLExecutor;
import jakarta.validation.constraints.NotEmpty;
import org.apache.commons.lang3.StringUtils;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.*;
import java.util.stream.Collectors;
import static ai.chat2db.spi.util.SortUtils.sortDatabase;
public class MysqlMetaData extends DefaultMetaService implements MetaData {
private List<String> systemDatabases = Arrays.asList("information_schema", "performance_schema", "mysql", "sys");
@Override
public List<Database> databases(Connection connection) {
List<Database> databases = SQLExecutor.getInstance().databases(connection);
return sortDatabase(databases,systemDatabases,connection);
return sortDatabase(databases, systemDatabases, connection);
}
@ -116,19 +116,25 @@ public class MysqlMetaData extends DefaultMetaService implements MetaData {
@Override
public Procedure procedure(Connection connection, @NotEmpty String databaseName, String schemaName,
String procedureName) {
String sql = String.format(ROUTINES_SQL, "PROCEDURE", databaseName, procedureName);
return SQLExecutor.getInstance().execute(connection, sql, resultSet -> {
Procedure procedure = new Procedure();
procedure.setDatabaseName(databaseName);
procedure.setSchemaName(schemaName);
procedure.setProcedureName(procedureName);
String routinesSql = String.format(ROUTINES_SQL, "PROCEDURE", databaseName, procedureName);
String showCreateProcedureSql = "SHOW CREATE PROCEDURE " + procedureName;
Procedure procedure = SQLExecutor.getInstance().execute(connection, routinesSql, resultSet -> {
Procedure p = new Procedure();
p.setDatabaseName(databaseName);
p.setSchemaName(schemaName);
p.setProcedureName(procedureName);
if (resultSet.next()) {
procedure.setSpecificName(resultSet.getString("SPECIFIC_NAME"));
procedure.setRemarks(resultSet.getString("ROUTINE_COMMENT"));
procedure.setProcedureBody(resultSet.getString("ROUTINE_DEFINITION"));
p.setSpecificName(resultSet.getString("SPECIFIC_NAME"));
p.setRemarks(resultSet.getString("ROUTINE_COMMENT"));
}
return procedure;
return p;
});
SQLExecutor.getInstance().execute(connection, showCreateProcedureSql, resultSet -> {
if (resultSet.next()) {
procedure.setProcedureBody(resultSet.getString("Create Procedure"));
}
});
return procedure;
}
private static String SELECT_TABLE_COLUMNS = "SELECT * FROM information_schema.COLUMNS WHERE TABLE_SCHEMA = '%s' AND TABLE_NAME = '%s' order by ORDINAL_POSITION";