Minor refactor in InsertClauseShardingConditionEngine (apache#29304)

hunkier · Dec 6, 2023 · 885c439 · 885c439
1 parent c99d91e
commit 885c439
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...ingsphere/sharding/route/engine/condition/engine/InsertClauseShardingConditionEngine.java b/...ingsphere/sharding/route/engine/condition/engine/InsertClauseShardingConditionEngine.java
@@ -172,10 +172,10 @@ private void appendGeneratedKeyConditions(final InsertStatementContext sqlStatem
         Optional<GeneratedKeyContext> generatedKey = sqlStatementContext.getGeneratedKeyContext();
         String tableName = sqlStatementContext.getSqlStatement().getTable().getTableName().getIdentifier().getValue();
         if (generatedKey.isPresent() && generatedKey.get().isGenerated() && shardingRule.findTableRule(tableName).isPresent()) {
-            String schemaName =
-                    sqlStatementContext.getTablesContext().getSchemaName().orElseGet(() -> new DatabaseTypeRegistry(sqlStatementContext.getDatabaseType()).getDefaultSchemaName(database.getName()));
-            generatedKey.get().getGeneratedValues()
-                    .addAll(shardingRule.generateKeys(new KeyGenerateContext(database.getName(), schemaName, tableName, generatedKey.get().getColumnName()), sqlStatementContext.getValueListCount()));
+            String schemaName = sqlStatementContext.getTablesContext().getSchemaName()
+                    .orElseGet(() -> new DatabaseTypeRegistry(sqlStatementContext.getDatabaseType()).getDefaultSchemaName(database.getName()));
+            KeyGenerateContext keyGenerateContext = new KeyGenerateContext(database.getName(), schemaName, tableName, generatedKey.get().getColumnName());
+            generatedKey.get().getGeneratedValues().addAll(shardingRule.generateKeys(keyGenerateContext, sqlStatementContext.getValueListCount()));
             generatedKey.get().setSupportAutoIncrement(shardingRule.isSupportAutoIncrement(tableName));
             if (shardingRule.findShardingColumn(generatedKey.get().getColumnName(), tableName).isPresent()) {
                 appendGeneratedKeyCondition(generatedKey.get(), tableName, shardingConditions);