mybatis對枚舉類型的處理

有時候，我們在數據庫中會插入一些字段的值時固定的，比如性別字段，它的值固定只有2個，男或者女； 或者季節字段，只有春夏秋冬4種。有時候在Java類中會采用枚舉類型來表達相應的數據庫字段，比如如下數據表：

create table student(
id int auto_increment comment '學生ID',
name varchar(50),
gender varchar(10),
primary key(id));

 

枚舉類定義如下：

package com.example.demo.dao;

public enum Gender {
    FEMAIL("女"),
    MAIL("男");

    private String sex;
    public String getSex() {
        return sex;
    }
    public void setSex(String sex) {
        this.sex = sex;
    }

    Gender(String s){
        this.sex=s;
    }
}

實體類定義如下：

package com.example.demo.dao;

public class Student {
    private int id;
    private String name;
    private Gender gender;

    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;
    }

    public Gender getGender() {
        return gender;
    }

    public void setGender(Gender gender) {
        this.gender = gender;
    }
}

一.存儲枚舉的名稱，Gender在數據庫中值為：FEMAIL

此方式是mybatis對枚舉類型的默認處理方式，使用的類型處理器是org.apache.ibatis.type.EnumTypeHandler

SQL如下：

<insert id="insertStudent" parameterType="com.example.demo.dao.Student">
      insert into student(name,gender) values(#{name},#{gender})
  </insert>

對應的Mapper為：

package com.example.demo.dao.mapper;

import com.example.demo.dao.Student;
import org.apache.ibatis.annotations.Mapper;


@Mapper
public interface StudentMapper {
    int insertStudent(Student student);
}

對應的Controller為：

package com.example.demo.controller;

import com.example.demo.dao.Gender;
import com.example.demo.dao.Student;
import com.example.demo.service.StudentService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@EnableAutoConfiguration
@RequestMapping("/student")
public class StudentController {
    @Autowired
    StudentService studentService;


    @GetMapping("/insertStudent")
    int insertStudent(){
        Student student=new Student();
       student.setName("張三");
       student.setGender(Gender.FEMAIL);
     return studentService.insertStudent(student);
    }
}

 

對應的service為：

 

package com.example.demo.service.impl;

import com.example.demo.dao.Student;
import com.example.demo.dao.mapper.StudentMapper;
import com.example.demo.service.StudentService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;


@Service
public class StudentServiceImpl implements StudentService {
    @Autowired
    StudentMapper studentMapper;

    @Override
    public int insertStudent(Student student) {
        return studentMapper.insertStudent(student);
    }
}

采用postman測試：

 

二.存儲枚舉的索引，FEMAIL在數據庫中值為：0

此方式使用的類型處理器是org.apache.ibatis.type.EnumOrdinalTypeHandler,使用 typeHandler 設置類型處理為org.apache.ibatis.type.EnumOrdinalTypeHandler

修改sql為：

  <insert id="insertStudent" parameterType="com.example.demo.dao.Student">
      insert into student(name,gender) values(#{name},#{gender,typeHandler=org.apache.ibatis.type.EnumOrdinalTypeHandler})
  </insert>

采用Postman測試：

 

 三.存儲枚舉的值，這里存儲“女”

前面兩種方式使用mybatis提供的類型處理器，如果想存儲上面枚舉的值，比如：Gender.FEMAIL的值女,就需要創建一個自定義的類型處理器,只需要繼承BaseTypeHandler

自定義類型處理器如下：

 

package com.example.demo.dao;

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 java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/* 數據庫中的數據類型 */
@MappedJdbcTypes(JdbcType.VARCHAR)
/* 轉化后的數據類型 */
@MappedTypes(value = Gender.class)
public class MyEnumHandler extends BaseTypeHandler<Gender> {
    @Override
    public void setNonNullParameter(PreparedStatement preparedStatement, int i, Gender gender, JdbcType jdbcType) throws SQLException {
        preparedStatement.setString(i,gender.getSex());
    }

    @Override
    public Gender getNullableResult(ResultSet resultSet, String s) throws SQLException {
        String val=resultSet.getString(s);
        return getEnum(val);
    }

    @Override
    public Gender getNullableResult(ResultSet resultSet, int i) throws SQLException {
        return getEnum(resultSet.getString(i));
    }

    @Override
    public Gender getNullableResult(CallableStatement callableStatement, int i) throws SQLException {
        return getEnum(callableStatement.getNString(i));
    }

    /**
     * 根據值獲得對應的枚舉
     * @param val
     * @return
     */
    private Gender getEnum(String val){
        Class<Gender> sexClass = Gender.class;
        Gender[] sexs = sexClass.getEnumConstants();

        for(Gender se:sexs){
            if(se.getSex().equals(val)){
                return se;
            }
        }
        return null;
    }
}

修改sql：

  <insert id="insertStudent" parameterType="com.example.demo.dao.Student">
      insert into student(name,gender) values(#{name},#{gender,typeHandler=com.example.demo.dao.mapper.MyEnumHandler})
  </insert>