From b51bec5b76999e917cd857fafa621bca238b3b5f Mon Sep 17 00:00:00 2001 From: harveyyue Date: Thu, 14 Dec 2023 11:03:41 +0800 Subject: [PATCH] DBZ-7167 Provide a public API from the connector implementations to retrieve the list of matching collections or tables based on the different include-/exclude lists --- .../io/debezium/connector/vitess/VitessConnector.java | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/main/java/io/debezium/connector/vitess/VitessConnector.java b/src/main/java/io/debezium/connector/vitess/VitessConnector.java index 1b27a163..f22bb414 100644 --- a/src/main/java/io/debezium/connector/vitess/VitessConnector.java +++ b/src/main/java/io/debezium/connector/vitess/VitessConnector.java @@ -32,7 +32,7 @@ import io.debezium.connector.common.RelationalBaseSourceConnector; import io.debezium.connector.vitess.connection.VitessReplicationConnection; import io.debezium.relational.RelationalDatabaseConnectorConfig; -import io.debezium.rest.model.DataCollection; +import io.debezium.relational.TableId; import io.debezium.util.Strings; import io.grpc.StatusRuntimeException; import io.vitess.proto.Query; @@ -393,13 +393,16 @@ protected Map validateAllFields(Configuration config) { return results; } + @SuppressWarnings("unchecked") @Override - public List getMatchingCollections(Configuration configuration) { + 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 DataCollection(keyspace, table)).collect(Collectors.toList()); + return includedTables.stream() + .map(table -> new TableId(keyspace, null, table)) + .collect(Collectors.toList()); } }