diff --git a/infra/rewrite/src/main/java/org/apache/shardingsphere/infra/rewrite/context/SQLRewriteContext.java b/infra/rewrite/src/main/java/org/apache/shardingsphere/infra/rewrite/context/SQLRewriteContext.java index a415711d8cbaf..95e68fd476db1 100644 --- a/infra/rewrite/src/main/java/org/apache/shardingsphere/infra/rewrite/context/SQLRewriteContext.java +++ b/infra/rewrite/src/main/java/org/apache/shardingsphere/infra/rewrite/context/SQLRewriteContext.java @@ -69,26 +69,17 @@ public SQLRewriteContext(final ShardingSphereDatabase database, final QueryConte if (!queryContext.getHintValueContext().isSkipSQLRewrite()) { addSQLTokenGenerators(new DefaultTokenGeneratorBuilder(sqlStatementContext).getSQLTokenGenerators()); } - List genericParams = - (sqlStatementContext.getSqlStatement() instanceof PostgreSQLInsertStatement) - ? ((InsertStatementContext) sqlStatementContext).getOnConflictKeyUpdateParameters() - : ((InsertStatementContext) sqlStatementContext).getOnDuplicateKeyUpdateParameters(); - parameterBuilder = containsInsertValues(sqlStatementContext) - ? new GroupedParameterBuilder(((InsertStatementContext) sqlStatementContext).getGroupedParameters(), - genericParams) - : new StandardParameterBuilder(parameters); - - // if (sqlStatementContext.getSqlStatement() instanceof PostgreSQLInsertStatement) { - // parameterBuilder = containsInsertValues(sqlStatementContext) - // ? new GroupedParameterBuilder(((InsertStatementContext) sqlStatementContext).getGroupedParameters(), - // ((InsertStatementContext) sqlStatementContext).getOnConflictKeyUpdateParameters()) - // : new StandardParameterBuilder(parameters); - // } else { - // parameterBuilder = containsInsertValues(sqlStatementContext) - // ? new GroupedParameterBuilder(((InsertStatementContext) sqlStatementContext).getGroupedParameters(), - // ((InsertStatementContext) sqlStatementContext).getOnDuplicateKeyUpdateParameters()) - // : new StandardParameterBuilder(parameters); - // } + if (sqlStatementContext.getSqlStatement() instanceof PostgreSQLInsertStatement) { + parameterBuilder = containsInsertValues(sqlStatementContext) + ? new GroupedParameterBuilder(((InsertStatementContext) sqlStatementContext).getGroupedParameters(), + ((InsertStatementContext) sqlStatementContext).getOnConflictKeyUpdateParameters()) + : new StandardParameterBuilder(parameters); + } else{ + parameterBuilder = containsInsertValues(sqlStatementContext) + ? new GroupedParameterBuilder(((InsertStatementContext) sqlStatementContext).getGroupedParameters(), + ((InsertStatementContext) sqlStatementContext).getOnDuplicateKeyUpdateParameters()) + : new StandardParameterBuilder(parameters); + } } private boolean containsInsertValues(final SQLStatementContext sqlStatementContext) {