MyBatis Plus复合主键问题解析

引言

在数据库设计中，有时候需要使用复合主键来唯一标识表中的一行数据。然而，在使用MyBatis Plus框架时，处理复合主键可能会带来一些挑战和困惑。本文将为您详细介绍MyBatis Plus中复合主键的问题以及解决方案。

复合主键定义

复合主键是由两个或多个列组成的主键，用于唯一标识一张表中的一行数据。与单一主键不同，复合主键可以通过多个列的值来确定唯一性。

MyBatis Plus支持的复合主键策略

MyBatis Plus提供了几种处理复合主键的策略，让我们一起了解它们：

1. 使用`@TableId`注解

我们可以在实体类中使用@TableId注解来标识复合主键字段。该注解接受一个type参数，用于指定主键生成策略。常见的值有AUTO、INPUT、NONE等。

AUTO：自动增长，适用于数据库自动生成主键的场景。
INPUT：用户输入，适用于用户手动输入主键的场景。
NONE：无主键，适用于没有主键或主键由其他方式生成的场景。

示例代码如下所示：

@Data
@TableName("my_table")
public class MyEntity {@TableId(type = IdType.AUTO)private Long id;@TableId(type = IdType.INPUT)private String name;
}

2. 使用`@TableId`和`@TableField`注解组合

除了在复合主键字段上使用@TableId注解外，我们还可以使用@TableField注解来标识其他的字段。这种方式需要在实体类中声明一个用于存储复合主键值的对象。

示例代码如下所示：

@Data
@TableName("my_table")
public class MyEntity {@TableId(type = IdType.NONE)private MyCompositeKey compositeKey;@TableField(exist = false) // 忽略该字段与数据库的映射private String otherField;
}@Data
public class MyCompositeKey implements Serializable {private Long id;private String name;
}

3. 自定义主键生成器

如果以上两种方法无法满足需求，我们可以自定义主键生成器。通过实现MyBatis Plus提供的IdentifierGenerator接口，我们可以灵活地生成复合主键。

示例代码如下所示：

public class MyCustomIdGenerator implements IdentifierGenerator {@Overridepublic Serializable nextId(Object entity) {// 实现自定义主键生成逻辑}
}

然后，在实体类的复合主键字段上使用@TableId指定主键生成器：

@Data
@TableName("my_table")
public class MyEntity {@TableId(type = IdType.NONE, generator = "myCustomIdGenerator")private MyCompositeKey compositeKey;
}