16. sharding-jdbc源码分析之重写

最新推荐文章于 2024-08-22 15:45:00 发布

阿飞的博客

最新推荐文章于 2024-08-22 15:45:00 发布

阅读量722

点赞数

分类专栏：数据库 Java sharding-jdbc源码分析文章标签：数据库分库分表 sharding-jdbc

本文链接：https://blog.csdn.net/feelwing1314/article/details/80404010

版权

本文主要分析了sharding-jdbc的源码，着重探讨了表名、offset、rowCount的重写过程，以及SQLBuilder.toString()在生成最终SQL语句中的作用。通过SQLRewriteEngineTest的测试用例，展示了如何实现数据库分片后的SQL改写，包括对limit、offset和order by的处理，并解释了appendRest()方法的作用。

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

阿飞Javaer，转载请注明原创出处，谢谢！！

核心源码就在sharding-jdbc-core模块的com.dangdang.ddframe.rdb.sharding.rewrite目录下，包含两个文件SQLBuilder和SQLRewriteEngine；测试用例入口为SQLRewriteEngineTest，下面从SQLRewriteEngineTest中debug源码分析sharding-jdbc的重写是如何实现的：

SQLRewriteEngineTest中某个测试用例如下–主要包括表名，offset，limit（rowCount）的重写：

@Test
public void assertRewriteForLimit() {
    selectStatement.setLimit(new Limit(true));
    // offset的值就是limit offset,rowCount中offset的值
    selectStatement.getLimit().setOffset(new LimitValue(2, -1));
    // rowCount的值就是limit offset,rowCount中rowCount的值
    selectStatement.getLimit().setRowCount(new LimitValue(2, -1));
    // TableToken的值表示表名table_x在原始SQL语句的偏移量是17的位置
    selectStatement.getSqlTokens().add(new TableToken(17, "table_x"));
    // OffsetToken的值表示offset在原始SQL语句的偏移量是33的位置（2就是offset的值）
    selectStatement.getSqlTokens().add(new OffsetToken(33, 2));
    // RowCountToken的值表示rowCount在原始SQL语句的偏移量是36的位置（2就是rowCount的值）
    selectStatement.getSqlTokens().add(new RowCountToken(36, 2));
    // selectStatement值模拟过程，实际上是SQL解释过程（SQL解释会单独分析）
    SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, "SELECT x.id FROM table_x x LIMIT 2, 2", selectStatement);
    // 重写的核心就是这里了：rewriteEngine.rewrite(true)
    assertThat(rewriteEngine.rewrite(true).toSQL(tableTokens), is("SELECT x.id FROM table_1 x LIMIT