溫馨提示×

MyBatis中怎么處理數(shù)據(jù)庫的枚舉類型

小億
125
2024-05-08 14:06:57

在MyBatis中處理數(shù)據(jù)庫的枚舉類型通常有兩種方式：

  1. 使用TypeHandler：MyBatis提供了TypeHandler接口，可以自定義處理數(shù)據(jù)庫字段和Java對象之間的轉(zhuǎn)換。可以實現(xiàn)自定義的TypeHandler來處理枚舉類型，例如：
public class EnumTypeHandler<E extends Enum<E>> extends BaseTypeHandler<E> {
    
    private Class<E> type;
    
    public EnumTypeHandler(Class<E> type) {
        this.type = type;
    }
    
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, E parameter, JdbcType jdbcType) throws SQLException {
        ps.setString(i, parameter.name());
    }
    
    @Override
    public E getNullableResult(ResultSet rs, String columnName) throws SQLException {
        String value = rs.getString(columnName);
        return Enum.valueOf(type, value);
    }
    
    @Override
    public E getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        String value = rs.getString(columnIndex);
        return Enum.valueOf(type, value);
    }
    
    @Override
    public E getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        String value = cs.getString(columnIndex);
        return Enum.valueOf(type, value);
    }
}

然后在MyBatis的配置文件中配置這個TypeHandler，例如：

<typeHandlers>
    <typeHandler handler="com.example.EnumTypeHandler"/>
</typeHandlers>
  1. 使用枚舉的toString方法：另一種方式是在枚舉類中實現(xiàn)toString方法，將枚舉值保存為數(shù)據(jù)庫中的字符串形式，然后直接使用MyBatis的默認TypeHandler來處理枚舉類型，例如：
public enum Status {
    ACTIVE("A"),
    INACTIVE("I");

    private String value;

    Status(String value) {
        this.value = value;
    }

    @Override
    public String toString() {
        return this.value;
    }

    public static Status fromString(String value) {
        for (Status status : Status.values()) {
            if (status.value.equals(value)) {
                return status;
            }
        }
        throw new IllegalArgumentException("Invalid value for Status: " + value);
    }
}

在Mapper接口中使用枚舉類型：

public interface UserMapper {
    
    void insertUser(@Param("name") String name, @Param("status") Status status);
    
    User getUserByName(String name);
}

在配置文件中使用枚舉類型：

<insert id="insertUser" parameterType="map">
    INSERT INTO user (name, status) VALUES (#{name}, #{status})
</insert>

0 踩