close if failed to install reference

dnhatn · dnhatn · commit 67badc6621b9 · 2019-08-14T12:14:47.000-04:00
diff --git a/server/src/main/java/org/elasticsearch/index/shard/IndexShard.java b/server/src/main/java/org/elasticsearch/index/shard/IndexShard.java
@@ -1606,21 +1606,19 @@ private void innerOpenEngineAndTranslog(LongSupplier globalCheckpointSupplier) t
         synchronized (engineMutex) {
             // we must create a new engine under mutex (see IndexShard#snapshotStoreMetadata).
             final Engine newEngine = engineFactory.newReadWriteEngine(config);
-            boolean success = false;
-            try {
-                synchronized (mutex) {
+            synchronized (mutex) {
+                try {
                     verifyNotClosed();
                     assert currentEngineReference.get() == null : "engine is running";
                     onNewEngine(newEngine);
                     currentEngineReference.set(newEngine);
                     // We set active because we are now writing operations to the engine; this way,
                     // if we go idle after some time and become inactive, we still give sync'd flush a chance to run.
                     active.set(true);
-                    success = true;
-                }
-            } finally {
-                if (success == false) {
-                    newEngine.close();
+                } finally {
+                    if (currentEngineReference.get() != newEngine) {
+                        newEngine.close();
+                    }
                 }
             }
         }
@@ -3372,16 +3370,14 @@ public void close() throws IOException {
                         IOUtils.close(super::close, newEngine);
                     }
                 };
-            boolean success = false;
-            try {
-                synchronized (mutex) {
+            synchronized (mutex) {
+                try {
                     verifyNotClosed();
                     IOUtils.close(currentEngineReference.getAndSet(readOnlyEngine));
-                    success = true;
-                }
-            } finally {
-                if (success == false) {
-                    readOnlyEngine.close();
+                } finally {
+                    if (currentEngineReference.get() != readOnlyEngine) {
+                        readOnlyEngine.close();
+                    }
                 }
             }
             newEngineReference.set(engineFactory.newReadWriteEngine(newEngineConfig(replicationTracker)));
