DBZ-7903 Align to core changes

debezium · Oct 31, 2024 · b321419 · b321419
1 parent 25d4f0c
commit b321419
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/io/debezium/connector/db2/Db2OffsetContext.java b/src/main/java/io/debezium/connector/db2/Db2OffsetContext.java
@@ -119,8 +119,8 @@ public Loader(Db2ConnectorConfig connectorConfig) {
         public Db2OffsetContext load(Map<String, ?> offset) {
             final Lsn changeLsn = Lsn.valueOf((String) offset.get(SourceInfo.CHANGE_LSN_KEY));
             final Lsn commitLsn = Lsn.valueOf((String) offset.get(SourceInfo.COMMIT_LSN_KEY));
-            final SnapshotType snapshot = loadSnapshot((Map<String, Object>) offset);
-            boolean snapshotCompleted = Boolean.TRUE.equals(offset.get(SNAPSHOT_COMPLETED_KEY));
+            final SnapshotType snapshot = loadSnapshot(offset).orElse(null);
+            boolean snapshotCompleted = loadSnapshotCompleted(offset);
 
             // only introduced in 0.10.Beta1, so it might be not present when upgrading from earlier versions
             Long eventSerialNo = ((Long) offset.get(EVENT_SERIAL_NO_KEY));