You are viewing a plain text version of this content. The canonical link for it is here.
Posted to oak-commits@jackrabbit.apache.org by mr...@apache.org on 2016/09/08 12:19:00 UTC

svn commit: r1759795 - in /jackrabbit/oak/trunk/oak-core/src: main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java

Author: mreutegg
Date: Thu Sep  8 12:19:00 2016
New Revision: 1759795

URL: http://svn.apache.org/viewvc?rev=1759795&view=rev
Log:
OAK-4770: Missing exception handling in ClusterNodeInfo.renewLease()

Add more tests. One ignored for the OAK-4770 issue.

Added:
    jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java   (with props)
Modified:
    jackrabbit/oak/trunk/oak-core/src/main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java

Modified: jackrabbit/oak/trunk/oak-core/src/main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java
URL: http://svn.apache.org/viewvc/jackrabbit/oak/trunk/oak-core/src/main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java?rev=1759795&r1=1759794&r2=1759795&view=diff
==============================================================================
--- jackrabbit/oak/trunk/oak-core/src/main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java (original)
+++ jackrabbit/oak/trunk/oak-core/src/main/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfo.java Thu Sep  8 12:19:00 2016
@@ -884,6 +884,10 @@ public class ClusterNodeInfo {
         return leaseTime;
     }
 
+    public long getLeaseEndTime() {
+        return leaseEndTime;
+    }
+
     public void setLeaseFailureHandler(LeaseFailureHandler leaseFailureHandler) {
         this.leaseFailureHandler = leaseFailureHandler;
     }

Added: jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java
URL: http://svn.apache.org/viewvc/jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java?rev=1759795&view=auto
==============================================================================
--- jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java (added)
+++ jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java Thu Sep  8 12:19:00 2016
@@ -0,0 +1,226 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.jackrabbit.oak.plugins.document;
+
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import org.apache.jackrabbit.oak.plugins.document.memory.MemoryDocumentStore;
+import org.apache.jackrabbit.oak.stats.Clock;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+public class ClusterNodeInfoTest {
+
+    private Clock clock;
+    private TestStore store;
+    private FailureHandler handler = new FailureHandler();
+
+    @Before
+    public void before() throws Exception {
+        clock = new Clock.Virtual();
+        clock.waitUntil(System.currentTimeMillis());
+        ClusterNodeInfo.setClock(clock);
+        store = new TestStore();
+    }
+
+    @After
+    public void after() throws Exception {
+        ClusterNodeInfo.resetClockToDefault();
+    }
+
+    @Test
+    public void renewLease() throws Exception {
+        ClusterNodeInfo info = newClusterNodeInfo(1);
+        long leaseEnd = info.getLeaseEndTime();
+        waitLeaseUpdateInterval();
+        assertTrue(info.renewLease());
+        assertTrue(info.getLeaseEndTime() > leaseEnd);
+        assertFalse(handler.isLeaseFailure());
+    }
+
+    @Test
+    public void renewLeaseExceptionBefore() throws Exception {
+        ClusterNodeInfo info = newClusterNodeInfo(1);
+        waitLeaseUpdateInterval();
+        store.setFailBeforeUpdate(1);
+        try {
+            info.renewLease();
+            fail("must fail with DocumentStoreException");
+        } catch (DocumentStoreException e) {
+            // expected
+        }
+        assertEquals(0, store.getFailBeforeUpdate());
+
+        long leaseEnd = info.getLeaseEndTime();
+        // must succeed next time
+        waitLeaseUpdateInterval();
+        assertTrue(info.renewLease());
+        assertTrue(info.getLeaseEndTime() > leaseEnd);
+        assertFalse(handler.isLeaseFailure());
+    }
+
+    // OAK-4770
+    @Ignore
+    @Test
+    public void renewLeaseExceptionAfter() throws Exception {
+        ClusterNodeInfo info = newClusterNodeInfo(1);
+        waitLeaseUpdateInterval();
+        store.setFailAfterUpdate(1);
+        try {
+            info.renewLease();
+            fail("must fail with DocumentStoreException");
+        } catch (DocumentStoreException e) {
+            // expected
+        }
+        assertEquals(0, store.getFailAfterUpdate());
+
+        long leaseEnd = info.getLeaseEndTime();
+        // must succeed next time
+        waitLeaseUpdateInterval();
+        assertTrue(info.renewLease());
+        assertTrue(info.getLeaseEndTime() > leaseEnd);
+        assertFalse(handler.isLeaseFailure());
+    }
+
+    @Test
+    public void renewLeaseDelayed() throws Exception {
+        ClusterNodeInfo info = newClusterNodeInfo(1);
+        clock.waitUntil(info.getLeaseEndTime() + ClusterNodeInfo.DEFAULT_LEASE_UPDATE_INTERVAL_MILLIS);
+        recoverClusterNode(1);
+        try {
+            assertFalse(info.renewLease());
+        } catch (AssertionError e) {
+            // TODO: expected !?
+        }
+        for (int i = 0; i < 10; i++) {
+            if (handler.isLeaseFailure()) {
+                return;
+            }
+            Thread.sleep(100);
+        }
+        fail("expected lease failure");
+    }
+
+    private void recoverClusterNode(int clusterId) throws Exception {
+        DocumentNodeStore ns = new DocumentMK.Builder()
+                .setDocumentStore(store.getStore()) // use unwrapped store
+                .setAsyncDelay(0).setClusterId(42).clock(clock).getNodeStore();
+        try {
+            LastRevRecoveryAgent recovery = new LastRevRecoveryAgent(ns);
+            recovery.recover(clusterId);
+        } finally {
+            ns.dispose();
+        }
+    }
+
+    private void waitLeaseUpdateInterval() throws Exception {
+        clock.waitUntil(clock.getTime() + ClusterNodeInfo.DEFAULT_LEASE_UPDATE_INTERVAL_MILLIS + 1);
+    }
+
+    private ClusterNodeInfo newClusterNodeInfo(int clusterId) {
+        ClusterNodeInfo info = ClusterNodeInfo.getInstance(store, clusterId);
+        info.setLeaseFailureHandler(handler);
+        assertTrue(info.renewLease()); // perform initial lease renew
+        return info;
+    }
+
+    static final class FailureHandler implements LeaseFailureHandler {
+
+        private final AtomicBoolean leaseFailure = new AtomicBoolean();
+
+        @Override
+        public void handleLeaseFailure() {
+            leaseFailure.set(true);
+        }
+
+        public boolean isLeaseFailure() {
+            return leaseFailure.get();
+        }
+    }
+
+    final class TestStore extends DocumentStoreWrapper {
+
+        private final AtomicInteger failBeforeUpdate = new AtomicInteger();
+        private final AtomicInteger failAfterUpdate = new AtomicInteger();
+        private long delayMillis;
+
+        TestStore() {
+            super(new MemoryDocumentStore());
+        }
+
+        DocumentStore getStore() {
+            return store;
+        }
+
+        @Override
+        public <T extends Document> T findAndUpdate(Collection<T> collection,
+                                                    UpdateOp update) {
+            maybeDelay();
+            maybeThrow(failBeforeUpdate, "update failed before");
+            T doc = super.findAndUpdate(collection, update);
+            maybeThrow(failAfterUpdate, "update failed after");
+            return doc;
+        }
+
+        private void maybeDelay() {
+            try {
+                clock.waitUntil(clock.getTime() + delayMillis);
+            } catch (InterruptedException e) {
+                throw new DocumentStoreException(e);
+            }
+        }
+
+        private void maybeThrow(AtomicInteger num, String msg) {
+            if (num.get() > 0) {
+                num.decrementAndGet();
+                throw new DocumentStoreException(msg);
+            }
+        }
+
+        public int getFailBeforeUpdate() {
+            return failBeforeUpdate.get();
+        }
+
+        public void setFailBeforeUpdate(int num) {
+            failBeforeUpdate.set(num);
+        }
+
+        public int getFailAfterUpdate() {
+            return failAfterUpdate.get();
+        }
+
+        public void setFailAfterUpdate(int num) {
+            this.failAfterUpdate.set(num);
+        }
+
+        public long getDelayMillis() {
+            return delayMillis;
+        }
+
+        public void setDelayMillis(long delayMillis) {
+            this.delayMillis = delayMillis;
+        }
+    }
+}

Propchange: jackrabbit/oak/trunk/oak-core/src/test/java/org/apache/jackrabbit/oak/plugins/document/ClusterNodeInfoTest.java
------------------------------------------------------------------------------
    svn:eol-style = native