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

Delete operations of functions

Browse Source
This commit is contained in:
suyue
2024-08-03 14:39:09 +08:00
parent 052a36b73e
commit 89e2eb30de
11 changed files with 173 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLDBManage.java
View File

@ -3,6 +3,7 @@ package ai.chat2db.plugin.postgresql;
import ai.chat2db.spi.DBManage;
import ai.chat2db.spi.jdbc.DefaultDBManage;
import ai.chat2db.spi.model.AsyncContext;
import ai.chat2db.spi.model.Function;
import ai.chat2db.spi.model.Procedure;
import ai.chat2db.spi.sql.Chat2DBContext;
import ai.chat2db.spi.sql.ConnectInfo;
@ -209,6 +210,15 @@ public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
SQLExecutor.getInstance().execute(connection, sql, resultSet -> {});
}
@Override
public void deleteFunction(Connection connection, String databaseName, String schemaName, Function function) {
String functionBody = function.getFunctionBody();
String parameterSignature = extractParameterSignature(functionBody);
String functionNewName = getSchemaOrFunctionName(functionBody, schemaName, function);
String sql = "DROP FUNCTION" + functionNewName + parameterSignature;
SQLExecutor.getInstance().execute(connection,sql,resultSet -> {});
}
private String extractParameterSignature(String input) {
int depth = 0, start = -1;
for (int i = 0; i < input.length(); i++) {
@ -232,4 +242,12 @@ public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
return schemaName + "." + procedure.getProcedureName();
}
}
private static String getSchemaOrFunctionName(String functionBody, String schemaName, Function function) {
if (functionBody.toLowerCase().contains(schemaName.toLowerCase())) {
return function.getFunctionName();
} else {
return schemaName + "." + function.getFunctionName();
}
}
}