From b02fa04f1d71d282f705d433f20514d5039ca966 Mon Sep 17 00:00:00 2001
From: Jiri Pechanec <jpechane@redhat.com>
Date: Wed, 12 Jun 2024 06:34:31 +0200
Subject: [PATCH] DBZ-7640 Get row count if snapshot ordering required

---
 .../connector/binlog/BinlogSnapshotChangeEventSource.java      | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/debezium-connector-binlog/src/main/java/io/debezium/connector/binlog/BinlogSnapshotChangeEventSource.java b/debezium-connector-binlog/src/main/java/io/debezium/connector/binlog/BinlogSnapshotChangeEventSource.java
index 1fcf68295..6b3fd0a22 100644
--- a/debezium-connector-binlog/src/main/java/io/debezium/connector/binlog/BinlogSnapshotChangeEventSource.java
+++ b/debezium-connector-binlog/src/main/java/io/debezium/connector/binlog/BinlogSnapshotChangeEventSource.java
@@ -48,6 +48,7 @@
 import io.debezium.pipeline.notification.NotificationService;
 import io.debezium.pipeline.source.SnapshottingTask;
 import io.debezium.pipeline.source.spi.SnapshotChangeEventSource;
+import io.debezium.relational.RelationalDatabaseConnectorConfig.SnapshotTablesRowCountOrder;
 import io.debezium.relational.RelationalSnapshotChangeEventSource;
 import io.debezium.relational.RelationalTableFilters;
 import io.debezium.relational.Table;
@@ -548,7 +549,7 @@ protected OptionalLong rowCountForTable(TableId tableId) {
         if (getSnapshotSelectOverridesByTable(tableId, connectorConfig.getSnapshotSelectOverridesByTable()) != null) {
             return super.rowCountForTable(tableId);
         }
-        if (ROW_ESTIMATE_LOGGER.isInfoEnabled()) {
+        if (ROW_ESTIMATE_LOGGER.isInfoEnabled() || connectorConfig.snapshotOrderByRowCount() != SnapshotTablesRowCountOrder.DISABLED) {
             OptionalLong rowCount = connection.getEstimatedTableSize(tableId);
             LOGGER.info("Estimated row count for table {} is {}", tableId, rowCount);
             return rowCount;