feat: 自定义es及接口完成

2024-03-06 17:18:23 +08:00
parent 964c4beb11
commit ce4387c7e3
20 changed files with 866 additions and 112 deletions
--- a/qing-yu-club-subject/qing-yu-club-common/src/main/java/com/landaiqing/subject/common/util/IdWorkerUtil.java
+++ b/qing-yu-club-subject/qing-yu-club-common/src/main/java/com/landaiqing/subject/common/util/IdWorkerUtil.java
@@ -0,0 +1,99 @@
+package com.landaiqing.subject.common.util;
+
+public class IdWorkerUtil {
+
+    private long workerId;
+
+    private long datacenterId;
+
+    private long sequence;
+
+    private long twepoch = 1585644268888L;
+
+    private long workerIdBits = 5L;
+
+    private long datacenterIdBits = 5L;
+
+    private long sequenceBits = 12L;
+
+    private long maxWorkerId = -1L ^ (-1L << workerIdBits);
+
+    private long maxDatacenterId = -1L ^ (-1L << datacenterIdBits);
+
+    private long workerIdShift = sequenceBits;
+
+    private long datacenterIdShift = sequenceBits + workerIdBits;
+
+    private long timestampLeftShift = sequenceBits + workerIdBits + datacenterIdBits;
+
+    private long sequenceMask = -1L ^ (-1L << sequenceBits);
+
+    private long lastTimestamp = -1L;
+
+    public long getWorkerId() {
+        return workerId;
+    }
+
+    public long getDatacenterId() {
+        return datacenterId;
+    }
+
+    public long getTimestamp() {
+        return System.currentTimeMillis();
+    }
+
+    public IdWorkerUtil(long workerId, long datacenterId, long sequence) {
+        if (workerId > maxWorkerId || workerId < 0) {
+            throw new IllegalArgumentException(
+                    String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
+        }
+        if (datacenterId > maxDatacenterId || datacenterId < 0) {
+
+            throw new IllegalArgumentException(
+                    String.format("datacenter Id can't be greater than %d or less than 0", maxDatacenterId));
+        }
+        this.workerId = workerId;
+        this.datacenterId = datacenterId;
+        this.sequence = sequence;
+    }
+
+    public synchronized long nextId() {
+        long timestamp = timeGen();
+        if (timestamp < lastTimestamp) {
+            System.err.printf(
+                    "clock is moving backwards. Rejecting requests until %d.", lastTimestamp);
+            throw new RuntimeException(
+                    String.format("Clock moved backwards. Refusing to generate id for %d milliseconds",
+                            lastTimestamp - timestamp));
+        }
+
+        if (lastTimestamp == timestamp) {
+
+            sequence = (sequence + 1) & sequenceMask;
+            if (sequence == 0) {
+                timestamp = tilNextMillis(lastTimestamp);
+            }
+
+        } else {
+            sequence = 0;
+        }
+        lastTimestamp = timestamp;
+        return ((timestamp - twepoch) << timestampLeftShift) |
+                (datacenterId << datacenterIdShift) |
+                (workerId << workerIdShift) | sequence;
+    }
+
+    private long tilNextMillis(long lastTimestamp) {
+        long timestamp = timeGen();
+        while (timestamp <= lastTimestamp) {
+            timestamp = timeGen();
+        }
+        return timestamp;
+    }
+
+    //获取当前时间戳
+    private long timeGen() {
+        return System.currentTimeMillis();
+    }
+
+}