Catch error that could cause LeaderScheduler thread to crash (#4850)

Signed-off-by: Taylor Gray <tylgry@amazon.com>
opensearch-project · Aug 21, 2024 · 292a225 · 292a225
1 parent 923145c
commit 292a225
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.../main/java/org/opensearch/dataprepper/plugins/source/dynamodb/leader/LeaderScheduler.java b/.../main/java/org/opensearch/dataprepper/plugins/source/dynamodb/leader/LeaderScheduler.java
@@ -127,7 +127,11 @@ public void run() {
                 if(leaderPartition != null) {
                     // Extend the timeout
                     // will always be a leader until shutdown
-                    coordinator.saveProgressStateForPartition(leaderPartition, Duration.ofMinutes(DEFAULT_EXTEND_LEASE_MINUTES));
+                    try {
+                        coordinator.saveProgressStateForPartition(leaderPartition, Duration.ofMinutes(DEFAULT_EXTEND_LEASE_MINUTES));
+                    } catch (final Exception e) {
+                        LOG.error("Failed to update ownership for leader partition. Retrying...");
+                    }
                 }
                 try {
                     Thread.sleep(leaseInterval.toMillis());