Commit 45a3e8bc authored by 马晓光's avatar 马晓光
Browse files

fix desiredPartitionSize calculate error

parent 180f0cce

sharding-core/src/main/java/io/shardingsphere/core/executor/sql/prepare/SQLExecutePrepareTemplate.java

+2 −1
Original line number Diff line number Diff line
@@ -40,6 +40,7 @@ import java.util.Map.Entry; 
 * @author zhaojun

 * @author zhangliang

 * @author panjuan

 * @author maxiaoguang

 */

@RequiredArgsConstructor

public final class SQLExecutePrepareTemplate {
@@ -82,7 +83,7 @@ public final class SQLExecutePrepareTemplate { 
    private List<ShardingExecuteGroup<StatementExecuteUnit>> getSQLExecuteGroups(

            final String dataSourceName, final List<SQLUnit> sqlUnits, final SQLExecutePrepareCallback callback) throws SQLException {

        List<ShardingExecuteGroup<StatementExecuteUnit>> result = new LinkedList<>();

        int desiredPartitionSize = Math.max(sqlUnits.size() / maxConnectionsSizePerQuery, 1);

        int desiredPartitionSize = Math.max(0 == sqlUnits.size() % maxConnectionsSizePerQuery ? sqlUnits.size() / maxConnectionsSizePerQuery : sqlUnits.size() / maxConnectionsSizePerQuery + 1, 1);

        List<List<SQLUnit>> sqlUnitPartitions = Lists.partition(sqlUnits, desiredPartitionSize);

        ConnectionMode connectionMode = maxConnectionsSizePerQuery < sqlUnits.size() ? ConnectionMode.CONNECTION_STRICTLY : ConnectionMode.MEMORY_STRICTLY;

        List<Connection> connections = callback.getConnections(connectionMode, dataSourceName, sqlUnitPartitions.size());