Examples with SQLExecutionUnit com.dangdang.ddframe.rdb.sharding.router.SQLExecutionUnit used on opensource projects

Example 1 with SQLExecutionUnit

use of com.dangdang.ddframe.rdb.sharding.router.SQLExecutionUnit in project sharding-jdbc by dangdangdotcom.

the class ShardingPreparedStatement method routeSQL.

private List<PreparedStatementExecutorWrapper> routeSQL() throws SQLException {
    List<PreparedStatementExecutorWrapper> result = new ArrayList<>();
    SQLRouteResult sqlRouteResult = preparedSQLRouter.route(getParameters());
    MergeContext mergeContext = sqlRouteResult.getMergeContext();
    setMergeContext(mergeContext);
    setGeneratedKeyContext(sqlRouteResult.getGeneratedKeyContext());
    for (SQLExecutionUnit each : sqlRouteResult.getExecutionUnits()) {
        PreparedStatement preparedStatement = (PreparedStatement) getStatement(getShardingConnection().getConnection(each.getDataSource(), sqlRouteResult.getSqlStatementType()), each.getSql());
        replayMethodsInvocation(preparedStatement);
        getParameters().replayMethodsInvocation(preparedStatement);
        result.add(wrap(preparedStatement, each));
    }
    return result;
}

Example 2 with SQLExecutionUnit

use of com.dangdang.ddframe.rdb.sharding.router.SQLExecutionUnit in project sharding-jdbc by dangdangdotcom.

the class ShardingStatement method generateExecutor.

private StatementExecutor generateExecutor(final String sql) throws SQLException {
    StatementExecutor result = new StatementExecutor(shardingConnection.getShardingContext().getExecutorEngine());
    SQLRouteResult sqlRouteResult = shardingConnection.getShardingContext().getSqlRouteEngine().route(sql);
    generatedKeyContext = sqlRouteResult.getGeneratedKeyContext();
    mergeContext = sqlRouteResult.getMergeContext();
    for (SQLExecutionUnit each : sqlRouteResult.getExecutionUnits()) {
        Statement statement = getStatement(shardingConnection.getConnection(each.getDataSource(), sqlRouteResult.getSqlStatementType()), each.getSql());
        replayMethodsInvocation(statement);
        result.addStatement(new StatementExecutorWrapper(statement, each));
    }
    return result;
}