当前位置: 代码迷 >> 综合 >> myBatis是怎么使用 @GeneratedValue(generator = “...“, strategy = ...) 注解
  详细解决方案

myBatis是怎么使用 @GeneratedValue(generator = “...“, strategy = ...) 注解

热度:52   发布时间:2023-12-25 02:44:10.0

一. @GeneratedValue注解id生成策略

??使用范围:方法和属性

@Target({
    METHOD, FIELD})
@Retention(RUNTIME)
public @interface GeneratedValue {
    /*** (Optional) The primary key generation strategy* that the persistence provider must use to* generate the annotated entity primary key.*/GenerationType strategy() default AUTO;/*** (Optional) The name of the primary key generator* to use as specified in the {@link SequenceGenerator} * or {@link TableGenerator} annotation.* <p> Defaults to the id generator supplied by persistence provider.*/String generator() default "";
}

GenerationType :主键生成策略,有TABLE,SEQUENCE,IDENTITY,AUTO四种。
generator:主键生成的来源,既由谁来执行生成主键,从哪里获取主键值。下面会讲到mybatis可以配置的值。

二. GenerationType 定义主键策略的枚举

此部分参考博客

/** * Defines the types of primary key generation. ** @since Java Persistence 1.0*/
public enum GenerationType {
     /*** Indicates that the persistence provider must assign * primary keys for the entity using an underlying * database table to ensure uniqueness.*/TABLE, /*** Indicates that the persistence provider must assign * primary keys for the entity using database sequence column.*/SEQUENCE, /*** Indicates that the persistence provider must assign * primary keys for the entity using database identity column.*/IDENTITY, /*** Indicates that the persistence provider should pick an * appropriate strategy for the particular database. The * <code>AUTO</code> generation strategy may expect a database * resource to exist, or it may attempt to create one. A vendor * may provide documentation on how to create such resources * in the event that it does not support schema generation * or cannot create the schema resource at runtime.*/AUTO
}

1.GenerationType.TABLE

??使用一个特定的数据库表格来保存主键,持久化引擎通过关系数据库的一张特定的表格来生成主键,这种策略的好处就是不依赖于外部环境和数据库的具体实现,在不同数据库间可以很容易的进行移植,但由于其不能充分利用数据库的特性,所以不会优先使用。该策略一般与另外一个注解一起使用@TableGenerator,@TableGenerator注解指定了生成主键的表(可以在实体类上指定也可以在主键字段或属性上指定),然后JPA将会根据注解内容自动生成一张表作为序列表(或使用现有的序列表)。如果不指定序列表,则会生成一张默认的序列表,表中的列名也是自动生成,数据库上会生成一张名为sequence的表(SEQ_NAME,SEQ_COUNT)。序列表一般只包含两个字段:第一个字段是该生成策略的名称,第二个字段是该关系表的最大序号,它会随着数据的插入逐渐累加。类似于

@Id
@GeneratedValue(strategy = GenerationType.TABLE, generator = "roleSeq")
@TableGenerator(name = "roleSeq", allocationSize = 1, table = "seq_table", pkColumnName = "seq_id", valueColumnName = "seq_count")
private Integer id;

??在以上例子中,roleSeq唯一的标识了该生成器,在@GeneratedValue注解中的generator属性可以根据此标识来声明主键生成器。

2.GenerationType.SEQUENCE

??在某些数据库中,不支持主键自增长,比如Oracle,其提供了一种叫做"序列(sequence)"的机制生成主键。此时,GenerationType.SEQUENCE就可以作为主键生成策略。该策略的不足之处正好与TABLE相反,由于只有部分数据库(Oracle,PostgreSQL,DB2)支持序列对象,所以该策略一般不应用于其他数据库。类似的,该策略一般与另外一个注解一起使用@SequenceGenerator,@SequenceGenerator注解指定了生成主键的序列.然后JPA会根据注解内容创建一个序列(或使用一个现有的序列)。如果不指定序列,则会自动生成一个序列SEQ_GEN_SEQUENCE。类似于

@Id
@GeneratedValue(strategy = GenerationType.SEQUENCE, generator = "menuSeq")
@SequenceGenerator(name = "menuSeq", initialValue = 1, allocationSize = 1, sequenceName = "MENU_SEQUENCE")
private Integer id;

??同样,在以上例子中,menuSeq唯一的标识了该生成器,@SequenceGenerator可以理解为将数据库中存在的序列进行了一个映射,在@GeneratedValue注解中的generator属性可以根据此标识来声明主键生成器。

3.GenerationType.IDENTITY

??此种主键生成策略就是通常所说的主键自增长,数据库在插入数据时,会自动给主键赋值,比如MYSQL可以在创建表时声明"auto_increment" 来指定主键自增长。该策略在大部分数据库中都提供了支持(指定方法或关键字可能不同),但还是有少数数据库不支持,所以可移植性略差。使用自增长主键生成策略是只需要声明strategy = GenerationType.IDENTITY即可。类似于

@Id
@GeneratedValue(strategy = GenerationType.IDENTITY,generator = "mysql")
private Integer id;

??需要注意的是,同一张表中自增列最多只能有一列。

4.GenerationType.AUTO

??把主键生成策略交给持久化引擎(persistence engine),持久化引擎会根据数据库在以上三种主键生成策略中选择其中一种。此种主键生成策略比较常用,由于JPA默认的生成策略就是GenerationType.AUTO,所以使用此种策略时.可以显式的指定@GeneratedValue(strategy = GenerationType.AUTO)也可以直接@GeneratedValue

类似于

    @GeneratedValue(strategy = GenerationType.AUTO)private Integer id;@GeneratedValueprivate Integer id;

三. myBatis 对@GeneratedValue的解析实现

// 包名
package tk.mybatis.mapper.mapperhelper;//主键策略 - Oracle序列,MySql自动增长,UUID// Oracle序列生成注解if (field.isAnnotationPresent(SequenceGenerator.class)) {
    SequenceGenerator sequenceGenerator = field.getAnnotation(SequenceGenerator.class);if (sequenceGenerator.sequenceName().equals("")) {
    throw new RuntimeException(entityTable.getEntityClass() + "字段" + field.getName() + "的注解@SequenceGenerator未指定sequenceName!");}entityColumn.setSequenceName(sequenceGenerator.sequenceName());//MySql自动增长,UUID} else if (field.isAnnotationPresent(GeneratedValue.class)) {
    GeneratedValue generatedValue = field.getAnnotation(GeneratedValue.class);// 首先 通过generator 类型判断主键生成类型if (generatedValue.generator().equals("UUID")) {
    entityColumn.setUuid(true);} else if (generatedValue.generator().equals("JDBC")) {
    entityColumn.setIdentity(true);entityColumn.setGenerator("JDBC");entityTable.setKeyProperties(entityColumn.getProperty());entityTable.setKeyColumns(entityColumn.getColumn());} else {
    //允许通过generator来设置获取id的sql,例如mysql=CALL IDENTITY(),hsqldb=SELECT SCOPE_IDENTITY()//允许通过拦截器参数设置公共的generator// generator 不是UUID 和 JDBC时 判断 strategy ;如果类型不是 GenerationType.IDENTITY 则会抛出异常if (generatedValue.strategy() == GenerationType.IDENTITY) {
    //mysql的自动增长entityColumn.setIdentity(true);if (!generatedValue.generator().equals("")) {
    String generator = null;// 获取generator 主键生成的来源 IdentityDialect identityDialect = IdentityDialect.getDatabaseDialect(generatedValue.generator());if (identityDialect != null) {
    generator = identityDialect.getIdentityRetrievalStatement();} else {
    generator = generatedValue.generator();}entityColumn.setGenerator(generator);}} else {
    throw new RuntimeException(field.getName()+ " - 该字段@GeneratedValue配置只允许以下几种形式:" +"\n1.全部数据库通用的@GeneratedValue(generator=\"UUID\")" +"\n2.useGeneratedKeys的@GeneratedValue(generator=\\\"JDBC\\\") " +"\n3.类似mysql数据库的@GeneratedValue(strategy=GenerationType.IDENTITY[,generator=\"Mysql\"])");}}}entityTable.getEntityClassColumns().add(entityColumn);if (entityColumn.isId()) {
    entityTable.getEntityClassPKColumns().add(entityColumn);}

?? 由上面的mybatis对@GeneratedValue注解的解析可知 :
?? 1. Mybatis先解析注解@GeneratedValue 中的generator,
?? 2. 如果generator的值不是UUID 和 JDBC时,再判断 strategy ,
?? 3. 如果类型不是 GenerationType.IDENTITY 则会抛出异常,
?? 4. 因此strategy在使用mybatis框架时,值必须是GenerationType.IDENTITY,
?? 5. 而generator的取值可以在下面的枚举中看到,且不区分大小写。

package tk.mybatis.mapper.code;public enum IdentityDialect {
    DB2("VALUES IDENTITY_VAL_LOCAL()"),MYSQL("SELECT LAST_INSERT_ID()"),SQLSERVER("SELECT SCOPE_IDENTITY()"),CLOUDSCAPE("VALUES IDENTITY_VAL_LOCAL()"),DERBY("VALUES IDENTITY_VAL_LOCAL()"),HSQLDB("CALL IDENTITY()"),SYBASE("SELECT @@IDENTITY"),DB2_MF("SELECT IDENTITY_VAL_LOCAL() FROM SYSIBM.SYSDUMMY1"),INFORMIX("select dbinfo('sqlca.sqlerrd1') from systables where tabid=1");private String identityRetrievalStatement;private IdentityDialect(String identityRetrievalStatement) {
    this.identityRetrievalStatement = identityRetrievalStatement;}public static IdentityDialect getDatabaseDialect(String database) {
    IdentityDialect returnValue = null;if ("DB2".equalsIgnoreCase(database)) {
    returnValue = DB2;} else if ("MySQL".equalsIgnoreCase(database)) {
    returnValue = MYSQL;} else if ("SqlServer".equalsIgnoreCase(database)) {
    returnValue = SQLSERVER;} else if ("Cloudscape".equalsIgnoreCase(database)) {
    returnValue = CLOUDSCAPE;} else if ("Derby".equalsIgnoreCase(database)) {
    returnValue = DERBY;} else if ("HSQLDB".equalsIgnoreCase(database)) {
    returnValue = HSQLDB;} else if ("SYBASE".equalsIgnoreCase(database)) {
    returnValue = SYBASE;} else if ("DB2_MF".equalsIgnoreCase(database)) {
    returnValue = DB2_MF;} else if ("Informix".equalsIgnoreCase(database)) {
    returnValue = INFORMIX;}return returnValue;}public String getIdentityRetrievalStatement() {
    return identityRetrievalStatement;}
}

?? 因此,项目集成mybatis 使用mysql数据库的主键配置为:

在这里插入图片描述
?? strategy:必须为 GenerationType.IDENTITY;
?? generator: mysql大写忽略。

  相关解决方案