for #1172, fix ot for concurrency on get connection

apache · Sep 25, 2018 · afcfab4 · afcfab4
1 parent 6bcd301
commit afcfab4
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.../src/main/java/io/shardingsphere/core/executor/sql/prepare/SQLExecutePrepareTemplate.java b/.../src/main/java/io/shardingsphere/core/executor/sql/prepare/SQLExecutePrepareTemplate.java
@@ -23,6 +23,7 @@
 import io.shardingsphere.core.executor.ShardingExecuteEngine;
 import io.shardingsphere.core.executor.ShardingExecuteGroup;
 import io.shardingsphere.core.executor.StatementExecuteUnit;
+import io.shardingsphere.core.executor.sql.execute.threadlocal.ExecutorDataMap;
 import io.shardingsphere.core.routing.RouteUnit;
 import io.shardingsphere.core.routing.SQLUnit;
 import lombok.AllArgsConstructor;
@@ -77,11 +78,13 @@ private Collection<ShardingExecuteGroup<StatementExecuteUnit>> getSynchronizedEx
     private Collection<ShardingExecuteGroup<StatementExecuteUnit>> getAsynchronizedExecuteUnitGroups(
             final Collection<RouteUnit> routeUnits, final SQLExecutePrepareCallback callback) throws SQLException {
         Map<String, List<SQLUnit>> sqlUnitGroups = getSQLUnitGroups(routeUnits);
+        final Map<String, Object> dataMap = ExecutorDataMap.getDataMap();
         List<Collection<ShardingExecuteGroup<StatementExecuteUnit>>> results = shardingExecuteEngine.execute(sqlUnitGroups.entrySet(),
                 new ShardingExecuteCallback<Entry<String, List<SQLUnit>>, Collection<ShardingExecuteGroup<StatementExecuteUnit>>>() {
 
                     @Override
                     public Collection<ShardingExecuteGroup<StatementExecuteUnit>> execute(final Entry<String, List<SQLUnit>> input) throws SQLException {
+                        ExecutorDataMap.setDataMap(dataMap);
                         return getSQLExecuteGroups(input.getKey(), input.getValue(), callback);
                     }
                 });