原生JDBC简单实现Mybatis核心功能
ztj100 2025-01-06 16:31 27 浏览 0 评论
前言
之前在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<PropertyEntity> page(Long startTime, Long endTime, Long start, Long size) {
Map<String, Object> 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<PropertyEntity> 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}
- 第一步将SQL中类似#{ts}替换成?，同时获取到参数占位标识符ts
- 解析传进来的参数，如果是实体类就转换成Map，Key是属性名称，Value是属性值。如果是Map就进行第三步
- 第1步获取的参数占位符ts，从第二步解析到的参数Map中获取到参数值存储到顺序List中
- 填充预编译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<String, Object> paramMap) {
if (paramMap == null || paramMap.isEmpty()) {
return SqlAndParamDTO.builder().paramList(Collections.emptyList()).sql(sql).build();
}
List<Object> 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<String, Object> poToMap(Object po, Boolean underScore) {
if (po == null) {
return Collections.emptyMap();
}
Map<String, Object> 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> R selectOne(String sql, Map<String, Object> valueMap, Class<R> rClass) {
List<R> 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 <R> List<R> selectList(String sql, Map<String, Object> valueMap, Class<R> 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<String, Object> 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但实现是一样的。主要步骤就是：
- 从返回的数据行中解析出列名称-列值
- 反射出需要返回的对象，根据属性值和列名称对应起来给属性赋值
/**
* 从resultSet获取结果
* @param resultSet
* @param targetClass
* @return {@link List}<{@link R}>
* @throws SQLException
*/
public static <R> List<R> listFromResultSet(ResultSet resultSet, Class<R> targetClass) throws SQLException {
if (resultSet == null) {
return Collections.emptyList();
}
List<R> 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<String, Object> 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> R mapToObj(Map<String, Object> valueMap, Class<R> 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
作者：南瓜大人
相关推荐
- WPS 隐藏黑科技!OCT2HEX 函数用法全攻略，数据转换不再愁
-
WPS隐藏黑科技！OCT2HEX函数用法全攻略，数据转换不再愁在WPS表格的强大函数库中，OCT2HEX函数堪称数据进制转换的“魔法钥匙”。无论是程序员处理代码数据，还是工程师进行电路设计...
- WPS 表格隐藏神器!LEFTB 函数让文本处理更高效
-
WPS表格隐藏神器！LEFTB函数让文本处理更高效在职场办公和日常数据处理中，WPS表格堪称我们的得力助手，而其中丰富多样的函数更是提升效率的关键。今天，要为大家介绍一个“宝藏函数”——LEF...
- Java lombok 使用教程（lombok.jar idea）
-
简介Lombok是...
- PART 48: 万能结果自定义，SWITCH函数!
-
公式解析SWITCH:根据值列表计算表达式并返回与第一个匹配值对应的结果。如果没有匹配项，则返回可选默认值用法解析1：评级=SWITCH(TRUE,C2>=90,"优秀",C2...
- Excel 必备if函数使用方法详解（excel表if函数使用）
-
excel表格if函数使用方法介绍打开Excel，在想输出数据的单元格点击工具栏上的“公式”--“插入函数”--“IF”,然后点击确定。...
- Jetty使用场景（jetty入门）
-
Jetty作为一款高性能、轻量级的嵌入式Web服务器和Servlet容器，其核心优势在于模块化设计、快速启动、低资源消耗...
- 【Java教程】基础语法到高级特性（java语言高级特性）
-
Java作为一门面向对象的编程语言，拥有清晰规范的语法体系。本文将系统性地介绍Java的核心语法特性，帮助开发者全面掌握Java编程基础。...
- WPS里这个EVEN 函数，90%的人都没用过!
-
一、开篇引入在日常工作中，我们常常会与各种数据打交道。比如，在统计员工绩效时，需要对绩效分数进行一系列处理；在计算销售数据时，可能要对销售额进行特定的运算。这些看似简单的数据处理任务，实则隐藏着许多技...
- 64 AI助力Excel，查函数查用法简单方便
-
在excel表格当中接入ai之后会是一种什么样的使用体验？今天就跟大家一起来分享一下小程序商店的下一步重大的版本更新。下一个版本将会加入ai功能，接下来会跟大家演示一下基础的用法。ai功能规划的是有三...
- python入门到脱坑 函数—函数的调用
-
Python函数调用详解函数调用是Python编程中最基础也是最重要的操作之一。下面我将详细介绍Python中函数调用的各种方式和注意事项。...
- 从简到繁，一文说清vlookup函数的常见用法
-
VLOOKUP函数是Excel中常用的查找与引用函数，用于在表格中按列查找数据。本文将从简单到复杂，逐步讲解VLOOKUP的用法、语法、应用场景及注意事项。一、VLOOKUP基础：快速入门1.什么是...
- Java新特性:Lambda表达式（java lambda表达式的3种简写方式）
-
1、Lambda表达式概述1.1、Lambda表达式的简介Lambda表达式（Lambdaexpression），也可称为闭包（Closure），是Java（SE）8中一个重要的新特性。Lam...
- WPS 冷门却超实用!ODD 函数用法大揭秘，轻松解决数据处理难题
-
WPS冷门却超实用！ODD函数用法大揭秘，轻松解决数据处理难题在WPS表格庞大的函数家族里，有一些函数虽然不像SUM、VLOOKUP那样广为人知，却在特定场景下能发挥出令人惊叹的作用，OD...
- Python 函数式编程的 8 大核心技巧，不允许你还不会
-
函数式编程是一种强调使用纯函数、避免共享状态和可变数据的编程范式。Python虽然不是纯函数式语言，但提供了丰富的函数式编程特性。以下是Python函数式编程的8个核心技巧：...
欢迎 你 发表评论:
- 一周热门
- 最近发表
- 标签列表
-
- idea eval reset (50)
- vue dispatch (70)
- update canceled (42)
- order by asc (53)
- spring gateway (67)
- 简单代码编程 贪吃蛇 (40)
- transforms.resize (33)
- redisson trylock (35)
- 卸载node (35)
- np.reshape (33)
- torch.arange (34)
- npm 源 (35)
- vue3 deep (35)
- win10 ssh (35)
- vue foreach (34)
- idea设置编码为utf8 (35)
- vue 数组添加元素 (34)
- std find (34)
- tablefield注解用途 (35)
- python str转json (34)
- java websocket客户端 (34)
- tensor.view (34)
- java jackson (34)
- vmware17pro最新密钥 (34)
- mysql单表最大数据量 (35)