TCC实现

Author: tianwu

pom.xml

@@ -8,6 +8,10 @@
     <artifactId>java-code</artifactId>
     <version>1.0-SNAPSHOT</version>
 
+    <properties>
+        <spring-boot.version>2.2.2.RELEASE</spring-boot.version>
+    </properties>
+
     <dependencies>
         <dependency>
             <groupId>org.javassist</groupId>
@@ -20,12 +24,28 @@
             <groupId>org.apache.zookeeper</groupId>
             <artifactId>zookeeper</artifactId>
             <version>3.4.8</version>
+            <exclusions>
+                <exclusion>
+                    <artifactId>slf4j-log4j12</artifactId>
+                    <groupId>org.slf4j</groupId>
+                </exclusion>
+            </exclusions>
         </dependency>
         <!-- ZkClient -->
         <dependency>
             <groupId>com.101tec</groupId>
             <artifactId>zkclient</artifactId>
             <version>0.8</version>
+            <exclusions>
+                <exclusion>
+                    <artifactId>log4j</artifactId>
+                    <groupId>log4j</groupId>
+                </exclusion>
+                <exclusion>
+                    <artifactId>slf4j-log4j12</artifactId>
+                    <groupId>org.slf4j</groupId>
+                </exclusion>
+            </exclusions>
         </dependency>
 
         <dependency>
@@ -94,10 +114,25 @@
             <version>3.4</version>
         </dependency>
 
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter</artifactId>
+            <version>${spring-boot.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+            <version>${spring-boot.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-test</artifactId>
+            <version>${spring-boot.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.springframework</groupId>
-            <artifactId>spring-core</artifactId>
-            <version>4.3.25.RELEASE</version>
+            <artifactId>spring-test</artifactId>
+            <version>5.2.2.RELEASE</version>
         </dependency>
 
         <dependency>
@@ -109,7 +144,7 @@
         <dependency>
             <groupId>junit</groupId>
             <artifactId>junit</artifactId>
-            <version>4.7</version>
+            <version>4.12</version>
         </dependency>
 
         <dependency>
@@ -154,6 +189,18 @@
             <version>3.3.7</version>
         </dependency>
 
+        <dependency>
+            <groupId>org.aspectj</groupId>
+            <artifactId>aspectjweaver</artifactId>
+            <version>1.9.6</version>
+        </dependency>
+
+        <dependency>
+            <groupId>aopalliance</groupId>
+            <artifactId>aopalliance</artifactId>
+            <version>1.0</version>
+        </dependency>
+
     </dependencies>
 
     <build>

src/main/java/code/CodeApplication.java

@@ -0,0 +1,24 @@
+package code;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
+import org.springframework.boot.web.servlet.ServletComponentScan;
+
+/**
+ * 〈启动入口〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author tianwu
+ * @date 2022/09/08
+ */
+@ServletComponentScan
+@EnableConfigurationProperties
+@SpringBootApplication
+public class CodeApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(CodeApplication.class);
+    }
+
+}

src/main/java/code/distribution/id/Snowflake/IdWorker.java renamed to src/main/java/code/distribution/id/snowflake/IdWorker.java

@@ -1,4 +1,4 @@
-package code.distribution.id.Snowflake;
+package code.distribution.id.snowflake;
 
 /**
  * 〈Snowflake算法〉<p>
@@ -141,7 +141,7 @@ public static IdWorker getInstance() {
         return idWorker;
     }
 
-    public static long initWorkerId() {
+    private static long initWorkerId() {
         InetAddress address;
         try {
             address = InetAddress.getLocalHost();
@@ -152,7 +152,7 @@ public static long initWorkerId() {
         return ((ipAddressByteArray[ipAddressByteArray.length - 2] & 0B11) << Byte.SIZE) + (ipAddressByteArray[ipAddressByteArray.length - 1] & 0xFF);
     }
 
-    public static void init(Long serverNodeId) {
+    private static void init(Long serverNodeId) {
         if (idWorker == null) {
             synchronized (IdWorker.class) {
                 if (idWorker == null) {

src/main/java/code/distribution/tcc/common/TccAction.java renamed to src/main/java/code/distribution/tcc/annotation/TccBranch.java

@@ -1,9 +1,9 @@
-package code.distribution.tcc.common;
+package code.distribution.tcc.annotation;
 
 import java.lang.annotation.*;
 
 /**
- * 〈一句话功能简述〉<p>
+ * 〈TCC分支事务〉<p>
  * 〈功能详细描述〉
  *
  * @author zixiao
@@ -12,13 +12,13 @@
 @Target({ElementType.METHOD})
 @Retention(RetentionPolicy.RUNTIME)
 @Documented
-public @interface TccAction {
+public @interface TccBranch {
 
     /**
-     * 预占方法
+     * 分支名称
      * @return
      */
-    String try1();
+    String actionName();
 
     /**
      * 确认方法

src/main/java/code/distribution/tcc/annotation/TccTransaction.java

@@ -0,0 +1,17 @@
+package code.distribution.tcc.annotation;
+
+import java.lang.annotation.*;
+
+/**
+ * 〈TCC事务〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author tianwu
+ * @date 2022/09/08
+ */
+@Target({ElementType.METHOD})
+@Retention(RetentionPolicy.RUNTIME)
+@Documented
+public @interface TccTransaction {
+
+}

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

@@ -12,15 +12,15 @@
 @Data
 public class BranchTx {
 
-    private String id;
+    private Long id;
 
     private BranchState state;
 
-    private TxMethod methods;
+    private TxMethod method;
 
-    public BranchTx(String id, BranchState state, TxMethod methods) {
+    public BranchTx(Long id, BranchState state, TxMethod methods) {
         this.id = id;
         this.state = state;
-        this.methods = methods;
+        this.method = methods;
     }
 }

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

@@ -0,0 +1,14 @@
+package code.distribution.tcc.common;
+
+/**
+ * 〈一句话功能简述〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author tianwu
+ * @date 2022/09/07
+ */
+public interface Callback<T> {
+
+    T execute() throws Throwable;
+
+}

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

@@ -0,0 +1,44 @@
+package code.distribution.tcc.common;
+
+import code.distribution.id.snowflake.IdWorker;
+
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * 〈事务容器〉<p>
+ * 〈功能详细描述〉
+ *
+ * @author tianwu
+ * @date 2022/09/07
+ */
+public class TxContainer {
+
+    private static final ThreadLocal<String> XID = new ThreadLocal<>();
+
+    private static final ThreadLocal<Map<String, Long>> BRANCH_ID_MAP = ThreadLocal.withInitial(HashMap::new);
+
+    public static String getXid(){
+        return XID.get();
+    }
+
+    public static void setXid(String xid) {
+        XID.set(xid);
+    }
+
+    public static Long getBranchId(String actionName){
+        return BRANCH_ID_MAP.get().get(actionName);
+    }
+
+    public static Long setBranchId(String actionName){
+        Long branchId = IdWorker.getInstance().nextId();
+        BRANCH_ID_MAP.get().put(actionName, branchId);
+        return branchId;
+    }
+
+    public static void remove(){
+        XID.remove();
+        BRANCH_ID_MAP.get().clear();
+    }
+
+}

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

@@ -0,0 +1,66 @@
+package code.distribution.tcc.common;
+
+import lombok.Data;
+import java.io.Serializable;
+import java.util.Date;
+
+/**
+ * 〈事务控制表〉<p>
+ `xid`           VARCHAR(128)  NOT NULL COMMENT 'global id',
+ `branch_id`     BIGINT        NOT NULL COMMENT 'branch id',
+ `action_name`   VARCHAR(64)   NOT NULL COMMENT 'action name',
+ `status`        TINYINT       NOT NULL COMMENT 'status(tried:1;committed:2;rollbacked:3;suspended:4)',
+ `gmt_create`    DATETIME(3)   NOT NULL COMMENT 'create time',
+ `gmt_modified`  DATETIME(3)   NOT NULL COMMENT 'update time',
+ PRIMARY KEY (`xid`, `branch_id`)
+ *
+ * @author zixiao
+ * @date 2020/1/7
+ */
+@Data
+public class TxFlowLogDO implements Serializable {
+
+    /**
+     * 全局事务id
+     */
+    private String xid;
+
+    /**
+     * 分支事务id
+     */
+    private Long branchId;
+
+    /**
+     * 已Try，已提交，已回滚
+     */
+    private TxState state;
+
+    private Date gmtCreate;
+
+    private Date gmtModified;
+
+    public TxFlowLogDO(String xid, Long branchId, TxState state) {
+        this.xid = xid;
+        this.branchId = branchId;
+        this.state = state;
+        this.gmtCreate = new Date();
+        this.gmtModified = new Date();
+    }
+
+    public static TxFlowLogDO buildTry(String xid, Long branchId){
+        return new TxFlowLogDO(xid, branchId, TxState.TRIED);
+    }
+
+    public static TxFlowLogDO buildRollback(String xid, Long branchId){
+        return new TxFlowLogDO(xid, branchId, TxState.ROLLBACKED);
+    }
+
+    public String getUniqueKey() {
+        return buildKey(xid, branchId);
+    }
+
+    public static String buildKey(String xid, Long branchId) {
+        return xid + "@" + branchId;
+    }
+
+}