Initial commit: bositao cloud (jnpf-java-cloud-v6x)

刘光辉

118 分钟以前 06f314545ac3514d7ce28939920866ae8188d31c

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package jnpf.flowable.service;
 
import com.baomidou.dynamic.datasource.tx.*;
import org.junit.After;
import org.junit.Test;
import java.util.concurrent.atomic.AtomicInteger;
import static org.junit.Assert.*;
 
/** Uses the installed dynamic-datasource transaction executor, not a simulated callback loader. */
public class DmsLifecycleEngineCompensationTest {
    @After public void cleanup() {TransactionContext.remove();TransactionContext.removeSynchronizations();}
    private void transaction(Runnable work) throws Throwable {
        var info=new TransactionalInfo();info.setPropagation(DsPropagation.REQUIRED);
        info.setRollbackFor(new Class[]{Exception.class});info.setNoRollbackFor(new Class[]{});
        new TransactionalTemplate().execute(new TransactionalExecutor() {
            public Object execute() {work.run();return null;}
            public TransactionalInfo getTransactionInfo() {return info;}
        });
    }
    @Test public void successfulCommitRetainsEngineInstance() throws Throwable {
        var deleted=new AtomicInteger();
        transaction(()->DmsLifecycleEngineCompensation.register("engine",deleted::incrementAndGet));
        assertEquals(0,deleted.get());assertTrue(TransactionContext.getSynchronizations().isEmpty());
    }
    @Test public void laterFailureRollsBackEveryPreviouslyStartedDmsInstance() throws Throwable {
        var deleted=new AtomicInteger();
        try {
            transaction(()->{
                DmsLifecycleEngineCompensation.register("first",deleted::incrementAndGet);
                DmsLifecycleEngineCompensation.register("second",deleted::incrementAndGet);
                throw new IllegalStateException("later native save failed");
            });fail();
        } catch(IllegalStateException expected) { }
        assertEquals(2,deleted.get());assertTrue(TransactionContext.getSynchronizations().isEmpty());
    }
    @Test public void immediateCompensationDoesNotDeleteTwiceOnRollback() throws Throwable {
        var deleted=new AtomicInteger();
        try {
            transaction(()->{
                var ticket=DmsLifecycleEngineCompensation.register("engine",deleted::incrementAndGet);
                var failure=new IllegalStateException("configuration read failed");
                ticket.compensate(failure);throw failure;
            });fail();
        } catch(IllegalStateException expected) { }
        assertEquals(1,deleted.get());
    }
    @Test public void failedImmediateDeletionRetriesAtOuterRollback() throws Throwable {
        var attempts=new AtomicInteger();var failure=new IllegalArgumentException("native failure");
        try {
            transaction(()->{
                var ticket=DmsLifecycleEngineCompensation.register("engine",()->{
                    if(attempts.incrementAndGet()==1) throw new IllegalStateException("temporary engine failure");
                });ticket.compensate(failure);throw failure;
            });fail();
        } catch(IllegalArgumentException expected) {assertSame(failure,expected);}
        assertEquals(2,attempts.get());assertEquals(1,failure.getSuppressed().length);
    }
}