Spring boot JPA 用自定义主键策略 生成自定义主键ID

最近学习Spring boot JPA 学习过程解决的一些问题写成随笔，大家一起成长。
这次遇到自定义主键的问题

 package javax.persistence;

 public enum GenerationType {
     TABLE,
     SEQUENCE,
     IDENTITY,
     AUTO;

     private GenerationType() {
     }
 }

GenerationType源码

从源码中可以看出JPA提供的四种标准主键策略TABLE,SEQUENCE,IDENTITY,AUTO

TABLE：使用一个特定的数据库表格来保存主键。

SEQUENCE：根据底层数据库的序列来生成主键，条件是数据库支持序列。 这个值要与generator一起使用，generator 指定生成主键使用的生成器（可能是orcale中自己编写的序列）。

IDENTITY：主键由数据库自动生成（主要是支持自动增长的数据库，如mysql）

AUTO：主键由程序控制，也是GenerationType的默认值。

不写GeneratedValue注解时即为GenerationType.AUTO 这时主键生成是根据数据库hibernate_sequence里的next_val来生成但我想用自已规则来生成自己的ID查了半天的资料终于解决这个问题。

自定义生成的ID，看到效果出来那个兴奋啊，新手对java的一堆注解还是真的不容易上手。

下面是具体代码

1.在实体主键get方法添加注解

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO, generator = "custom-id")     @GenericGenerator(name = "custom-id", strategy = "com.muyuer.springdemo.core.CustomIDGenerator")
    @Column(name = "user_id")
    public Long getUserId() {
        return userId;
    }

注意：GeneratedValue中的generator要与GenericGenerator中的name相等 上面代码中是"custom-id"

2.添加自定义ID生成类

 package com.muyuer.springdemo.core;

 import com.muyuer.springdemo.utils.SnowflakeIdHelper;
 import org.hibernate.MappingException;
 import org.hibernate.engine.spi.SharedSessionContractImplementor;
 import org.hibernate.id.IdentityGenerator;
 import java.io.Serializable;

 /**
  * 自定义ID生成器
  * @author muyuer 182443947@qq.com
  * @version 1.0
  * @date 2018-12-08 15:42
  */
 public class CustomIDGenerator extends IdentityGenerator {
     @Override
     public Serializable generate(SharedSessionContractImplementor session, Object object) throws MappingException {
         Object id =  SnowflakeIdHelper.getId();
         if (id != null) {
             return (Serializable) id;
         }
         return super.generate(session, object);
     }
 }

这里Override了generate方法通过SnowflakeIdHelper.getId();返回了自定义的ID。

注意：我测试的ID是Long类型所以这里继承的是IdentityGenerator类，如果ID为String类型的话应该继承 UUIDGenerator 或者 UUIDGenerator