From 8dbb2bf00538694b0fd2f1bef37e5d8aa52a96f4 Mon Sep 17 00:00:00 2001
From: mfvitale <mvitale@redhat.com>
Date: Fri, 5 Apr 2024 11:54:03 +0200
Subject: [PATCH] DBZ-7732 Add connector specific annotation to SnapshotLock
 and SnapshotQuery implementations

---
 .../connector/db2/snapshot/lock/ExclusiveSnapshotLock.java     | 3 +++
 .../debezium/connector/db2/snapshot/lock/NoSnapshotLock.java   | 3 +++
 .../connector/db2/snapshot/query/SelectAllSnapshotQuery.java   | 3 +++
 3 files changed, 9 insertions(+)

diff --git a/src/main/java/io/debezium/connector/db2/snapshot/lock/ExclusiveSnapshotLock.java b/src/main/java/io/debezium/connector/db2/snapshot/lock/ExclusiveSnapshotLock.java
index 2f58272..7827085 100644
--- a/src/main/java/io/debezium/connector/db2/snapshot/lock/ExclusiveSnapshotLock.java
+++ b/src/main/java/io/debezium/connector/db2/snapshot/lock/ExclusiveSnapshotLock.java
@@ -9,9 +9,12 @@
 import java.util.Map;
 import java.util.Optional;
 
+import io.debezium.annotation.ConnectorSpecific;
+import io.debezium.connector.db2.Db2Connector;
 import io.debezium.connector.db2.Db2ConnectorConfig;
 import io.debezium.snapshot.spi.SnapshotLock;
 
+@ConnectorSpecific(connector = Db2Connector.class)
 public class ExclusiveSnapshotLock implements SnapshotLock {
 
     @Override
diff --git a/src/main/java/io/debezium/connector/db2/snapshot/lock/NoSnapshotLock.java b/src/main/java/io/debezium/connector/db2/snapshot/lock/NoSnapshotLock.java
index ab4294d..e170faf 100644
--- a/src/main/java/io/debezium/connector/db2/snapshot/lock/NoSnapshotLock.java
+++ b/src/main/java/io/debezium/connector/db2/snapshot/lock/NoSnapshotLock.java
@@ -9,9 +9,12 @@
 import java.util.Map;
 import java.util.Optional;
 
+import io.debezium.annotation.ConnectorSpecific;
+import io.debezium.connector.db2.Db2Connector;
 import io.debezium.connector.db2.Db2ConnectorConfig;
 import io.debezium.snapshot.spi.SnapshotLock;
 
+@ConnectorSpecific(connector = Db2Connector.class)
 public class NoSnapshotLock implements SnapshotLock {
 
     @Override
diff --git a/src/main/java/io/debezium/connector/db2/snapshot/query/SelectAllSnapshotQuery.java b/src/main/java/io/debezium/connector/db2/snapshot/query/SelectAllSnapshotQuery.java
index 1238203..0242005 100644
--- a/src/main/java/io/debezium/connector/db2/snapshot/query/SelectAllSnapshotQuery.java
+++ b/src/main/java/io/debezium/connector/db2/snapshot/query/SelectAllSnapshotQuery.java
@@ -9,9 +9,12 @@
 import java.util.Map;
 import java.util.Optional;
 
+import io.debezium.annotation.ConnectorSpecific;
 import io.debezium.config.CommonConnectorConfig;
+import io.debezium.connector.db2.Db2Connector;
 import io.debezium.snapshot.spi.SnapshotQuery;
 
+@ConnectorSpecific(connector = Db2Connector.class)
 public class SelectAllSnapshotQuery implements SnapshotQuery {
 
     @Override