|
21 | 21 |
|
22 | 22 | import org.apache.iotdb.confignode.procedure.entity.SimpleSTMProcedure; |
23 | 23 | import org.apache.iotdb.confignode.procedure.env.TestProcEnv; |
| 24 | +import org.apache.iotdb.confignode.procedure.exception.ProcedureException; |
24 | 25 | import org.apache.iotdb.confignode.procedure.impl.StateMachineProcedure; |
25 | 26 | import org.apache.iotdb.confignode.procedure.state.ProcedureState; |
26 | 27 | import org.apache.iotdb.confignode.procedure.util.ProcedureTestUtil; |
27 | 28 |
|
28 | 29 | import org.junit.Assert; |
29 | 30 | import org.junit.Test; |
30 | 31 |
|
| 32 | +import java.io.ByteArrayOutputStream; |
| 33 | +import java.io.DataOutputStream; |
| 34 | +import java.io.IOException; |
31 | 35 | import java.lang.reflect.Field; |
| 36 | +import java.nio.ByteBuffer; |
| 37 | +import java.util.ArrayList; |
| 38 | +import java.util.Arrays; |
| 39 | +import java.util.List; |
32 | 40 | import java.util.concurrent.ConcurrentLinkedDeque; |
| 41 | +import java.util.concurrent.CountDownLatch; |
| 42 | +import java.util.concurrent.TimeUnit; |
33 | 43 | import java.util.concurrent.atomic.AtomicInteger; |
34 | 44 |
|
35 | 45 | public class STMProcedureTest extends TestProcedureBase { |
@@ -60,6 +70,104 @@ public void testRolledBackProcedure() { |
60 | 70 | Assert.assertEquals(1 + success - rolledback, acc.get()); |
61 | 71 | } |
62 | 72 |
|
| 73 | + @Test |
| 74 | + public void testFailedBeforeExecutionCanRollbackWithoutState() throws Exception { |
| 75 | + final RetryingRollbackProcedure procedure = new RetryingRollbackProcedure(); |
| 76 | + procedure.setFailure(procedure.originalFailure); |
| 77 | + procedure.doRollback(env); |
| 78 | + Assert.assertEquals(Arrays.asList(0), procedure.attemptedStates); |
| 79 | + Assert.assertSame(procedure.originalFailure, procedure.getException()); |
| 80 | + } |
| 81 | + |
| 82 | + @Test |
| 83 | + public void testFailedRollbackRetainsStateAfterSerialization() throws Exception { |
| 84 | + final RetryingRollbackProcedure procedure = new RetryingRollbackProcedure(); |
| 85 | + procedure.setState(ProcedureState.RUNNABLE); |
| 86 | + procedure.doExecute(env); |
| 87 | + procedure.doExecute(env); |
| 88 | + try { |
| 89 | + procedure.doRollback(env); |
| 90 | + Assert.fail("Compensation should fail while the Region is unavailable"); |
| 91 | + } catch (IOException expected) { |
| 92 | + Assert.assertEquals(Arrays.asList(1), procedure.attemptedStates); |
| 93 | + } |
| 94 | + procedure.setTimeout(1000); |
| 95 | + procedure.setState(ProcedureState.WAITING_TIMEOUT); |
| 96 | + final ByteArrayOutputStream bytes = new ByteArrayOutputStream(); |
| 97 | + procedure.serialize(new DataOutputStream(bytes)); |
| 98 | + final RetryingRollbackProcedure restored = new RetryingRollbackProcedure(); |
| 99 | + restored.deserialize(ByteBuffer.wrap(bytes.toByteArray())); |
| 100 | + Assert.assertTrue(restored.isFailed()); |
| 101 | + Assert.assertEquals( |
| 102 | + procedure.originalFailure.getMessage(), restored.getException().getMessage()); |
| 103 | + restored.attemptedStates.addAll(Arrays.asList(1, 1)); |
| 104 | + restored.doRollback(env); |
| 105 | + restored.doRollback(env); |
| 106 | + Assert.assertEquals(Arrays.asList(1, 1, 1, 0), restored.attemptedStates); |
| 107 | + } |
| 108 | + |
| 109 | + @Test |
| 110 | + public void testFailedRollbackRetriesSameStateWithoutOverwritingFailure() throws Exception { |
| 111 | + final RetryingRollbackProcedure procedure = new RetryingRollbackProcedure(); |
| 112 | + final long procId = procExecutor.submitProcedure(procedure); |
| 113 | + Assert.assertTrue(procedure.firstFailure.await(5, TimeUnit.SECONDS)); |
| 114 | + Assert.assertTrue(procedure.completed.await(5, TimeUnit.SECONDS)); |
| 115 | + ProcedureTestUtil.waitForProcedure(procExecutor, procId); |
| 116 | + Assert.assertTrue(procedure.isFinished()); |
| 117 | + Assert.assertEquals(Arrays.asList(1, 1, 1, 0), procedure.attemptedStates); |
| 118 | + Assert.assertSame(procedure.originalFailure, procedure.getException()); |
| 119 | + } |
| 120 | + |
| 121 | + private static class RetryingRollbackProcedure |
| 122 | + extends StateMachineProcedure<TestProcEnv, Integer> { |
| 123 | + private final CountDownLatch firstFailure = new CountDownLatch(1); |
| 124 | + private final CountDownLatch completed = new CountDownLatch(1); |
| 125 | + private final List<Integer> attemptedStates = new ArrayList<>(); |
| 126 | + private final ProcedureException originalFailure = new ProcedureException("Execution failed"); |
| 127 | + |
| 128 | + @Override |
| 129 | + protected Flow executeFromState(TestProcEnv env, Integer state) { |
| 130 | + if (state == 0) { |
| 131 | + setNextState(1); |
| 132 | + return Flow.HAS_MORE_STATE; |
| 133 | + } |
| 134 | + setFailure(originalFailure); |
| 135 | + return Flow.NO_MORE_STATE; |
| 136 | + } |
| 137 | + |
| 138 | + @Override |
| 139 | + protected void rollbackState(TestProcEnv env, Integer state) throws IOException { |
| 140 | + attemptedStates.add(state); |
| 141 | + if (state == 1 && attemptedStates.size() < 3) { |
| 142 | + firstFailure.countDown(); |
| 143 | + throw new IOException("Region temporarily unavailable"); |
| 144 | + } |
| 145 | + if (state == 0) { |
| 146 | + completed.countDown(); |
| 147 | + } |
| 148 | + } |
| 149 | + |
| 150 | + @Override |
| 151 | + protected long getRollbackRetryTimeout() { |
| 152 | + return 50; |
| 153 | + } |
| 154 | + |
| 155 | + @Override |
| 156 | + protected Integer getState(int stateId) { |
| 157 | + return stateId; |
| 158 | + } |
| 159 | + |
| 160 | + @Override |
| 161 | + protected int getStateId(Integer state) { |
| 162 | + return state; |
| 163 | + } |
| 164 | + |
| 165 | + @Override |
| 166 | + protected Integer getInitialState() { |
| 167 | + return 0; |
| 168 | + } |
| 169 | + } |
| 170 | + |
63 | 171 | @Test |
64 | 172 | public void testEofStateReexecutionDoesNotCallExecuteFromState() throws Exception { |
65 | 173 | EofReexecutionProcedure procedure = new EofReexecutionProcedure(); |
|
0 commit comments