From 9cd58d3733402e91453207a65294411f0215e09e Mon Sep 17 00:00:00 2001
From: Marc Handalian <handalm@amazon.com>
Date: Mon, 28 Aug 2023 22:15:22 -0700
Subject: [PATCH] Update ReadOnlyEngine inside of resetEngineToGlobalCheckpoint
 to implement getSegmentInfosSnapshot. This ensures access to this function is
 not permitted on the ReadOnlyEngine and is delegated to the new IE once
 opened.

Signed-off-by: Marc Handalian <handalm@amazon.com>
---
 .../java/org/opensearch/index/shard/IndexShard.java    | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server/src/main/java/org/opensearch/index/shard/IndexShard.java b/server/src/main/java/org/opensearch/index/shard/IndexShard.java
index 1d0184de9d93c..136f93fdc1e21 100644
--- a/server/src/main/java/org/opensearch/index/shard/IndexShard.java
+++ b/server/src/main/java/org/opensearch/index/shard/IndexShard.java
@@ -4582,6 +4582,16 @@ public GatedCloseable<IndexCommit> acquireSafeIndexCommit() {
                     }
                 }
 
+                @Override
+                public GatedCloseable<SegmentInfos> getSegmentInfosSnapshot() {
+                    synchronized (engineMutex) {
+                        if (newEngineReference.get() == null) {
+                            throw new AlreadyClosedException("engine was closed");
+                        }
+                        return newEngineReference.get().getSegmentInfosSnapshot();
+                    }
+                }
+
                 @Override
                 public void close() throws IOException {
                     assert Thread.holdsLock(engineMutex);