Mybatis处理枚举（快速上手）

现在有一个原料类需要存入数据库，并且查询。

@Data
@NoArgsConstructor
@AllArgsConstructor
@Alias(value = "ingredient")

public class Ingredient {

    private String id;
    private String name;
    private IngredientEnum type;
    private String info;

}

 其中IngredientEnum字段是枚举类型，表示原料的类型。

public enum IngredientEnum {
    WRAP, PROTEIN, VEGGIES, CHEESE, SAUCE;

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

 首先，编写TypeHandler类来处理枚举转换：

@MappedJdbcTypes(JdbcType.VARCHAR)

@MappedTypes(value = IngredientEnum.class)
public class IngredientsTypeHandler extends BaseTypeHandler<IngredientEnum> {

    //设置数据
    @Override
    public void setNonNullParameter(PreparedStatement preparedStatement, int i, IngredientEnum ingredientEnum, JdbcType jdbcType) throws SQLException {
        preparedStatement.setString(i,ingredientEnum.name());
    }
    //根据名字查找
    @Override
    public IngredientEnum getNullableResult(ResultSet resultSet, String s) throws SQLException {
        return IngredientEnum.valueOf(resultSet.getString(s));
    }
    //根据位置查找
    @Override
    public IngredientEnum getNullableResult(ResultSet resultSet, int i) throws SQLException {
        return IngredientEnum.valueOf(resultSet.getString(i));
    }
    //根据存储模过程来查找
    @Override
    public IngredientEnum getNullableResult(CallableStatement callableStatement, int i) throws SQLException {
        return IngredientEnum.valueOf(callableStatement.getString(i));
    }
}

其次，在对应的mapper.xml文件中，在需要转换的字段上添加上需要的typehandler转换 。

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE mapper
        PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
        "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="tacos.data.mapper.IngredientMapper">
    <insert id="add" parameterType="tacos.pojo.entity.Ingredient" >
        insert into ingredient(id,name,ingredientType) values (#{id},#{name},#{type,typeHandler=tacos.data.typehandlers.IngredientsTypeHandler})
    </insert>

    <select id="getIngredient" parameterType="string" resultMap="BaseResultMap">
        select * from ingredient where name=#{name}
    </select>
    <resultMap id="BaseResultMap" type="tacos.pojo.entity.Ingredient">
        <!-- 定义主键 ,非常重要。如果是多个字段,则定义多个id -->
        <!-- property：主键在pojo中的属性名 -->
        <!-- column：主键在数据库中的列名 -->
        <id property="id" column="id" jdbcType="VARCHAR"/>
        <result property="name" column="name" jdbcType="VARCHAR"/>
        <result property="type" column="ingredientType" jdbcType="VARCHAR" typeHandler="tacos.data.typehandlers.IngredientsTypeHandler"/>
        <result property="info" column="info" jdbcType="VARCHAR"/>
    </resultMap>

</mapper>

最后，调用mapper接口函数即可实现。 

@SpringBootTest
class TacoCloudApplicationTests {
    @Resource
    DataSource dataSource;
    @Resource
    IngredientMapper ingredientMapper;

    @Test
    public void testMybatis() throws Exception {
        ingredientMapper.add(new Ingredient("FLTO","Flour Tortilla", IngredientEnum.WRAP,""));
        ingredientMapper.add(new Ingredient("COTO","Corn Tortilla", IngredientEnum.WRAP,""));
        ingredientMapper.add(new Ingredient("GRBF","Ground Beef", IngredientEnum.PROTEIN,""));
        ingredientMapper.add(new Ingredient("CARN","Carnitas", IngredientEnum.PROTEIN,""));
        ingredientMapper.add(new Ingredient("TMTO","Diced Tomatoes", IngredientEnum.VEGGIES,""));
        ingredientMapper.add(new Ingredient("LETC","Lettuce", IngredientEnum.VEGGIES,""));
        ingredientMapper.add(new Ingredient("CHED","Cheddar", IngredientEnum.CHEESE,""));
        ingredientMapper.add(new Ingredient("JACK","Monterrey Jack", IngredientEnum.CHEESE,""));
        ingredientMapper.add(new Ingredient("SLSA","Salsa", IngredientEnum.SAUCE,""));
        ingredientMapper.add(new Ingredient("SRCR","Sour Cream", IngredientEnum.SAUCE,""));
        System.out.println(1);
    }

    @Test
    public void testMybatis2() throws Exception{
        System.out.println(ingredientMapper.getIngredient("Sour Cream"));
        System.out.println(2);
    }



}