diff --git a/src/main/java/io/debezium/connector/vitess/VitessConnector.java b/src/main/java/io/debezium/connector/vitess/VitessConnector.java index da6e8920..f22bb414 100644 --- a/src/main/java/io/debezium/connector/vitess/VitessConnector.java +++ b/src/main/java/io/debezium/connector/vitess/VitessConnector.java @@ -32,6 +32,7 @@ import io.debezium.connector.common.RelationalBaseSourceConnector; import io.debezium.connector.vitess.connection.VitessReplicationConnection; import io.debezium.relational.RelationalDatabaseConnectorConfig; +import io.debezium.relational.TableId; import io.debezium.util.Strings; import io.grpc.StatusRuntimeException; import io.vitess.proto.Query; @@ -391,4 +392,17 @@ protected Map validateAllFields(Configuration config) { } return results; } + + @SuppressWarnings("unchecked") + @Override + public List getMatchingCollections(Configuration configuration) { + VitessConnectorConfig vitessConnectorConfig = new VitessConnectorConfig(configuration); + String keyspace = vitessConnectorConfig.getKeyspace(); + List allTables = getKeyspaceTables(vitessConnectorConfig); + List includedTables = getIncludedTables(keyspace, + vitessConnectorConfig.tableIncludeList(), allTables); + return includedTables.stream() + .map(table -> new TableId(keyspace, null, table)) + .collect(Collectors.toList()); + } }