mybatis字段映射枚舉類型

在底層使用mybatis的時候，我們可能會需要把表里的字段映射到Java里面的枚舉類，現總結下工作中的用法：

sku表里一個status_type字段為int類型。(這里是postgresql的腳本)

status_type int4 NULL

Java里面對應生成的枚舉類為：

　

public enum SkuStatusType implements DbEnum, LocalString {

    PRODUCT(0, false, "masterdata.skuStatus.product"), 
    BUYING(1, false, "masterdata.skuStatus.buying"), 
    SALE(2, false, "masterdata.skuStatus.sale");


    private int dbConstant;
    private String messageKey;

    private OrderState(int dbConstant, boolean hasBeenDispatched, String messageKey) {
        this.dbConstant = dbConstant;
        this.hasBeenDispatched = hasBeenDispatched;
        this.messageKey = messageKey;
    }

    @Override
    public Integer getConstant() {
        return this.dbConstant;
    }

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

在配置xml文件之前我們需要為自己的項目建一個工具類：

package com.lcc.util;

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 com.lcc.api.domain.DbEnum;

public class DbEnumTypeHandler extends BaseTypeHandler<DbEnum> {
    
    private Class<DbEnum> type; 
    
    public DbEnumTypeHandler(Class<DbEnum> type){
        this.type = type;
    }
    
    
    @Override
    public void setNonNullParameter(PreparedStatement ps, int i,
            DbEnum parameter, JdbcType jdbcType) throws SQLException {
        ps.setInt(i, parameter.getConstant());
    }

    @Override
    public DbEnum getNullableResult(ResultSet rs, String columnName)
            throws SQLException {
        int constant = rs.getInt(columnName);
        if(rs.wasNull()){    
            return null;
        }
        else{
            return convert(constant);
        }
    }

    @Override
    public DbEnum getNullableResult(ResultSet rs, int columnIndex)
            throws SQLException {
        int constant = rs.getInt(columnIndex);
        if(rs.wasNull()){    
            return null;
        }
        else{
            return convert(constant);
        }
    }

    @Override
    public DbEnum getNullableResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        int constant = cs.getInt(columnIndex);
        if(cs.wasNull()){    
            return null;
        }
        else{
            return convert(constant);
        }
    }
    
    private DbEnum convert(int constant) {
        DbEnum[] dbEnums = type.getEnumConstants();
        for (DbEnum dbEnum : dbEnums) {
            if(dbEnum.getConstant().equals(constant)){
                return dbEnum;
            }
        }
        return null;
    }
    
    
}

util class里面的DbEnum domain 和 Localisable domain:（一般項目里的枚舉實現這兩個domain interface）

package com.lcc.api.domain;

import java.io.Serializable;

public interface DbEnum extends Serializable {

    public Integer getConstant();
}

package com.lcc.api.domain;

public interface Localisable {

    String messageKey();
}

 

看下mybatis的xml文件配置：

<result column="status_type" property="statusType"
                typeHandler="com.lcc.util.StringEnumTypeHandler" />

 

此時表字段映射枚舉類就大功告成了，看完之后你可以自己上手試一下。

這只是其中的一種方法，還有其他的方法可以在度娘和谷哥上面搜索一下。