主要给大家介绍了关于spring boot mybatis枚举映射的相关资料，文中通过示例代码介绍的非常详细，对大家学习或者使用Spring Boot具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧
在Spring Boot集成MyBatis的项目中，我们经常会遇到实体类中的字段是枚举类型，而数据库中的对应字段是整型或字符串型的情况。这种情况下，我们需要进行枚举与数据库字段之间的映射处理，以便正确地保存和读取数据。本篇文章将详细讲解如何在Spring Boot MyBatis中实现枚举的映射，并提供示例代码。
我们需要创建一个枚举类，例如`YesOrNo`，它包含两个枚举元素：`Yes`和`No`，每个元素都有一个对应的整数值（code）和文本描述（text）。这是一个简单的枚举类示例：
```java
public enum YesOrNo implements BaseEnum {
Yes(1, "是"),
No(0, "否");
private Integer code;
private String text;
YesOrNo(Integer code, String text) {
this.code = code;
this.text = text;
}
// 实现BaseEnum接口的方法
@Override
public Integer getCode() {
return this.code;
}
@Override
public String getText() {
return this.text;
}
// 用于反序列化，根据code获取枚举对象
public static YesOrNo jsonCreate(Integer code) {
return EnumUtils.codeOf(YesOrNo.class, code);
}
}
```
这里的`BaseEnum`接口定义了`getCode()`和`getText()`方法，以供其他枚举类继承并实现。`jsonCreate`方法用于根据code反序列化为枚举对象，这在JSON转换时可能会用到。
接下来，我们需要创建一个自定义的枚举处理器，这个处理器会处理枚举到整型的转换。这里我们创建了一个名为`UniversalEnumHandler`的类，它继承自MyBatis的`BaseTypeHandler`，并处理特定类型的枚举：
```java
@MappedTypes(value = {YesOrNo.class})
public class UniversalEnumHandler<E extends Enum<E> & BaseEnum> extends BaseTypeHandler<E> {
private final Class<E> type;
public UniversalEnumHandler(Class<E> type) {
if (type == null) {
throw new IllegalArgumentException("Type argument cannot be null");
}
this.type = type;
}
// 将枚举对象设置为PreparedStatement参数
@Override
public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException {
ps.setInt(i, parameter.getCode());
}
// 从ResultSet中获取枚举对象
@Override
public E getNullableResult(ResultSet rs, String columnName) throws SQLException {
int code = rs.getInt(columnName);
return rs.wasNull() ? null : EnumUtils.codeOf(this.type, code);
}
// 从ResultSet中获取枚举对象
@Override
public E getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
int code = rs.getInt(columnIndex);
return rs.wasNull() ? null : EnumUtils.codeOf(this.type, code);
}
// 从CallableStatement中获取枚举对象
@Override
public E getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
int code = cs.getInt(columnIndex);
return cs.wasNull() ? null : EnumUtils.codeOf(this.type, code);
}
}
```
`UniversalEnumHandler`类通过`@MappedTypes`注解指定了处理的枚举类型。在`setNonNullParameter`方法中，我们将枚举的code设置为PreparedStatement的参数；在`getNullableResult`方法中，我们从ResultSet中读取整数值，并根据此值反向查找枚举对象。
在Spring Boot的配置文件中，我们需要指定MyBatis Plus的`typeHandlersPackage`，这样MyBatis Plus在扫描时会找到我们的枚举处理器：
```yaml
mybatis-plus:
type-handlers-package: cn.pilipa.account.cerebrum.client.enums
```
确保`type-handlers-package`的值与`UniversalEnumHandler`类所在的包路径匹配。
通过以上步骤，我们已经成功地实现了Spring Boot MyBatis中的枚举映射。现在，当我们在实体类中使用`YesOrNo`枚举类型，并映射到整型字段时，MyBatis Plus会自动调用我们自定义的枚举处理器，实现枚举与数据库字段之间的转换，从而避免了因数据类型不匹配导致的异常。这种方法提高了代码的可读性和可维护性，同时也简化了开发过程。
