refactor: simplify offset update

Co-authored-by: Christophe Camel <camel.christophe@gmail.com>
okp4 · Apr 8, 2022 · a6e4829 · a6e4829
1 parent b4da165
commit a6e4829
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/com/okp4/connect/cosmos/CosmosSourceTask.kt b/src/main/kotlin/com/okp4/connect/cosmos/CosmosSourceTask.kt
@@ -58,8 +58,8 @@ class CosmosSourceTask : SourceTask() {
                     .toList()
             }
         }.also {
-            it.takeIf { it.isNotEmpty() }?.let { records ->
-                lastBlockHeightFromOffsetStorage = records[records.size - 1].sourceOffset()[HEIGHT_FIELD] as Long
+            it.lastOrNull()?.run {
+                lastBlockHeightFromOffsetStorage = sourceOffset()[HEIGHT_FIELD] as Long
             }
         }