From d71858b095649a842b2e307eb3742a1e8a5ecc93 Mon Sep 17 00:00:00 2001
From: mfvitale <mvitale@redhat.com>
Date: Mon, 18 Mar 2024 16:47:53 +0100
Subject: [PATCH] DBZ-7308 tableLockingStatement from SnapshotLock interface
 now takes just one table in input

---
 .../debezium/connector/db2/Db2SnapshotChangeEventSource.java  | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/main/java/io/debezium/connector/db2/Db2SnapshotChangeEventSource.java b/src/main/java/io/debezium/connector/db2/Db2SnapshotChangeEventSource.java
index b0edb3b..7ca4ff4 100644
--- a/src/main/java/io/debezium/connector/db2/Db2SnapshotChangeEventSource.java
+++ b/src/main/java/io/debezium/connector/db2/Db2SnapshotChangeEventSource.java
@@ -11,7 +11,6 @@
 import java.sql.Savepoint;
 import java.sql.Statement;
 import java.util.List;
-import java.util.Map;
 import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -32,7 +31,6 @@
 import io.debezium.relational.Tables;
 import io.debezium.schema.SchemaChangeEvent;
 import io.debezium.snapshot.SnapshotterService;
-import io.debezium.spi.snapshot.Snapshotter;
 import io.debezium.util.Clock;
 
 public class Db2SnapshotChangeEventSource extends RelationalSnapshotChangeEventSource<Db2Partition, Db2OffsetContext> {
@@ -90,7 +88,7 @@ else if (connectorConfig.getSnapshotIsolationMode() == SnapshotIsolationMode.EXC
                     }
 
                     Optional<String> lockingStatement = snapshotterService.getSnapshotLock().tableLockingStatement(connectorConfig.snapshotLockTimeout(),
-                            Set.of(quoteTableName(tableId)));
+                            quoteTableName(tableId));
 
                     if (lockingStatement.isPresent()) {
                         LOGGER.info("Locking table {}", tableId);