feat:support get table trigger from oracle

This commit is contained in:
vran 2022-05-17 15:39:30 +08:00
parent 348a6e85dd
commit ae8dc7ebcc
2 changed files with 82 additions and 0 deletions

View File

@ -3,6 +3,7 @@ package com.databasir.core.meta.provider;
import com.databasir.core.meta.provider.jdbc.*; import com.databasir.core.meta.provider.jdbc.*;
import com.databasir.core.meta.provider.maria.MariaTriggerMetaProvider; import com.databasir.core.meta.provider.maria.MariaTriggerMetaProvider;
import com.databasir.core.meta.provider.mysql.MysqlTableTriggerMetaProvider; import com.databasir.core.meta.provider.mysql.MysqlTableTriggerMetaProvider;
import com.databasir.core.meta.provider.oracle.OracleTriggerMetaProvider;
import com.databasir.core.meta.provider.postgresql.PostgresqlTriggerMetaProvider; import com.databasir.core.meta.provider.postgresql.PostgresqlTriggerMetaProvider;
import com.databasir.core.meta.provider.sqlserver.SqlServerColumnMetaProvider; import com.databasir.core.meta.provider.sqlserver.SqlServerColumnMetaProvider;
import com.databasir.core.meta.provider.sqlserver.SqlServerTableMetaProvider; import com.databasir.core.meta.provider.sqlserver.SqlServerTableMetaProvider;
@ -51,6 +52,9 @@ public class MetaProviders {
if (url.contains(":mariadb:")) { if (url.contains(":mariadb:")) {
return mariaDB(); return mariaDB();
} }
if (url.contains(":oracle:")) {
return oracle();
}
return jdbc(); return jdbc();
} }
@ -109,4 +113,18 @@ public class MetaProviders {
); );
return new JdbcDatabaseMetaProvider(tableMetaProvider); return new JdbcDatabaseMetaProvider(tableMetaProvider);
} }
private static DatabaseMetaProvider oracle() {
var columnMetaProvider = new JdbcColumnMetaProvider();
var foreignKeyMetaProvider = new JdbcForeignKeyMetaProvider();
var indexMetaProvider = new JdbcIndexMetaProvider();
var triggerMetaProvider = new OracleTriggerMetaProvider();
var tableMetaProvider = new JdbcTableMetaProvider(
columnMetaProvider,
indexMetaProvider,
triggerMetaProvider,
foreignKeyMetaProvider
);
return new JdbcDatabaseMetaProvider(tableMetaProvider);
}
} }

View File

@ -0,0 +1,64 @@
package com.databasir.core.meta.provider.oracle;
import com.databasir.core.meta.data.TriggerMeta;
import com.databasir.core.meta.provider.TriggerMetaProvider;
import com.databasir.core.meta.provider.condition.TableCondition;
import lombok.extern.slf4j.Slf4j;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
@Slf4j
public class OracleTriggerMetaProvider implements TriggerMetaProvider {
@Override
public List<TriggerMeta> selectTriggers(Connection connection, TableCondition condition) {
String sql = "SELECT trig.table_owner AS schema_name,\n"
+ " trig.table_name,\n"
+ " trig.owner AS trigger_schema_name,\n"
+ " trig.trigger_name,\n"
+ " trig.trigger_type,\n"
+ " trig.triggering_event,\n"
+ " trig.status,\n"
+ " trig.trigger_body AS script\n"
+ "FROM sys.all_triggers trig\n"
+ " INNER JOIN sys.all_tables tab ON trig.table_owner = tab.owner\n"
+ " AND trig.table_name = tab.table_name\n"
+ "WHERE trig.base_object_type = 'TABLE' AND trig.owner = ? AND trig.TABLE_NAME = ?";
try {
PreparedStatement preparedStatement = connection.prepareStatement(sql);
preparedStatement.setObject(1, condition.getSchemaName());
preparedStatement.setObject(2, condition.getTableName());
ResultSet results = preparedStatement.executeQuery();
List<TriggerMeta> triggers = new ArrayList<>();
while (results.next()) {
String status = results.getString("status");
String name = Objects.requireNonNullElse(results.getString("trigger_name"), "")
+ " ("
+ status
+ ")";
String statement = results.getString("script");
String timing = results.getString("trigger_type");
String manipulation = results.getString("triggering_event");
String created = "unknown";
TriggerMeta meta = TriggerMeta.builder()
.name(name)
.manipulation(manipulation)
.timing(timing)
.statement(statement)
.createAt(created)
.build();
triggers.add(meta);
}
return triggers;
} catch (SQLException e) {
log.warn("get trigger meta failed", e);
return Collections.emptyList();
}
}
}