分布式事务demo：TCC模式

Author: beston123

src/main/java/code/distribution/tcc/common/BranchState.java

@@ -0,0 +1,24 @@
+package code.distribution.tcc.common;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/8
+ */
+public enum BranchState {
+
+    INITIAL,
+
+    ONE_PHASE_OK,
+
+    TWO_COMMIT_OK,
+
+    TWO_COMMITTING,
+
+    TWO_ROLLBACK_OK,
+
+    TWO_ROLLING_BACK,
+
+}

src/main/java/code/distribution/tcc/common/BranchTx.java

@@ -0,0 +1,26 @@
+package code.distribution.tcc.common;
+
+import lombok.Data;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Data
+public class BranchTx {
+
+    private String id;
+
+    private BranchState state;
+
+    private TxMethod methods;
+
+    public BranchTx(String id, BranchState state, TxMethod methods) {
+        this.id = id;
+        this.state = state;
+        this.methods = methods;
+    }
+}

src/main/java/code/distribution/tcc/common/GlobalState.java

@@ -0,0 +1,25 @@
+package code.distribution.tcc.common;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+public enum GlobalState {
+
+    PROCESSING,
+
+//    ONE_PHASE_OK,
+//
+//    ONE_PHASE_FAIL,
+
+    TWO_COMMITTING,
+
+    TWO_COMMITTED,
+
+    TWO_ROLLING_BACK,
+
+    TWO_ROLLED_BACK
+}

src/main/java/code/distribution/tcc/common/GlobalTx.java

@@ -0,0 +1,32 @@
+package code.distribution.tcc.common;
+
+import code.distribution.at.common.GlobalStatus;
+import code.distribution.at.utils.Xid;
+import lombok.Data;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Data
+public class GlobalTx {
+
+    private String xid;
+
+    private GlobalState state;
+
+    private List<BranchTx> branches;
+
+    public GlobalTx() {
+        xid = Xid.newXid();
+        state = GlobalState.PROCESSING;
+        branches = new ArrayList<>(4);
+    }
+
+}

src/main/java/code/distribution/tcc/common/TccAction.java

@@ -0,0 +1,32 @@
+package code.distribution.tcc.common;
+
+import java.lang.annotation.*;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Target({ElementType.METHOD})
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface TccAction {
+
+    /**
+     * 预占方法
+     * @return
+     */
+    String try1();
+
+    /**
+     * 确认方法
+     */
+    String confirm();
+
+    /**
+     * 回滚方法
+     */
+    String cancel();
+}

src/main/java/code/distribution/tcc/common/TxFlowDo.java

@@ -0,0 +1,54 @@
+package code.distribution.tcc.common;
+
+import lombok.Data;
+import java.io.Serializable;
+import java.util.Date;
+
+/**
+ * 〈事务控制表〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Data
+public class TxFlowDo implements Serializable {
+
+    private String txId;
+
+    private String branchId;
+
+    /**
+     * 初始，已提交，已回滚
+     */
+    private TxState state;
+
+    private Date createTime;
+
+    private Date modifyTime;
+
+    public TxFlowDo(String txId, String branchId, TxState state) {
+        this.txId = txId;
+        this.branchId = branchId;
+        this.state = state;
+        this.createTime = new Date();
+    }
+
+    public static TxFlowDo buildInit(String txId, String branchId){
+        return new TxFlowDo(txId, branchId, TxState.INIT);
+    }
+
+    public static TxFlowDo buildCommit(String txId, String branchId){
+        return new TxFlowDo(txId, branchId, TxState.COMMIT);
+    }
+
+    public static TxFlowDo buildRollback(String txId, String branchId){
+        return new TxFlowDo(txId, branchId, TxState.ROLLBACK);
+    }
+
+    public static void main(String[] args) {
+        TxFlowDo txFlowDo = TxFlowDo.buildCommit("txId", "branchId");
+        System.out.println(txFlowDo);
+    }
+
+}

src/main/java/code/distribution/tcc/common/TxMethod.java

@@ -0,0 +1,62 @@
+package code.distribution.tcc.common;
+
+import lombok.Data;
+import org.apache.commons.lang3.reflect.MethodUtils;
+
+import java.io.Serializable;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Data
+public class TxMethod implements Serializable {
+
+    private Object object;
+
+    private Method tryMethod;
+
+    private Method confirmMethod;
+
+    private Method cancelMethod;
+
+    private Object[] args;
+
+    public TxMethod(Object object, Method tryMethod, Method confirmMethod, Method cancelMethod, Object[] args) {
+        this.object = object;
+        this.tryMethod = tryMethod;
+        this.confirmMethod = confirmMethod;
+        this.cancelMethod = cancelMethod;
+        this.args = args;
+    }
+
+    public boolean doTry(){
+        return invoke(tryMethod);
+    }
+
+    public boolean doConfirm(){
+        return invoke(confirmMethod);
+    }
+
+    public boolean doCancel(){
+        return invoke(cancelMethod);
+    }
+
+    private boolean invoke(Method method){
+        try {
+            method.invoke(object, args);
+            return true;
+        } catch (IllegalAccessException e) {
+            return false;
+        } catch (InvocationTargetException e) {
+            return false;
+        }
+    }
+
+
+}

src/main/java/code/distribution/tcc/common/TxState.java

@@ -0,0 +1,14 @@
+package code.distribution.tcc.common;
+
+/**
+ * 〈事务状态〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+public enum TxState {
+
+    INIT, COMMIT, ROLLBACK
+
+}

src/main/java/code/distribution/tcc/exception/TccException.java

@@ -0,0 +1,20 @@
+package code.distribution.tcc.exception;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author zixiao
+ * @date 2020/1/8
+ */
+public class TccException extends RuntimeException {
+
+    public TccException(String message) {
+        super(message);
+    }
+
+    public TccException(String message, Throwable cause) {
+        super(message, cause);
+    }
+
+}

src/main/java/code/distribution/tcc/package-info.java

@@ -0,0 +1,10 @@
+/**
+ * 〈TCC模式的分布式事务〉<p>
+ * 1 空回滚，try没执行，允许cancel空操作
+ * 2 防悬挂，先cancel后，再try导致资源不释放
+ * 3 幂等控制，多次confirm/cancel
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+package code.distribution.tcc;