snowflake id (#1517)

Co-authored-by: Ben <[email protected]>
apache · Jan 29, 2024 · 8d5770c · 8d5770c
1 parent 68c6fc7
commit 8d5770c
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 3 deletions.
diff --git a/common/src/main/java/org/dromara/hertzbeat/common/entity/alerter/Alert.java b/common/src/main/java/org/dromara/hertzbeat/common/entity/alerter/Alert.java
@@ -23,6 +23,7 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import org.hibernate.annotations.GenericGenerator;
 import org.hibernate.validator.constraints.Length;
 import org.springframework.data.annotation.CreatedBy;
 import org.springframework.data.annotation.CreatedDate;
@@ -55,7 +56,8 @@
 public class Alert {
 
     @Id
-    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @GeneratedValue(strategy = GenerationType.IDENTITY, generator = "myid")
+    @GenericGenerator(name = "myid", strategy = "org.dromara.hertzbeat.common.util.SnowFlakeIdGenerator")
     @Schema(title = "Alarm record entity primary key index ID",
             description = "告警记录实体主键索引ID",
             example = "87584674384", accessMode = READ_ONLY)

diff --git a/common/src/main/java/org/dromara/hertzbeat/common/entity/warehouse/History.java b/common/src/main/java/org/dromara/hertzbeat/common/entity/warehouse/History.java
@@ -5,6 +5,7 @@
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import org.hibernate.annotations.GenericGenerator;
 
 import javax.persistence.*;
 
@@ -30,7 +31,8 @@
 public class History {
 
     @Id
-    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    @GeneratedValue(strategy = GenerationType.IDENTITY, generator = "myid")
+    @GenericGenerator(name = "myid", strategy = "org.dromara.hertzbeat.common.util.SnowFlakeIdGenerator")
     @Schema(description = "指标数据历史实体主键索引ID", example = "87584674384", accessMode = READ_ONLY)
     private Long id;
 

diff --git a/common/src/main/java/org/dromara/hertzbeat/common/util/SnowFlakeIdGenerator.java b/common/src/main/java/org/dromara/hertzbeat/common/util/SnowFlakeIdGenerator.java
@@ -17,18 +17,35 @@
 
 package org.dromara.hertzbeat.common.util;
 
+import org.hibernate.HibernateException;
+import org.hibernate.engine.spi.SharedSessionContractImplementor;
+import org.hibernate.id.IdentityGenerator;
+
+import java.io.Serializable;
+
 /**
  * Snowflake Algorithm Generator Tool
  *
  */
-public class SnowFlakeIdGenerator {
+public class SnowFlakeIdGenerator extends IdentityGenerator {
 
     private static final SnowFlakeIdWorker ID_WORKER;
 
     static {
         ID_WORKER = new SnowFlakeIdWorker();
     }
 
+    @Override
+    public Serializable generate(SharedSessionContractImplementor s, Object obj) throws HibernateException {
+        Serializable id = s.getEntityPersister(null, obj).getClassMetadata().getIdentifier(obj, s);
+
+        if (id != null && Long.valueOf(id.toString()) > 0) {
+            return id;
+        } else {
+            return SnowFlakeIdGenerator.generateId();
+        }
+    }
+
     public static long generateId() {
         return ID_WORKER.nextId();
     }