Refactor DriverExecuteBatchExecutor (#31597)

apache · Jun 5, 2024 · feb2de8 · feb2de8
1 parent 156475b
commit feb2de8
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/jdbc/src/main/java/org/apache/shardingsphere/driver/executor/DriverExecuteBatchExecutor.java b/jdbc/src/main/java/org/apache/shardingsphere/driver/executor/DriverExecuteBatchExecutor.java
@@ -81,9 +81,8 @@ public DriverExecuteBatchExecutor(final ShardingSphereConnection connection, fin
      */
     public void addBatch(final QueryContext queryContext, final ShardingSphereDatabase database) {
         Optional<String> trafficInstanceId = connection.getTrafficInstanceId(metaData.getGlobalRuleMetaData().getSingleRule(TrafficRule.class), queryContext);
-        ExecutionContext result = trafficInstanceId.map(optional -> createExecutionContext(queryContext, optional)).orElseGet(() -> createExecutionContext(queryContext, database));
-        batchPreparedStatementExecutor.addBatchForExecutionUnits(result.getExecutionUnits());
-        executionContext = result;
+        executionContext = trafficInstanceId.map(optional -> createExecutionContext(queryContext, optional)).orElseGet(() -> createExecutionContext(queryContext, database));
+        batchPreparedStatementExecutor.addBatchForExecutionUnits(executionContext.getExecutionUnits());
     }
 
     private ExecutionContext createExecutionContext(final QueryContext queryContext, final String trafficInstanceId) {