For segrep enabled indices, use NRTReplicationEngine for replica shar…

…ds (#486) * Update engine factory to return NRTReplicationEngine for replica shards Signed-off-by: Suraj Singh <[email protected]> * Address review comments Signed-off-by: Suraj Singh <[email protected]> Signed-off-by: Suraj Singh <[email protected]> (cherry picked from commit 81c2002)
opensearch-project · Aug 29, 2022 · c2bd1ba · c2bd1ba
1 parent 61aaa49
commit c2bd1ba
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/main/kotlin/org/opensearch/replication/ReplicationPlugin.kt b/src/main/kotlin/org/opensearch/replication/ReplicationPlugin.kt
@@ -143,6 +143,9 @@ import org.opensearch.watcher.ResourceWatcherService
 import java.util.Optional
 import java.util.function.Supplier
 
+import org.opensearch.index.engine.NRTReplicationEngine
+
+
 @OpenForTesting
 internal class ReplicationPlugin : Plugin(), ActionPlugin, PersistentTaskPlugin, RepositoryPlugin, EnginePlugin {
 
@@ -359,7 +362,14 @@ internal class ReplicationPlugin : Plugin(), ActionPlugin, PersistentTaskPlugin,
 
     override fun getEngineFactory(indexSettings: IndexSettings): Optional<EngineFactory> {
         return if (indexSettings.settings.get(REPLICATED_INDEX_SETTING.key) != null) {
-            Optional.of(EngineFactory { config -> ReplicationEngine(config) })
+            Optional.of(EngineFactory { config ->
+                // Use NRTSegmentReplicationEngine for SEGMENT replication type indices replica shards
+                if (config.isReadOnlyReplica) {
+                    NRTReplicationEngine(config)
+                } else {
+                    ReplicationEngine(config)
+                }
+            })
         } else {
             Optional.empty()
         }