sharding-jdbc的源码分析一

路由的输入与执行

ShardingPreparedStatement的route()方法如下,routingEngine在构造时会传入需要路由的sql字符串以及对应的Connection关联的ShardingContext(一个javaBean),在执行路由的时候,route()方法会传入setParameter获得的参数,根据参数、参数值以及SQL进行路由分析产生执行单元PreparedStatementUnit列表,之后将PreparedStatementUnit列表传到谷歌的Excutor框架去对里面路由后的各个SQL通过PreparedStatement执行。
这里的getParameters()实际上获取的是父类的属性,在父类已经定义了,父类是AbstractShardingPreparedStatementAdapter抽象类,再通过几层关系实现了PreparedStatement的接口,ShardingPreparedStatement通过子类继承的方式封装并实现了PreparedStatement接口规范,实现了excuteQuery等方法,ShardingPreparedStatement的创建在ShardingConnection里进行创建

 private Collection<PreparedStatementUnit> route() throws SQLException {
        Collection<PreparedStatementUnit> result = new LinkedList<>();
        routeResult = routingEngine.route(getParameters());
        for (SQLExecutionUnit each : routeResult.getExecutionUnits()) {
            SQLType sqlType = routeResult.getSqlStatement().getType();
            Collection<PreparedStatement> preparedStatements;
            if (SQLType.DDL == sqlType) {
                preparedStatements = generatePreparedStatementForDDL(each);
            } else {
                preparedStatements = Collections.singletonList(generatePreparedStatement(each));
            }
            routedStatements.addAll(preparedStatements);
            for (PreparedStatement preparedStatement : preparedStatements) {
                replaySetParameter(preparedStatement);
                result.add(new PreparedStatementUnit(each, preparedStatement));
            }
        }
        return result;
    }
    public ShardingPreparedStatement(final ShardingConnection connection, final String sql, final int resultSetType, final int resultSetConcurrency, final int resultSetHoldability) {
        this.connection = connection;
        this.resultSetType = resultSetType;
        this.resultSetConcurrency = resultSetConcurrency;
        this.resultSetHoldability = resultSetHoldability;
        routingEngine = new PreparedStatementRoutingEngine(sql, connection.getShardingContext());
    }

SQL的解析到路由的解析到SQL的执行

从ShardingPreparedStatement的route()方法跟进下去,可以发现路由解析的过程是在PreparedStatementRoutingEngine里面,PreparedStatementRoutingEngine是上面说到的routingEngine的具体实现,在构造他的时候,他获取了sql字符串和Connection连接关联的ShardingContext(一个javaBean),在调用route()的时候获取了PreparedStatement的参数,因此路由解析的参数的三个输入就确定了下来;PreparedStatementRoutingEngine里面有三个属性以及一个方法,看到这里,基本上可以确定路由算法就是在SQLRouter接口的实现类实现的,从构造方法里面跟进去可以发现SQLRouter的实现类在2.0版本有两个,一个是DatabaseHintSQLRouter,一个是ParsingSQLRouter,他们都实现了SQLRouter的parse和route方法,parse解析的结果封装在SQLStatement接口,route的结果封装在SQLRouteResult(一个Bean)。DMLStatement继承AbstractSQLStatement从而实现SQLStatement接口,DMLStatement根据SQL的类型派生了例如InsertStatement、updateStatement等子类(alter、create、drop、insert、select、truncate、update)。
实际上这里的SQLStatement命名应该是参考了mysql-connector包里面的从下面的代码 route(final List<Object> parameters)我们也可以知道,sqlStatement封装的是SQL解析后的结果,sqlRouter利用了该结果去对SQL进行路由,封装成SQLRouteResult,最后SQL的执行就是利用了SQLRouteResult里面的信息,sqlRouter里面方法的执行顺序是先parse后route。

public final class PreparedStatementRoutingEngine {
    
    private final String logicSQL;
    
    private final SQLRouter sqlRouter;
    
    private SQLStatement sqlStatement;
    
    public PreparedStatementRoutingEngine(final String logicSQL, final ShardingContext shardingContext) {
        this.logicSQL = logicSQL;
        sqlRouter = SQLRouterFactory.createSQLRouter(shardingContext);
    }
    
    /**
     * SQL route.
     * 
     * <p>First routing time will parse SQL, after second time will reuse first parsed result.</p>
     * 
     * @param parameters parameters of SQL placeholder
     * @return route result
     */
    public SQLRouteResult route(final List<Object> parameters) {
        if (null == sqlStatement) {
            sqlStatement = sqlRouter.parse(logicSQL, parameters.size());
        }
        return sqlRouter.route(logicSQL, parameters, sqlStatement);
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 关于Mongodb的全面总结 MongoDB的内部构造《MongoDB The Definitive Guide》...
    中v中阅读 32,109评论 2 89
  • 首先去官网下载python3.5直接安装,安装完毕后自带pip可以在命令行中使用python查看版本,pip -V...
    晓函阅读 3,110评论 0 0
  • 第一份工作是在广告公司做策划专员。这一行,每一天都风起云涌,变幻莫测。折射在日常的工作中就成了临下班前布置的任务、...
    是闪闪发光的晓宇宙阅读 3,122评论 5 4
  • 她穿上新买的羽绒服 笑着 在原地转了一个圈 轻盈地 张开的手臂旋起来像花朵 我裹在 落了尘灰的旧大衣里 看着她 想...
    江山宜唤酒阅读 2,695评论 0 0