百度360必应搜狗淘宝本站头条
当前位置:网站首页 > IT知识 > 正文

原生JDBC简单实现Mybatis核心功能

liuian 2025-03-04 13:04 42 浏览

前言

之前在Vertx项目中使用Tdengine,但Vertx没有Tdengine的异步JDBC驱动。所以只能使用Tdengine提供的同步JDBC驱动配合vertx.executeBlocking实现异步数据库查询。
原生的JDBC在SQL参数绑定和返回数据映射时很不方便。但当时第一印象是Mybatis和Spring结合太紧密了(实际是可以的哈),所以自己写了一个简单的SQL解析和返回数据映射的简单JDBC工具。但不能像Mybaits那样支持复杂的嵌套的类参数绑定和映射,但已经够用了。

首先要实现的功能:

  • 实现SQL参数绑定,支持实体类和MAP绑定到SQL
  • 实现返回值映射到实体类

实现效果:

public class PropertyMapper extends BaseMapper {


    public PropertyMapper(DataSource dataSource) {
        super(dataSource);
    }
    
    public Integer add(PropertyEntity propertyEntity) {
        // 实体类参数绑定到SQL
        return update("INSERT INTO property VALUES (#{ts}, #{value});", propertyEntity);
    }


    public List page(Long startTime, Long endTime, Long start, Long size) {
        Map params = new MapBuilder().put("startTime", startTime)
                .put("endTime", endTime)
                .put("start", start)
                .put("size", size)
                .build();
         // MAP参数绑定到SQL
        return selectList("SELECT ts, value, property_code FROM property WHERE ts >= #{startTime} AND ts <= #{endTime} ORDER BY ts DESC LIMIT #{size} OFFSET #{start}", params, PropertyEntity.class);
    }
    

public class  PropertyEntity {

    /**
     * 时间戳, 精确到ms
     */
    @TableField(typeHandler = OffsetDateTimeToLongHandler.class, value = "ts")
    private Long ts;


    /**
     * 值, 不同产品属性定义的超级表值类型不同
     */
    private Object value;
    
     /**
     * 属性编码
     */
    private String propertyCode;
}

对比下原生实现,原生的很多重复代码,同时预编译SQL中参数占位符号都是?区分度差了,每次写的时候还要将参数和?的序列对应起来(JDBC中?index从1开始)可能导致绑定参数错误,

PreparedStatementpsmt = null; 
ResultSetrs = null;
try {
    Connectionconn = null;
    conn =JdbcUtils.getConnection(); 
    Stringsql = "SELECT ts, value, property_code FROM property WHERE ts >= ? AND ts <= ? ORDER BY ts DESC LIMIT ? OFFSET ?";
    psmt =conn.prepareStatement(sql);
    //绑定参数到预编译SQL
    psmt.setLong(1,startTime);
    psmt.setLong(1,endTime);
    psmt.setLong(1,start);
    psmt.setLong(1,size);
    rs =psmt.executeQuery();
    List result = new ArrayList<>();
    while(rs.next()){
        //解析返回值
        Long ts = rs.getLong("ts"); 
        Object value = rs.getObject("value"); 
        String propertyCode = rs.getObject("property_code"); 
        PropertyEntity property = new PropertyEntity();
        property.setTs(ts);
        property.setValue(value);
        property.setPropertyCode(propertyCode);
        result.add(property);
    } 
    return result;
} catch(Exception e) { 
    throw new RuntimeException(e);
} finally{
    JdbcUtils.closeResource(conn,psmt, rs); 
}

实现SQL参数绑定,支持实体类和MAP绑定到SQL

INSERT INTO property VALUES (#{ts}, #{value});
SELECT ts, value, property_code FROM property WHERE ts >= #{startTime} AND ts <= #{endTime} ORDER BY ts DESC LIMIT #{size} OFFSET #{start}
  1. 第一步将SQL中类似#{ts}替换成?,同时获取到参数占位标识符ts
  2. 解析传进来的参数,如果是实体类就转换成Map,Key是属性名称,Value是属性值。如果是Map就进行第三步
  3. 第1步获取的参数占位符ts,从第二步解析到的参数Map中获取到参数值存储到顺序List中
  4. 填充预编译SQL参数值
/**
 * obj根据属性名映射到sql 占位符 #{ts}
 */
public static SqlAndParamDTO objMapToSqlParam(String sql, Object po) {
    return toSqlParam(sql, poToMap(po, false));
}


/**
 * map参数更加key映射到sql
 */
public static SqlAndParamDTO toSqlParam(String sql, Map paramMap) {
    if (paramMap == null || paramMap.isEmpty()) {
        return SqlAndParamDTO.builder().paramList(Collections.emptyList()).sql(sql).build();
    }
    List params = new ArrayList<>();
    String regex = "#\{\s*([^\{\}\s]*)\s*\}";
    Pattern pattern = Pattern.compile(regex);
    Matcher matcher = pattern.matcher(sql);
    StringBuilder sb = new StringBuilder();
    String placeholder = "?";
    while (matcher.find()) {
        //占位符
        String poParamName = matcher.group(1);
        Object paramValue = paramMap.get(poParamName);
        matcher.appendReplacement(sb, placeholder);
        params.add(paramValue);
    }
    matcher.appendTail(sb);
    return SqlAndParamDTO.builder().paramList(params).sql(sql).build();
}



/**
 * 将po转Map
 * 属性名称--->属性值
 *
 * @param po         需要转换的对象
 * @param underScore 是否取下划线即注解值
 */
public static Map poToMap(Object po, Boolean underScore) {
    if (po == null) {
        return Collections.emptyMap();
    }
    Map paramMap = new HashMap<>(16);
    // getField只能够获取类的共有属性字段 也就是public修饰的
    Field[] fields = FieldUtil.getAllFields(po.getClass());
    for (Field field : fields) {
        field.setAccessible(true);
        TableField tableField = field.getDeclaredAnnotation(TableField.class);
        String fieldName = underScore ? tableField.value() : field.getName();
        try {
            paramMap.put(fieldName, field.get(po));
        } catch (IllegalAccessException e) {
            throw new RuntimeException("参数绑定出错");
        }
    }
    return paramMap;
}


public static PreparedStatement getPreparedStatement(Connection connection, String sql, List params) throws SQLException {
    PreparedStatement preparedStatement = connection.prepareStatement(sql);
    for (int i = 1; i <= params.size(); i++) {
        //jdbc预编译参数从1起步
        preparedStatement.setObject(i, params.get(i - 1));
    }
    return preparedStatement;
}


@Data
@Builder
public static class SqlAndParamDTO {

    /**
     * 生成的预编译sql
     */
    public String sql;

    /**
     * 生成对应预编译参数位置的数组
     */
    private List paramList;
}


?

这里没做集合类型支持,比如当参数类型为List时可以生成对应数量的?,golang中的gorm也是这样实现的。

核心BaseMapper
很神奇只需要写两个方法就可以解决所有SQL查询了
SQL只有两类更新SQL和查询SQL,分别对应JDBC的 preparedStatement.executeQuery(),reparedStatement.executeUpdate()。对应我们的也就两个函数:selectList和update

public class BaseMapper {


    private DataSource dataSource;


    /**
     * getOne之类查询
     *
     * @param sql      sql
     * @param valueMap 参数, 因为查询一般参数都是零散的没有提供Obj参数绑定方法要写页很简单
     * @param rClass   返回值类型
     */
    public  R selectOne(String sql, Map valueMap, Class rClass) {
        List result = selectList(sql, valueMap, rClass);
        if (result == null || result.isEmpty()) {
            return null;
        }
        if (result.size() == 1) {
            return result.get(0);
        }
        throw new RuntimeException("except one but more than one");
    }


    /**
     * 查询list
     *
     * @param sql      sql
     * @param valueMap 参数
     * @param rClass   返回值类型
     */
    public  List selectList(String sql, Map valueMap, Class rClass) {
        ParamUtil.SqlAndParamDTO sqlAndParamDTO = ParamUtil.toSqlParam(sql, valueMap);
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        ResultSet resultSet = null;
        try {
            connection = dataSource.getConnection();
            preparedStatement = getPreparedStatement(connection, sqlAndParamDTO.getSql(), sqlAndParamDTO.getParamList());
            resultSet = preparedStatement.executeQuery();
            return ResultConvertUtil.listFromResultSet(resultSet, rClass);
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            close(resultSet, preparedStatement, connection);
        }
    }


    /**
     * 更新sql: insert, delete, update
     *
     * @param sql      sql
     * @param paramObj Obj参数
     * @return {@link Future}<{@link Integer}>
     */
    public Integer update(String sql, Object paramObj) {
        //将Object转换为Map
        return update(sql, ParamUtil.poToMap(paramObj, false));
    }

    /**
     * 更新sql: insert, delete, update
     *
     * @param sql      sql
     * @param paramMap map类型参数
     * @return {@link Future}<{@link Integer}>
     */
    public Integer update(String sql, Map paramMap) {
        ParamUtil.SqlAndParamDTO sqlAndParamDTO = ParamUtil.toSqlParam(sql, paramMap);
        Connection connection = null;
        PreparedStatement preparedStatement = null;
        try {
            connection = dataSource.getConnection();
            preparedStatement = getPreparedStatement(connection, sqlAndParamDTO.getSql(), sqlAndParamDTO.getParamList());
            return preparedStatement.executeUpdate();
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            //关闭资源让线程池回收
            close(null, preparedStatement, connection);
        }
    }
}

实现返回值映射到实体类 Tdengine返回的是ResultSet和其他JDBC驱动有些不同,Postgres返回的是 RowSet rowSet但实现是一样的。主要步骤就是:

  1. 从返回的数据行中解析出列名称-列值
  2. 反射出需要返回的对象,根据属性值和列名称对应起来给属性赋值
/**
 * 从resultSet获取结果
 * @param resultSet
 * @param targetClass
 * @return {@link List}<{@link R}>
 * @throws SQLException
 */
public static  List listFromResultSet(ResultSet resultSet, Class targetClass) throws SQLException {
    if (resultSet == null) {
        return Collections.emptyList();
    }
    List resultList = new ArrayList<>(20);
    //如果是基本类型或者String直接取第一个返回
    if (CommonUtil.isPrimitive(targetClass) || String.class == targetClass) {
        while (resultSet.next()) {
            if (String.class == targetClass) {
                resultList.add((R) resultSet.getString(1));
            } else {
                resultList.add((R) resultSet.getObject(1));
            }
        }
        return resultList;
    }
    //如果是对象映射结果返回
    while (resultSet.next()) {
        Map sqlRowToValueMap = new HashMap<>(8);
        ResultSetMetaData rowMetaData = resultSet.getMetaData();
        int columnCount = rowMetaData.getColumnCount();
        for (int i = 1; i <= columnCount; i++) {
            sqlRowToValueMap.put(rowMetaData.getColumnLabel(i), resultSet.getObject(i));
        }
        resultList.add(mapToObj(sqlRowToValueMap, targetClass));
    }
    return resultList;
}


/**
 * map值映射Obj
 *
 * @param valueMap
 * @param rClass
 * @return {@link R}
 */
public static  R mapToObj(Map valueMap, Class rClass) {
    try {
        //反射出对象
        R r = rClass.getDeclaredConstructor().newInstance();
        Field[] fields = FieldUtil.getAllFields(rClass);
        for (Field field : fields) {
            field.setAccessible(true);
            TableField tableField = field.getDeclaredAnnotation(TableField.class);
            TypeHandler typeHandler = null;
            if (tableField != null && tableField.typeHandler() != null) {
                typeHandler = tableField.typeHandler().getDeclaredConstructor().newInstance();
            }
            //如果有注解用注解没有用字段名改为下划线,支持注解定义别名。同时避免数据库short和java int long转换异常提供了变量提升
            String realFieldName = tableField == null ? poParamNameToDb(field.getName()) : tableField.value();
            field.set(r, typeHandler == null || typeHandler instanceof UnknownTypeHandler ? varLift(valueMap.get(realFieldName), field.getType()) : typeHandler.getResult(valueMap.get(realFieldName)));
        }
        return r;
    } catch (Exception e) {
        throw new RuntimeException("sql result to Object error:", e);
    }
}


/**
 * 变量提升
 *
 * @param oldVar
 * @param targetClass
 * @return {@link Object}
 */
public static Object varLift(Object oldVar, Class targetClass) {
    if (oldVar == null) {
        return null;
    }
    if (oldVar instanceof Integer && targetClass.equals(Long.class)) {
        return ((Integer) oldVar).longValue();
    }
    if (oldVar instanceof Short && targetClass.equals(Long.class)) {
        return ((Short) oldVar).longValue();
    }
    if (oldVar instanceof Short && targetClass.equals(Integer.class)) {
        return ((Short) oldVar).intValue();
    }
    if (targetClass.equals(String.class) && oldVar instanceof byte[]) {
        return new String((byte[]) oldVar, StandardCharsets.UTF_8);
    }
    return oldVar;
}

总结

基本实现了Mybatis核心功能,其实不复杂,但够用了。但是Mybatis支持的太多啦,加上需要整合SpringBoot所以比较复杂。

原文:https://juejin.cn/post/7408849436431073316

作者:南瓜大人

#记录我的9月生活#

相关推荐

驱动网卡(怎么从新驱动网卡)
驱动网卡(怎么从新驱动网卡)

网卡一般是指为电脑主机提供有线无线网络功能的适配器。而网卡驱动指的就是电脑连接识别这些网卡型号的桥梁。网卡只有打上了网卡驱动才能正常使用。并不是说所有的网卡一插到电脑上面就能进行数据传输了,他都需要里面芯片组的驱动文件才能支持他进行数据传输...

2026-01-30 00:37 liuian

win10更新助手装系统(微软win10更新助手)

1、点击首页“系统升级”的按钮,给出弹框,告诉用户需要上传IMEI码才能使用升级服务。同时给出同意和取消按钮。华为手机助手2、点击同意,则进入到“系统升级”功能华为手机助手华为手机助手3、在检测界面,...

windows11专业版密钥最新(windows11专业版激活码永久)

 Windows11专业版的正版密钥,我们是对windows的激活所必备的工具。该密钥我们可以通过微软商城或者通过计算机的硬件供应商去购买获得。获得了windows11专业版的正版密钥后,我...

手机删过的软件恢复(手机删除过的软件怎么恢复)
手机删过的软件恢复(手机删除过的软件怎么恢复)

操作步骤:1、首先,我们需要先打开手机。然后在许多图标中找到带有[文件管理]文本的图标,然后单击“文件管理”进入页面。2、进入页面后,我们将在顶部看到一行文本:手机,最新信息,文档,视频,图片,音乐,收藏,最后是我们正在寻找的[更多],单击...

2026-01-29 23:55 liuian

一键ghost手动备份系统步骤(一键ghost 备份)

  步骤1、首先把装有一键GHOST装系统的U盘插在电脑上,然后打开电脑马上按F2或DEL键入BIOS界面,然后就选择BOOT打USDHDD模式选择好,然后按F10键保存,电脑就会马上重启。  步骤...

怎么创建局域网(怎么创建局域网打游戏)

  1、购买路由器一台。进入路由器把dhcp功能打开  2、购买一台交换机。从路由器lan端口拉出一条网线查到交换机的任意一个端口上。  3、两台以上电脑。从交换机任意端口拉出网线插到电脑上(电脑设置...

精灵驱动器官方下载(精灵驱动手机版下载)

是的。驱动精灵是一款集驱动管理和硬件检测于一体的、专业级的驱动管理和维护工具。驱动精灵为用户提供驱动备份、恢复、安装、删除、在线更新等实用功能。1、全新驱动精灵2012引擎,大幅提升硬件和驱动辨识能力...

一键还原系统步骤(一键还原系统有哪些)

1、首先需要下载安装一下Windows一键还原程序,在安装程序窗口中,点击“下一步”,弹出“用户许可协议”窗口,选择“我同意该许可协议的条款”,并点击“下一步”。  2、在弹出的“准备安装”窗口中,可...

电脑加速器哪个好(电脑加速器哪款好)

我认为pp加速器最好用,飞速土豆太懒,急速酷六根本不工作。pp加速器什么网页都加速,太任劳任怨了!以上是个人观点,具体性能请自己试。ps:我家电脑性能很好。迅游加速盒子是可以加速电脑的。因为有过之...

任何u盘都可以做启动盘吗(u盘必须做成启动盘才能装系统吗)

是的,需要注意,U盘的大小要在4G以上,最好是8G以上,因为启动盘里面需要装系统,内存小的话,不能用来安装系统。内存卡或者U盘或者移动硬盘都可以用来做启动盘安装系统。普通的U盘就可以,不过最好U盘...

u盘怎么恢复文件(u盘文件恢复的方法)

开360安全卫士,点击上面的“功能大全”。点击文件恢复然后点击“数据”下的“文件恢复”功能。选择驱动接着选择需要恢复的驱动,选择接入的U盘。点击开始扫描选好就点击中间的“开始扫描”,开始扫描U盘数据。...

系统虚拟内存太低怎么办(系统虚拟内存占用过高什么原因)

1.检查系统虚拟内存使用情况,如果发现有大量的空闲内存,可以尝试释放一些不必要的进程,以释放内存空间。2.如果系统虚拟内存使用率较高,可以尝试增加系统虚拟内存的大小,以便更多的应用程序可以使用更多...

剪贴板权限设置方法(剪贴板访问权限)
剪贴板权限设置方法(剪贴板访问权限)

1、首先打开iphone手机,触碰并按住单词或图像直到显示选择选项。2、其次,然后选取“拷贝”或“剪贴板”。3、勾选需要的“权限”,最后选择开启,即可完成苹果剪贴板权限设置。仅参考1.打开苹果手机设置按钮,点击【通用】。2.点击【键盘】,再...

2026-01-29 21:37 liuian

平板系统重装大师(平板重装win系统)

如果你的平板开不了机,但可以连接上电脑,那就能好办,楼主下载安装个平板刷机王到你的个人电脑上,然后连接你的平板,平板刷机王会自动识别你的平板,平板刷机王上有你平板的我刷机包,楼主点击下载一个,下载完成...

联想官网售后服务网点(联想官网售后服务热线)

联想3c服务中心是联想旗下的官方售后,是基于互联网O2O模式开发的全新服务平台。可以为终端用户提供多品牌手机、电脑以及其他3C类产品的维修、保养和保险服务。根据客户需求层次,联想服务针对个人及家庭客户...