美文网首页
在gradle6.9环境中对Mybatis Generators

在gradle6.9环境中对Mybatis Generators

作者: 冬天里的懒喵 | 来源:发表于2021-09-07 16:15 被阅读0次

在进行mybatis开发的过程中,我们经常会遇到此类问题:如mysql的表中的日期为字符串,但是我们在java中希望按照Date类型进行操作。因此,我们需要用到mybatis中的typeHandler。
本文就是一个将数据库中 String类型存储的字段 createDate
,其格式为“yyyy-MM-ddHH:mm:ss”,通过TypeHandler转换为Date操作的实例。

1.TypeHandler

自定义的TypeHandler需要继承BaseTypeHandler。代码如下:

import com.google.common.base.Strings;
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;
import java.text.SimpleDateFormat;
import java.util.Date;

@MappedJdbcTypes(JdbcType.VARCHAR)
@MappedTypes(Date.class)
public class MyDateTypeHandler extends BaseTypeHandler<Date> {

    private SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss SSS");

    private Date convert(String source) {
        Date result = new Date(0);
        try {
            if (!Strings.isNullOrEmpty(source)) {
                result = dateFormat.parse(source);
            }
        } catch (Exception e) {
            e.printStackTrace();
        } 
        return result;
    }
    

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, Date parameter, JdbcType jdbcType) throws SQLException {
        String result = dateFormat.format(parameter);
        ps.setString(i, result);
    }

    @Override
    public Date getNullableResult(ResultSet rs, String columnName) throws SQLException {
        String source = rs.getString(columnName);
        return convert(source);
    }

    @Override
    public Date getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        String source = rs.getString(columnIndex);
        return convert(source);
    }

    @Override
    public Date getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        String source = cs.getString(columnIndex);
        return convert(source);
    }
}

在这个类中,首先需要通过注解,@MappedJdbcTypes(JdbcType.VARCHAR)定义在数据库中的数据类型。@MappedTypes(Date.class)定义java对象中的数据类型。
MyDateTypeHandler 继承BaseTypeHandler
,需要实现四个方法,其中一个是将Date转换为String的setNonNullParameter方法,其他三个都是将数据库操作结果的ResultSet中的String转换为String的方法。
在这个示例中,如果字符串为空,则将创建一个默认为Date(0)的对象。

2.generatorConfig.xml

由于将采用generators逆向生成ORM代码,因此,需要在table的定义中增加columnOverride标签对Handler进行配置,这样逆向工程的时候就会创建所需要的代码。

<table tableName="users" domainObjectName="Users">
    <columnOverride column="createDate" jdbcType="VARCHAR" javaType="java.util.Date" typeHandler="com.dhb.gts.javacourse.week5.typehandler.MyDateTypeHandler"/>
</table>

3.代码生成

通过 grale->other->MybatisGenerate 就能生成所需要的代码.


mybatis generator 生成代码

可以看到此时生成的代码entity中,users的createDate的类型为Date了。


Entity中Users的CreateDate类型

主要的操作是在UsersMapper中进行了修改,@insert和select方法的@Results中都添加了MyDateTypeHandler的处理。

package com.dhb.gts.javacourse.week5.mapper;

import com.dhb.gts.javacourse.week5.entity.Users;
import com.dhb.gts.javacourse.week5.entity.UsersExample;
import com.dhb.gts.javacourse.week5.typehandler.MyDateTypeHandler;
import java.util.List;
import org.apache.ibatis.annotations.DeleteProvider;
import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.InsertProvider;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Result;
import org.apache.ibatis.annotations.Results;
import org.apache.ibatis.annotations.SelectProvider;
import org.apache.ibatis.annotations.UpdateProvider;
import org.apache.ibatis.type.JdbcType;

public interface UsersMapper {
    @SelectProvider(type=UsersSqlProvider.class, method="countByExample")
    long countByExample(UsersExample example);

    @DeleteProvider(type=UsersSqlProvider.class, method="deleteByExample")
    int deleteByExample(UsersExample example);

    @Insert({
        "insert into users (id, username, ",
        "`password`, createDate)",
        "values (#{id,jdbcType=INTEGER}, #{username,jdbcType=VARCHAR}, ",
        "#{password,jdbcType=VARCHAR}, #{createdate,jdbcType=VARCHAR,typeHandler=com.dhb.gts.javacourse.week5.typehandler.MyDateTypeHandler})"
    })
    int insert(Users record);

    @InsertProvider(type=UsersSqlProvider.class, method="insertSelective")
    int insertSelective(Users record);

    @SelectProvider(type=UsersSqlProvider.class, method="selectByExample")
    @Results({
        @Result(column="id", property="id", jdbcType=JdbcType.INTEGER),
        @Result(column="username", property="username", jdbcType=JdbcType.VARCHAR),
        @Result(column="password", property="password", jdbcType=JdbcType.VARCHAR),
        @Result(column="createDate", property="createdate", typeHandler=MyDateTypeHandler.class, jdbcType=JdbcType.VARCHAR)
    })
    List<Users> selectByExample(UsersExample example);

    @UpdateProvider(type=UsersSqlProvider.class, method="updateByExampleSelective")
    int updateByExampleSelective(@Param("record") Users record, @Param("example") UsersExample example);

    @UpdateProvider(type=UsersSqlProvider.class, method="updateByExample")
    int updateByExample(@Param("record") Users record, @Param("example") UsersExample example);
}

4.Starter

在springboot中,通过Startler启动,分别插入数据和select进行测试。
启动器:

package com.dhb.gts.javacourse.week5.generator;


import org.mybatis.spring.annotation.MapperScan;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
@MapperScan(basePackages = {"com.dhb.gts.javacourse.week5.mapper,com.dhb.gts.javacourse.week5.entity"})
public class GeneratorStarter {

    public static void main(String[] args) {
        SpringApplication.run(GeneratorStarter.class, args);
    }

}

rest的controller:

package com.dhb.gts.javacourse.week5.generator;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.serializer.SerializerFeature;
import com.dhb.gts.javacourse.week5.entity.Users;
import com.dhb.gts.javacourse.week5.entity.UsersExample;
import com.dhb.gts.javacourse.week5.mapper.UsersMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Date;
import java.util.List;
import java.util.Random;

@RestController
public class GeneratorController {

    @Autowired
    private UsersMapper usersMapper;
    
    private Random random = new Random();

    @RequestMapping("/generatorquery")
    @ResponseBody
    public String generatorQuery(HttpServletRequest request, HttpServletResponse response) {
        UsersExample example = new UsersExample();
        example.createCriteria();

        List<Users> list = usersMapper.selectByExample(example);
        return JSON.toJSONString(list, SerializerFeature.PrettyFormat, SerializerFeature.WriteMapNullValue,
                SerializerFeature.WriteDateUseDateFormat);
    }

    @RequestMapping("/generatorinsert")
    @ResponseBody
    public String generatorInsert(HttpServletRequest request, HttpServletResponse response) {
        Users users = new Users();
        users.setPassword(String.valueOf(System.currentTimeMillis()));
        users.setUsername("generatorinsert"+random.nextInt(100));
        users.setCreatedate(new Date());
        usersMapper.insert(users);
        return "success";
    }
}

此外还需注意的是,在springboot的yml文件中要增加如下内容:

# mybatis typehandler
mybatis.type-handlers-package: com.dhb.gts.javacouse.week5.typehandler

5.运行测试

建表语句如下:

CREATE TABLE users  ( 
    id          int(11) AUTO_INCREMENT NOT NULL,
    username    varchar(50) NULL,
    password    varchar(50) NULL,
    createDate  varchar(50) NULL,
    PRIMARY KEY(id)
)
ENGINE = InnoDB

通过insert接口http://127.0.0.1:8084/generatorinsert插入数据之后,可以看到数据库中的值:

insert之后数据库表中的数据情况

通过query接口查询如下:


query数据库users表的结果

可以看到我们的MyDateTypeHandler对数据进行了转换,在insert和select方法中都起到了作用。

相关文章

网友评论

      本文标题:在gradle6.9环境中对Mybatis Generators

      本文链接:https://www.haomeiwen.com/subject/nzkpwltx.html