Fix SharedClusterSnapshotRestoreIT and RestoreSnapshotIT

Signed-off-by: Sachin Kale <kalsac@amazon.com>
soosinha · Sep 6, 2023 · 62feea0 · 62feea0
1 parent fdd357c
commit 62feea0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/server/src/main/java/org/opensearch/index/engine/NRTReplicationEngine.java b/server/src/main/java/org/opensearch/index/engine/NRTReplicationEngine.java
@@ -24,6 +24,7 @@
 import org.opensearch.index.seqno.SeqNoStats;
 import org.opensearch.index.seqno.SequenceNumbers;
 import org.opensearch.index.translog.Translog;
+import org.opensearch.index.translog.TranslogCorruptedException;
 import org.opensearch.index.translog.TranslogDeletionPolicy;
 import org.opensearch.index.translog.TranslogException;
 import org.opensearch.index.translog.TranslogManager;
@@ -127,8 +128,9 @@ public void onAfterTranslogSync() {
                 engineConfig.getPrimaryModeSupplier()
             );
             this.translogManager = translogManagerRef;
-        } catch (IOException e) {
-            IOUtils.closeWhileHandlingException(store::decRef, readerManager, translogManagerRef);
+        } catch (IOException | TranslogCorruptedException e) {
+            store.decRef();
+            IOUtils.closeWhileHandlingException(readerManager, translogManagerRef);
             throw new EngineCreationFailureException(shardId, "failed to create engine", e);
         }
     }