Examples with RetryFactory org.apache.accumulo.fate.util.Retry.RetryFactory used on opensource projects

Example 1 with RetryFactory

use of org.apache.accumulo.fate.util.Retry.RetryFactory in project accumulo by apache.

the class ZooReaderWriterTest method testMutateNodeCreationFails.

@Test
public void testMutateNodeCreationFails() throws Exception {
    final String path = "/foo";
    final byte[] value = { 0 };
    Mutator mutator = currentValue -> new byte[] { 1 };
    zk.create(path, value, ZooUtil.PUBLIC, CreateMode.PERSISTENT);
    expectLastCall().andThrow(new SessionExpiredException()).once();
    expect(retry.canRetry()).andReturn(false);
    expect(retry.retriesCompleted()).andReturn(1L).once();
    replay(zk, zrw, retryFactory, retry);
    assertThrows(SessionExpiredException.class, () -> zrw.mutateOrCreate(path, value, mutator));
}

Example 2 with RetryFactory

use of org.apache.accumulo.fate.util.Retry.RetryFactory in project accumulo by apache.

the class ZooReaderWriterTest method testMutateWithBadVersion.

@Test
public void testMutateWithBadVersion() throws Exception {
    final String path = "/foo";
    final byte[] value = { 0 };
    final byte[] mutatedBytes = { 1 };
    Mutator mutator = currentValue -> mutatedBytes;
    zrw = createMockBuilder(ZooReaderWriter.class).addMockedMethods("getRetryFactory", "getZooKeeper").createMock();
    expect(zrw.getRetryFactory()).andReturn(retryFactory).anyTimes();
    expect(zrw.getZooKeeper()).andReturn(zk).anyTimes();
    Stat stat = new Stat();
    zk.create(path, value, ZooUtil.PUBLIC, CreateMode.PERSISTENT);
    expectLastCall().andThrow(new NodeExistsException()).once();
    expect(zk.getData(path, null, stat)).andReturn(new byte[] { 3 }).times(2);
    // BadVersionException should retry
    expect(zk.setData(path, mutatedBytes, 0)).andThrow(new BadVersionException());
    // Let 2nd setData succeed
    expect(zk.setData(path, mutatedBytes, 0)).andReturn(null);
    replay(zk, zrw, retryFactory, retry);
    assertArrayEquals(new byte[] { 1 }, zrw.mutateOrCreate(path, value, mutator));
    verify(zk, zrw, retryFactory, retry);
}

Example 3 with RetryFactory

use of org.apache.accumulo.fate.util.Retry.RetryFactory in project accumulo by apache.

the class RetryTest method properArgumentsInUnlimitedRetry.

@Test
public void properArgumentsInUnlimitedRetry() {
    long startWait = 50L, maxWait = 5000L, waitIncrement = 500L, logInterval = 10000L;
    double waitFactor = 1.0;
    RetryFactory factory = Retry.builder().infiniteRetries().retryAfter(startWait, MS).incrementBy(waitIncrement, MS).maxWait(maxWait, MS).backOffFactor(waitFactor).logInterval(logInterval, MS).createFactory();
    Retry retry = factory.createRetry();
    assertEquals(-1, retry.getMaxRetries());
    assertEquals(startWait, retry.getCurrentWait());
    assertEquals(maxWait, retry.getMaxWait());
    assertEquals(waitIncrement, retry.getWaitIncrement());
    assertEquals(logInterval, retry.getLogInterval());
}

Example 4 with RetryFactory

use of org.apache.accumulo.fate.util.Retry.RetryFactory in project accumulo by apache.

the class ZooReaderWriterTest method testMutateWithRetryOnSetData.

@Test
public void testMutateWithRetryOnSetData() throws Exception {
    final String path = "/foo";
    final byte[] value = { 0 };
    final byte[] mutatedBytes = { 1 };
    Mutator mutator = currentValue -> mutatedBytes;
    zrw = createMockBuilder(ZooReaderWriter.class).addMockedMethods("getRetryFactory", "getZooKeeper").createMock();
    expect(zrw.getRetryFactory()).andReturn(retryFactory).anyTimes();
    expect(zrw.getZooKeeper()).andReturn(zk).anyTimes();
    Stat stat = new Stat();
    zk.create(path, value, ZooUtil.PUBLIC, CreateMode.PERSISTENT);
    expectLastCall().andThrow(new NodeExistsException()).once();
    expect(zk.getData(path, null, stat)).andReturn(new byte[] { 3 }).times(2);
    // BadVersionException should retry
    expect(zk.setData(path, mutatedBytes, 0)).andThrow(new ConnectionLossException());
    expect(retry.canRetry()).andReturn(true);
    retry.useRetry();
    expectLastCall();
    retry.waitForNextAttempt();
    expectLastCall();
    // Let 2nd setData succeed
    expect(zk.setData(path, mutatedBytes, 0)).andReturn(null);
    replay(zk, zrw, retryFactory, retry);
    assertArrayEquals(new byte[] { 1 }, zrw.mutateOrCreate(path, value, mutator));
    verify(zk, zrw, retryFactory, retry);
}

Example 5 with RetryFactory

use of org.apache.accumulo.fate.util.Retry.RetryFactory in project accumulo by apache.

the class RetryTest method properArgumentsInRetry.

@Test
public void properArgumentsInRetry() {
    long maxRetries = 10, startWait = 50L, maxWait = 5000L, waitIncrement = 500L, logInterval = 10000L;
    RetryFactory factory = Retry.builder().maxRetries(maxRetries).retryAfter(startWait, MS).incrementBy(waitIncrement, MS).maxWait(maxWait, MS).backOffFactor(1).logInterval(logInterval, MS).createFactory();
    Retry retry = factory.createRetry();
    assertEquals(maxRetries, retry.getMaxRetries());
    assertEquals(startWait, retry.getCurrentWait());
    assertEquals(maxWait, retry.getMaxWait());
    assertEquals(waitIncrement, retry.getWaitIncrement());
    assertEquals(logInterval, retry.getLogInterval());
}