前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >mybatis扩展之自定义类型处理器处理枚举类型

mybatis扩展之自定义类型处理器处理枚举类型

作者头像
西西嘛呦
发布2020-08-26 14:15:33
6730
发布2020-08-26 14:15:33
举报

全局配置:

代码语言:javascript
复制
    <typeHandlers>
        <typeHandler handler="com.gong.mybatis.typeHandler.MyEnumEmpStatusTypeHandler"
        javaType="com.gong.mybatis.bean.EmpStatus"/>
        <!-- <typeHandler handler="org.apache.ibatis.type.EnumOrdinalTypeHandler" 
        javaType="com.gong.mybatis.bean.EmpStatus"/> -->
    </typeHandlers>

MyEnumEmpStatusTypeHandler.java

代码语言:javascript
复制
package com.gong.mybatis.typeHandler;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.TypeHandler;

import com.gong.mybatis.bean.EmpStatus;

/**
 * 1、实现TypeHandler接口。或者继承BaseTypeHandler
 *
 *
 */
public class MyEnumEmpStatusTypeHandler implements TypeHandler<EmpStatus> {

    /**
     * 定义当前数据如何保存到数据库中
     */
    @Override
    public void setParameter(PreparedStatement ps, int i, EmpStatus parameter,
            JdbcType jdbcType) throws SQLException {
        // TODO Auto-generated method stub
        System.out.println("要保存的状态码:"+parameter.getCode());
        ps.setString(i, parameter.getCode().toString());
    }

    @Override
    public EmpStatus getResult(ResultSet rs, String columnName)
            throws SQLException {
        // TODO Auto-generated method stub
        //需要根据从数据库中拿到的枚举的状态码返回一个枚举对象
        int code = rs.getInt(columnName);
        System.out.println("从数据库中获取的状态码:"+code);
        EmpStatus status = EmpStatus.getEmpStatusByCode(code);
        return status;
    }

    @Override
    public EmpStatus getResult(ResultSet rs, int columnIndex)
            throws SQLException {
        // TODO Auto-generated method stub
        int code = rs.getInt(columnIndex);
        System.out.println("从数据库中获取的状态码:"+code);
        EmpStatus status = EmpStatus.getEmpStatusByCode(code);
        return status;
    }

    @Override
    public EmpStatus getResult(CallableStatement cs, int columnIndex)
            throws SQLException {
        // TODO Auto-generated method stub
        int code = cs.getInt(columnIndex);
        System.out.println("从数据库中获取的状态码:"+code);
        EmpStatus status = EmpStatus.getEmpStatusByCode(code);
        return status;
    }

}

EmpSatus.java

代码语言:javascript
复制
package com.gong.mybatis.bean;

public enum EmpStatus {
    LOGIN(100,"用户登录"),LOGOUT(200,"用户登出"),REMOVE(300,"用户不存在");
    private Integer code;
    private String msg;
    private EmpStatus(Integer code,String msg) {
        this.code = code;
        this.msg = msg;
    }
    public Integer getCode() {
        return code;
    }
    public void setCode(Integer code) {
        this.code = code;
    }
    public String getMsg() {
        return msg;
    }
    public void setMsg(String msg) {
        this.msg = msg;
    }
    public static EmpStatus getEmpStatusByCode(int code) {
        // TODO Auto-generated method stub
        switch(code){
        case 100:
            return LOGIN;
        case 200:
            return LOGOUT;
        case 300:
            return REMOVE;
        default:        
            return LOGOUT;
        }
    }
}

Employee.java

代码语言:javascript
复制
    private Integer id;
    private String lastName;
    private String gender;
    private String email;
    private EmpStatus empStatus=EmpStatus.LOGOUT;
    Department dept;

测试:

代码语言:javascript
复制
package com.gong.mybatis.test;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Test;

import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.gong.mybatis.bean.Department;
import com.gong.mybatis.bean.Employee;
import com.gong.mybatis.dao.EmployeeMapper;
import com.gong.mybatis.mapper.EmployeeMapperDynamicSql;

public class TestMybatis5 {
    
    public SqlSessionFactory getSqlSessionFactory() throws IOException {
        String resource = "mybatis-config.xml";
        InputStream is = Resources.getResourceAsStream(resource);
        return new SqlSessionFactoryBuilder().build(is);
    }

    @Test
    public void test() throws IOException {
        SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();
        SqlSession openSession = sqlSessionFactory.openSession();
        
        try {
            EmployeeMapper mapper = openSession.getMapper(EmployeeMapper.class);
            //第一个参数为当前页码,第二个参数是每页的记录数
            Page<Object> page = PageHelper.startPage(1,2);
            List<Employee> emps = mapper.getAllEmp();
            //第一个参数为传入的数据,第二个参数为连续显示多少页
            PageInfo<Employee> pageInfo = new PageInfo<>(emps,3);
            for(Employee emp:emps) {
                System.out.println(emp);
            }
//            System.out.println("当前页码:"+page.getPageNum());
//            System.out.println("总记录数:"+page.getTotal());
//            System.out.println("每页几条记录:"+page.getPageSize());
//            System.out.println("总页码数:"+page.getPages());
            System.out.println("当前页码:"+pageInfo.getPageNum());
            System.out.println("总记录数:"+pageInfo.getTotal());
            System.out.println("每页几条记录:"+pageInfo.getPageSize());
            System.out.println("总页码数:"+pageInfo.getPages());
            System.out.println("是否第一页:"+pageInfo.isIsFirstPage());
            System.out.println("是否最后一页:"+pageInfo.isIsLastPage());
            System.out.println("连续显示的页码:");
            int[] navigatepageNums = pageInfo.getNavigatepageNums();
            for (int i = 0; i < navigatepageNums.length; i++) {
                System.out.println(navigatepageNums[i]);
            }
        } finally {
            // TODO: handle finally clause
            openSession.close();
        }
    }
    
    @Test
    public void testEnum() throws IOException {
        SqlSessionFactory sqlSessionFactory = getSqlSessionFactory();
        SqlSession openSession = sqlSessionFactory.openSession();
        
        try {
            EmployeeMapper mapper = openSession.getMapper(EmployeeMapper.class);
            Employee employee = new Employee();
            employee.setLastName("bom");
            employee.setGender("1");
            employee.setEmail("bom@qq.com");
            mapper.addEmp(employee);
            System.out.println("保存成功:"+employee.getId());
            openSession.commit();
        } finally {
            // TODO: handle finally clause
            openSession.close();
        }        
    }
}

结果:

代码语言:javascript
复制
DEBUG 01-23 16:02:56,976 ==>  Preparing: insert into tbl_employee(last_name,gender,email,empStatus) values(?,?,?,?)   (BaseJdbcLogger.java:145) 
要保存的状态码:200
DEBUG 01-23 16:02:57,030 ==> Parameters: bom(String), 1(String), bom@qq.com(String), 200(String)  (BaseJdbcLogger.java:145) 
DEBUG 01-23 16:02:57,046 <==    Updates: 1  (BaseJdbcLogger.java:145) 
保存成功:11

数据库中:

存储的是自己定义的枚举类型中的code。

再进行查询:

代码语言:javascript
复制
            Employee employee = mapper.getEmpById(11);
            System.out.println(employee);
            System.out.println(employee.getEmpstatus());
            openSession.commit();

输出:

代码语言:javascript
复制
DEBUG 01-23 16:11:09,121 ==>  Preparing: select id,last_name lastName,email,gender,empStatus from tbl_employee where id = ?   (BaseJdbcLogger.java:145) 
DEBUG 01-23 16:11:09,180 ==> Parameters: 11(Integer)  (BaseJdbcLogger.java:145) 
从数据库中获取的状态码:200
DEBUG 01-23 16:11:09,237 <==      Total: 1  (BaseJdbcLogger.java:145) 
Employee [id=11, lastName=bom, gender=1, email=bom@qq.com, dept=null]
LOGOUT

说明确实是定义成功的。

同时,

代码语言:javascript
复制
        <!--2、也可以在处理某个字段的时候告诉MyBatis用什么类型处理器
                保存:#{empStatus,typeHandler=xxxx}
                查询:
                    <resultMap type="com.atguigu.mybatis.bean.Employee" id="MyEmp">
                         <id column="id" property="id"/>
                         <result column="empStatus" property="empStatus" typeHandler=""/>
                     </resultMap>
                注意:如果在参数位置修改TypeHandler,应该保证保存数据和查询数据用的TypeHandler是一样的。
          -->
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2020-01-23 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档