default id gen

Author: ujued

txlcn-common/src/main/java/com/codingapi/txlcn/common/util/id/DefaultIdGen.java

@@ -0,0 +1,62 @@
+package com.codingapi.txlcn.common.util.id;
+
+/**
+ * Description:
+ * Date: 2/1/19
+ *
+ * @author ujued
+ */
+public class DefaultIdGen implements IdGen {
+
+    /**
+     * 2019-2-1
+     */
+    private static final long START_TIME = 1548992829394L;
+
+    private long machineId;
+
+    private int machineOffset;
+
+    private long seq;
+
+    private int seqLen;
+
+    private long lastTime;
+
+    public DefaultIdGen(int seqLen, long machineId) {
+        this.seqLen = seqLen;
+        this.machineOffset = seqLen;
+        this.machineId = machineId;
+    }
+
+    @Override
+    public synchronized String nextId() {
+        long curTime = System.currentTimeMillis();
+        if (curTime < lastTime) {
+            throw new IllegalStateException("");
+        }
+
+        if (curTime == lastTime) {
+            seq = (seq + 1) & (~(-1 << seqLen));
+            if (seq == 0L) {
+                curTime = tilNextMillis();
+            }
+        } else {
+            seq = 0L;
+        }
+
+        lastTime = curTime;
+
+        long seqWithMachine = (machineId << machineOffset & Long.MAX_VALUE) | seq;
+
+        return String.valueOf(curTime - START_TIME) + seqWithMachine;
+    }
+
+    private long tilNextMillis() {
+        long newTime = System.currentTimeMillis();
+        while (newTime <= lastTime) {
+            newTime = System.currentTimeMillis();
+        }
+        return newTime;
+    }
+}

txlcn-common/src/main/java/com/codingapi/txlcn/common/util/id/IdGen.java

@@ -22,5 +22,5 @@
  * @author ujued
  */
 public interface IdGen {
-    long nextId();
+    String nextId();
 }

txlcn-common/src/main/java/com/codingapi/txlcn/common/util/id/IdGenInit.java

@@ -14,6 +14,10 @@ public abstract class IdGenInit {
     public static void applySnowFlakeIdGen(int machineLen, int machineId) {
         SnowFlakeGenerator.Factory factory = new SnowFlakeGenerator.Factory(machineLen, 0);
         SnowFlakeGenerator snowFlakeGenerator = factory.create(0, machineId);
-        RandomUtils.init(snowFlakeGenerator::nextId);
+        RandomUtils.init(() -> String.valueOf(snowFlakeGenerator.nextId()));
+    }
+
+    public static void applyDefaultIdGen(int seqLen, int machineId) {
+        RandomUtils.init(new DefaultIdGen(seqLen, machineId));
     }
 }

txlcn-common/src/main/java/com/codingapi/txlcn/common/util/id/RandomUtils.java

@@ -35,9 +35,9 @@ public static String getUUID() {
 
     public static String randomKey() {
         if (Objects.isNull(theIdGen)) {
-            theIdGen = System::nanoTime;
+            theIdGen = new DefaultIdGen(12, 0);
         }
-        return String.valueOf(theIdGen.nextId());
+        return theIdGen.nextId();
     }
 
     public static String simpleKey() {

txlcn-tc/src/main/java/com/codingapi/txlcn/tc/txmsg/TCSideRpcInitCallBack.java

@@ -82,7 +82,7 @@ public void connected(String remoteKey) {
                     txClientConfig.applyMachineId(resParams.getMachineId());
 
                     // 2. IdGen 初始化
-                    IdGenInit.applySnowFlakeIdGen(resParams.getMachineLen(), resParams.getMachineId());
+                    IdGenInit.applyDefaultIdGen(resParams.getMachineLen(), resParams.getMachineId());
 
                     // 3. 日志
                     log.info("Finally, determined dtx time is {}ms, tm rpc timeout is {} ms, machineId is {}",

txlcn-tm/src/main/java/com/codingapi/txlcn/tm/txmsg/EnsureIdGenEngine.java

@@ -55,7 +55,7 @@ public void disconnect(String remoteKey, String appName) {
 
     @Override
     public void init() throws Exception {
-        IdGenInit.applySnowFlakeIdGen(managerConfig.getMachineIdLen(), managerService.machineIdSync());
+        IdGenInit.applyDefaultIdGen(managerConfig.getMachineIdLen(), managerService.machineIdSync());
 
         Transactions.setApplicationIdWhenRunning(ApplicationInformation.modId(environment, serverProperties));
     }