在一个pojo类中定义了一个枚举类型:
package com.springboot.chapter05.pojo;
import com.springboot.chapter05.enumeration.SexEnum;
public class JDBCUser {
private Integer id;
private String user_name;
private SexEnum sex; //枚举类型
private String note;
public SexEnum getSex() {
return sex;
}
public void setSex(SexEnum sex) {
this.sex = sex;
}
@Override
public String toString() {
return "JDBCUser [id=" + id + ", user_name=" + user_name + ", sex=" + sex + ", note=" + note + "]";
}
}
枚举类的实现:
package com.springboot.chapter05.enumeration;
public enum SexEnum {
MALE(1,"男"),
FEMALE(2,"女");
//成员变量
private int id;
private String name;
//构造方法
SexEnum(int id,String name){
this.id=id;
this.name=name;
}
//普通方法,根据ID得到值,ID可以是从前端传过来的整型变量,getEnumById转换成SexEnum类型
public static SexEnum getEnumById(int id) {
for(SexEnum sex : SexEnum.values()) {
if(sex.getId()==id) {
return sex;
}
}
return null;
}
//SexEnum类型转换成int类型
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
}
使用JPA(Hibernate)操作数据时
因为属性列名和数据库字段要一致,所以这里的枚举类型需要转换
使用@Convert指定转换类
@Convert(converter = SexConverter.class) //定义转换器
private SexEnum sex;
这里将SexConverter作为转换器,具体代码如下:
package com.springboot.chapter05.converter;
import javax.persistence.AttributeConverter;
import com.springboot.chapter05.enumeration.SexEnum;
public class SexConverter implements AttributeConverter<SexEnum, Integer> {
//将枚举转换成数据库列
@Override
public Integer convertToDatabaseColumn(SexEnum attribute) {
// TODO Auto-generated method stub
return attribute.getId();
}
//将数据库列转换成枚举
@Override
public SexEnum convertToEntityAttribute(Integer dbData) {
// TODO Auto-generated method stub
return SexEnum.getEnumById(dbData);
}
}
这个类定义了从数据库读出的转换规则和从属性装换为数据库列的规则,这样就能够使性别枚举类与数据库的列对应起来了
下面简单介绍一下AttributeConverter<X,Y>接口:
AttributeConverterr<X,Y>是实体属性类型转换器,主要用来持久化enum,加解密数据,持久化日期,它有两个方法:
- y convertToDatabaseColumn(x)的作用:将实体属性X转化为Y存到数据库中,即插入和更新操作时使用
- x convertToEntityAttribute(y)的作用:将数据库列y转化为实体属性x,即查询操作时使用
使用MyBatis操作数据时
在MyBatis中,枚举可以通过typeHandler进行数据转换的。mybatis对于typehandler的要求是实现typehandler接口,而他自身为了更方便也通过抽象类basetypehandler实现了typehandler接口。所以自定义转换类的时候直接继承抽象类basetypehandler
package com.springboot.chapter05.typeHandler;
import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedJdbcTypes;
import org.apache.ibatis.type.MappedTypes;
import com.springboot.chapter05.enumeration.SexEnum;
//声明jdbcType为整型
@MappedJdbcTypes(JdbcType.INTEGER)
//声明javatype为SexEnum
@MappedTypes(value = SexEnum.class)
public class SexTypeHandler extends BaseTypeHandler<SexEnum> {
//通过列明获取性别
@Override
public SexEnum getNullableResult(ResultSet rs, String columnName) throws SQLException {
// TODO Auto-generated method stub
int sex = rs.getInt(columnName);
if(sex!=1&&sex!=2) {
return null;
}
return SexEnum.getEnumById(sex);
}
//通过下标获取性别
@Override
public SexEnum getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
// TODO Auto-generated method stub
int sex = rs.getInt(columnIndex);
if(sex!=1&&sex!=2) {
return null;
}
return SexEnum.getEnumById(sex);
}
//通过存储过程获取性别
@Override
public SexEnum getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
// TODO Auto-generated method stub
int sex = cs.getInt(columnIndex);
if(sex!=1&&sex!=2) {
return null;
}
return SexEnum.getEnumById(sex);
}
//设置非空性别参数
@Override
public void setNonNullParameter(PreparedStatement ps, int i, SexEnum sex, JdbcType jdbcType)
throws SQLException {
// TODO Auto-generated method stub
ps.setInt(i, sex.getId());
}
}
其中注解@MappedJdbcTypes声明jcbctype为数据库的整型,@MappedTypes声明javatype为sexenum,这样mybatis即可据此对对应的数据类型进行转换了