ShardingJdbc2.X学习总结系列（六）：源码解析—SQL改写

最新推荐文章于 2024-06-18 09:04:35 发布

jadebai

最新推荐文章于 2024-06-18 09:04:35 发布

阅读量254

点赞数

分类专栏： ShardingJdbc 文章标签： ShardingJdbc java

本文链接：https://blog.csdn.net/jadebai/article/details/98470177

版权

本文是ShardingJdbc学习系列的第六篇，主要讲解SQL改写过程。从SQL改写的关键词入手，通过源码分析SQLRewriteEngine的生成，SQLBuilder的构建，以及如何生成最终执行的SQL。整个改写过程包括实例化SQLRewriteEngine，处理SQLToken，生成改写后的SQL，完成逻辑表到真实表的替换。

摘要由CSDN通过智能技术生成

前两篇我们介绍了SQL解析和SQL路由两个功能，这篇文章我们主要是介绍SQL改写。

1.首先还是介绍一些关键词语

1.SQL 改写：改写 SQL，解决分库分表后，查询结果需要聚合，需要对 SQL 进行调整，例如分页
SQLToken，SQL标记对象接口。SQLRewriteEngine 基于 SQLToken 实现 SQL改写。SQL解析器在 SQL解析过程中，很重要的一个目的是标记需要SQL改写的部分，也就是 SQLToken。
2.SQL 生成：生成分表分库的执行 SQL
SQL路由完后，会生成各数据分片的执行SQL。
调用RewriteEngine#generateSQL() 生成执行SQL。对于笛卡尔积路由结果和简单路由结果传递的参数略有不同：前者使用 CartesianDataSource ( CartesianTableReference )，后者使用路由表单元 ( TableUnit )。
对于两种情况，处理上大体是一致的：
1. 获得 SQL 相关逻辑表对应的真实表映射
2. 根据映射改写 SQL 相关逻辑表为真实表。

2.源码解析

public SQLRouteResult route(final String logicSQL, final List<Object> parameters, final SQLStatement sqlStatement) {
        SQLRouteResult result = new SQLRouteResult(sqlStatement);
        if (sqlStatement instanceof InsertStatement && null != ((InsertStatement) sqlStatement).getGeneratedKey()) {
            processGeneratedKey(parameters, (InsertStatement) sqlStatement, result);
        }
        RoutingResult routingResult = route(parameters, sqlStatement);
        ① SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, logicSQL, databaseType, sqlStatement);
        boolean isSingleRouting = routingResult.isSingleRouting();
        if (sqlStatement instanceof SelectStatement && null != ((SelectStatement) sqlStatement).getLimit()) {
            processLimit(parameters, (SelectStatement) sqlStatement, isSingleRouting);
        }
        SQLBuilder sqlBuilder = rewriteEngine.rewrite(!isSingleRouting);
        if (routingResult instanceof CartesianRoutingResult) {
            for (CartesianDataSource cartesianDataSource : ((CartesianRoutingResult) routingResult).getRoutingDataSources()) {
                for (CartesianTableReference cartesianTableReference : cartesianDataSource.getRoutingTableReferences()) {
                    result.getExecutionUnits().add(new SQLExecutionUnit(cartesianDataSource.getDataSource(), rewriteEngine.generateSQL(cartesianTableReference, sqlBuilder)));
                }
            }
        } else {
            for (TableUnit each : routingResult.getTableUnits().getTableUnits()) {
                result.getExecutionUnits().add(new SQLExecutionUnit(each.getDataSourceName(), rewriteEngine.generateSQL(each, sqlBuilder)));
            }
        } ②