Skip to content

Commit aef004d

Browse files
beikovbeikov
committed
HHH-18563 Add set clause foreign key target tables to affected tables
1 parent 2a1e6d5 commit aef004d

File tree

12 files changed

+348
-70
lines changed

12 files changed

+348
-70
lines changed

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/OracleLegacySqlAstTranslator.java‎

Lines changed: 48 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import org.hibernate.query.common.FetchClauseType;
2323
import org.hibernate.query.common.FrameExclusion;
2424
import org.hibernate.query.common.FrameKind;
25+
import org.hibernate.query.sqm.sql.internal.SqmPathInterpretation;
2526
import org.hibernate.sql.ast.Clause;
2627
import org.hibernate.sql.ast.SqlAstNodeRenderingMode;
2728
import org.hibernate.sql.ast.spi.AbstractSqlAstTranslator;
@@ -55,7 +56,9 @@
5556
import org.hibernate.sql.ast.tree.select.QueryPart;
5657
import org.hibernate.sql.ast.tree.select.QuerySpec;
5758
import org.hibernate.sql.ast.tree.select.SelectClause;
59+
import org.hibernate.sql.ast.tree.select.SelectStatement;
5860
import org.hibernate.sql.ast.tree.select.SortSpecification;
61+
import org.hibernate.sql.ast.tree.update.Assignable;
5962
import org.hibernate.sql.ast.tree.update.Assignment;
6063
import org.hibernate.sql.ast.tree.update.UpdateStatement;
6164
import org.hibernate.sql.exec.spi.JdbcOperation;
@@ -668,13 +671,43 @@ private boolean supportsOffsetFetchClause() {
668671
return getDialect().supportsFetchClause( FetchClauseType.ROWS_ONLY );
669672
}
670673

674+
@Override
675+
protected void renderNull(Literal literal) {
676+
if ( getParameterRenderingMode() == SqlAstNodeRenderingMode.NO_UNTYPED ) {
677+
switch ( literal.getJdbcMapping().getJdbcType().getDdlTypeCode() ) {
678+
case SqlTypes.BLOB:
679+
appendSql( "to_blob(null)" );
680+
break;
681+
case SqlTypes.CLOB:
682+
appendSql( "to_clob(null)" );
683+
break;
684+
case SqlTypes.NCLOB:
685+
appendSql( "to_nclob(null)" );
686+
break;
687+
default:
688+
super.renderNull( literal );
689+
break;
690+
}
691+
}
692+
else {
693+
super.renderNull( literal );
694+
}
695+
}
696+
671697
@Override
672698
protected void visitSetAssignment(Assignment assignment) {
699+
final Assignable assignable = assignment.getAssignable();
700+
if ( assignable instanceof SqmPathInterpretation<?> ) {
701+
final String affectedTableName = ( (SqmPathInterpretation<?>) assignable ).getAffectedTableName();
702+
if ( affectedTableName != null ) {
703+
addAffectedTableName( affectedTableName );
704+
}
705+
}
673706
final List<ColumnReference> columnReferences = assignment.getAssignable().getColumnReferences();
707+
final Expression assignedValue = assignment.getAssignedValue();
674708
if ( columnReferences.size() == 1 ) {
675709
columnReferences.get( 0 ).appendColumnForWrite( this );
676710
appendSql( '=' );
677-
final Expression assignedValue = assignment.getAssignedValue();
678711
final SqlTuple sqlTuple = SqlTupleContainer.getSqlTuple( assignedValue );
679712
if ( sqlTuple != null ) {
680713
assert sqlTuple.getExpressions().size() == 1;
@@ -684,7 +717,7 @@ protected void visitSetAssignment(Assignment assignment) {
684717
assignedValue.accept( this );
685718
}
686719
}
687-
else {
720+
else if ( assignedValue instanceof SelectStatement ) {
688721
char separator = OPEN_PARENTHESIS;
689722
for ( ColumnReference columnReference : columnReferences ) {
690723
appendSql( separator );
@@ -694,5 +727,18 @@ protected void visitSetAssignment(Assignment assignment) {
694727
appendSql( ")=" );
695728
assignment.getAssignedValue().accept( this );
696729
}
730+
else {
731+
assert assignedValue instanceof SqlTupleContainer;
732+
final List<? extends Expression> expressions = ( (SqlTupleContainer) assignedValue ).getSqlTuple().getExpressions();
733+
columnReferences.get( 0 ).appendColumnForWrite( this, null );
734+
appendSql( '=' );
735+
expressions.get( 0 ).accept( this );
736+
for ( int i = 1; i < columnReferences.size(); i++ ) {
737+
appendSql( ',' );
738+
columnReferences.get( i ).appendColumnForWrite( this, null );
739+
appendSql( '=' );
740+
expressions.get( i ).accept( this );
741+
}
742+
}
697743
}
698744
}

‎hibernate-core/src/main/java/org/hibernate/dialect/OracleSqlAstTranslator.java‎

Lines changed: 49 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import org.hibernate.query.common.FetchClauseType;
2121
import org.hibernate.query.common.FrameExclusion;
2222
import org.hibernate.query.common.FrameKind;
23+
import org.hibernate.query.sqm.sql.internal.SqmPathInterpretation;
2324
import org.hibernate.sql.ast.Clause;
2425
import org.hibernate.sql.ast.SqlAstNodeRenderingMode;
2526
import org.hibernate.sql.ast.spi.SqlSelection;
@@ -53,7 +54,9 @@
5354
import org.hibernate.sql.ast.tree.select.QueryPart;
5455
import org.hibernate.sql.ast.tree.select.QuerySpec;
5556
import org.hibernate.sql.ast.tree.select.SelectClause;
57+
import org.hibernate.sql.ast.tree.select.SelectStatement;
5658
import org.hibernate.sql.ast.tree.select.SortSpecification;
59+
import org.hibernate.sql.ast.tree.update.Assignable;
5760
import org.hibernate.sql.ast.tree.update.Assignment;
5861
import org.hibernate.sql.ast.tree.update.UpdateStatement;
5962
import org.hibernate.sql.exec.spi.JdbcOperation;
@@ -625,13 +628,43 @@ private boolean supportsOffsetFetchClause() {
625628
return getDialect().supportsFetchClause( FetchClauseType.ROWS_ONLY );
626629
}
627630

631+
@Override
632+
protected void renderNull(Literal literal) {
633+
if ( getParameterRenderingMode() == SqlAstNodeRenderingMode.NO_UNTYPED ) {
634+
switch ( literal.getJdbcMapping().getJdbcType().getDdlTypeCode() ) {
635+
case SqlTypes.BLOB:
636+
appendSql( "to_blob(null)" );
637+
break;
638+
case SqlTypes.CLOB:
639+
appendSql( "to_clob(null)" );
640+
break;
641+
case SqlTypes.NCLOB:
642+
appendSql( "to_nclob(null)" );
643+
break;
644+
default:
645+
super.renderNull( literal );
646+
break;
647+
}
648+
}
649+
else {
650+
super.renderNull( literal );
651+
}
652+
}
653+
628654
@Override
629655
protected void visitSetAssignment(Assignment assignment) {
656+
final Assignable assignable = assignment.getAssignable();
657+
if ( assignable instanceof SqmPathInterpretation<?> ) {
658+
final String affectedTableName = ( (SqmPathInterpretation<?>) assignable ).getAffectedTableName();
659+
if ( affectedTableName != null ) {
660+
addAffectedTableName( affectedTableName );
661+
}
662+
}
630663
final List<ColumnReference> columnReferences = assignment.getAssignable().getColumnReferences();
664+
final Expression assignedValue = assignment.getAssignedValue();
631665
if ( columnReferences.size() == 1 ) {
632666
columnReferences.get( 0 ).appendColumnForWrite( this );
633667
appendSql( '=' );
634-
final Expression assignedValue = assignment.getAssignedValue();
635668
final SqlTuple sqlTuple = SqlTupleContainer.getSqlTuple( assignedValue );
636669
if ( sqlTuple != null ) {
637670
assert sqlTuple.getExpressions().size() == 1;
@@ -641,15 +674,28 @@ protected void visitSetAssignment(Assignment assignment) {
641674
assignedValue.accept( this );
642675
}
643676
}
644-
else {
677+
else if ( assignedValue instanceof SelectStatement ) {
645678
char separator = OPEN_PARENTHESIS;
646679
for ( ColumnReference columnReference : columnReferences ) {
647680
appendSql( separator );
648681
columnReference.appendColumnForWrite( this );
649682
separator = COMMA_SEPARATOR_CHAR;
650683
}
651684
appendSql( ")=" );
652-
assignment.getAssignedValue().accept( this );
685+
assignedValue.accept( this );
686+
}
687+
else {
688+
assert assignedValue instanceof SqlTupleContainer;
689+
final List<? extends Expression> expressions = ( (SqlTupleContainer) assignedValue ).getSqlTuple().getExpressions();
690+
columnReferences.get( 0 ).appendColumnForWrite( this, null );
691+
appendSql( '=' );
692+
expressions.get( 0 ).accept( this );
693+
for ( int i = 1; i < columnReferences.size(); i++ ) {
694+
appendSql( ',' );
695+
columnReferences.get( i ).appendColumnForWrite( this, null );
696+
appendSql( '=' );
697+
expressions.get( i ).accept( this );
698+
}
653699
}
654700
}
655701

‎hibernate-core/src/main/java/org/hibernate/query/sqm/mutation/internal/cte/CteInsertHandler.java‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1299,7 +1299,7 @@ private List<Assignment> getCompatibleAssignments(InsertSelectStatement dmlState
12991299
final List<Assignment> assignments = conflictClause.getAssignments();
13001300
for ( Assignment assignment : assignments ) {
13011301
for ( ColumnReference targetColumn : dmlStatement.getTargetColumns() ) {
1302-
if ( targetColumn.equals( assignment.getAssignable() ) ) {
1302+
if ( assignment.getAssignable().getColumnReferences().contains( targetColumn ) ) {
13031303
if ( compatibleAssignments == null ) {
13041304
compatibleAssignments = new ArrayList<>( assignments.size() );
13051305
}

‎hibernate-core/src/main/java/org/hibernate/query/sqm/sql/BaseSqmToSqlAstConverter.java‎

Lines changed: 53 additions & 46 deletions
Original file line numberDiff line numberDiff line change
@@ -858,36 +858,57 @@ public List<Assignment> visitSetClause(SqmSetClause setClause) {
858858
final SqmExpression<?> assignmentValue = sqmAssignment.getValue();
859859
final SqmParameter<?> assignmentValueParameter = getSqmParameter( assignmentValue );
860860
final Expression pathSqlExpression = assignedPathInterpretation.getSqlExpression();
861-
final List<? extends Expression> targetColumnReferences =
862-
pathSqlExpression instanceof SqlTuple sqlTuple
863-
? sqlTuple.getExpressions()
861+
//noinspection unchecked
862+
final List<ColumnReference> targetColumnReferences =
863+
pathSqlExpression instanceof SqlTupleContainer sqlTuple
864+
? (List<ColumnReference>) sqlTuple.getSqlTuple().getExpressions()
864865
: pathSqlExpression.getColumnReference().getColumnReferences();
865866
if ( assignmentValueParameter != null ) {
867+
final ArrayList<Expression> expressions = new ArrayList<>( targetColumnReferences.size() );
866868
consumeSqmParameter(
867869
assignmentValueParameter,
868870
assignedPathInterpretation.getExpressionType(),
869-
(index, jdbcParameter) -> addAssignment(
870-
assignments,
871-
aggregateColumnAssignmentHandler,
872-
targetColumnReferences.get( index ),
873-
jdbcParameter
874-
)
871+
(index, jdbcParameter) -> expressions.add( jdbcParameter )
875872
);
876-
}
877-
else if ( assignmentValue instanceof SqmLiteralNull<?> ) {
878-
for ( Expression columnReference : targetColumnReferences ) {
873+
if ( pathSqlExpression instanceof SqlTupleContainer ) {
879874
addAssignment(
880875
assignments,
881876
aggregateColumnAssignmentHandler,
882-
columnReference,
883-
new QueryLiteral<>( null, (BasicValuedMapping) columnReference.getExpressionType() )
877+
(Assignable) assignedPathInterpretation,
878+
targetColumnReferences,
879+
new SqlTuple( expressions, assignedPathInterpretation.getExpressionType() )
884880
);
885881
}
882+
else {
883+
assert expressions.size() == 1;
884+
addAssignment(
885+
assignments,
886+
aggregateColumnAssignmentHandler,
887+
(Assignable) assignedPathInterpretation,
888+
targetColumnReferences,
889+
expressions.get( 0 )
890+
);
891+
}
892+
}
893+
else if ( pathSqlExpression instanceof SqlTupleContainer
894+
&& assignmentValue instanceof SqmLiteralNull<?> ) {
895+
final ArrayList<Expression> expressions = new ArrayList<>( targetColumnReferences.size() );
896+
for ( ColumnReference targetColumnReference : targetColumnReferences ) {
897+
expressions.add( new QueryLiteral<>( null,
898+
(SqlExpressible) targetColumnReference.getExpressionType() ) );
899+
}
900+
addAssignment(
901+
assignments,
902+
aggregateColumnAssignmentHandler,
903+
(Assignable) assignedPathInterpretation,
904+
targetColumnReferences,
905+
new SqlTuple( expressions, assignedPathInterpretation.getExpressionType() )
906+
);
886907
}
887908
else {
888909
addAssignments(
889910
(Expression) assignmentValue.accept( this ),
890-
assignedPathInterpretation.getExpressionType(),
911+
assignedPathInterpretation,
891912
targetColumnReferences,
892913
assignments,
893914
aggregateColumnAssignmentHandler
@@ -909,35 +930,18 @@ else if ( assignmentValue instanceof SqmLiteralNull<?> ) {
909930

910931
private void addAssignments(
911932
Expression valueExpression,
912-
ModelPart assignedPathType,
913-
List<? extends Expression> targetColumnReferences,
933+
SqmPathInterpretation<?> assignedPathInterpretation,
934+
List<ColumnReference> targetColumnReferences,
914935
ArrayList<Assignment> assignments,
915-
AggregateColumnAssignmentHandler assignmentHandler) {
916-
checkAssignment( valueExpression, assignedPathType );
917-
if ( valueExpression instanceof SqlTuple sqlTuple ) {
918-
addTupleAssignments( targetColumnReferences, assignments, assignmentHandler, sqlTuple );
919-
}
920-
else if ( valueExpression instanceof EmbeddableValuedPathInterpretation<?> embeddable ) {
921-
addTupleAssignments( targetColumnReferences, assignments, assignmentHandler, embeddable.getSqlTuple() );
922-
}
923-
else {
924-
for ( Expression columnReference : targetColumnReferences ) {
925-
addAssignment( assignments, assignmentHandler, columnReference, valueExpression );
926-
}
927-
}
928-
}
929-
930-
private void addTupleAssignments(
931-
List<? extends Expression> targetColumnReferences,
932-
ArrayList<Assignment> assignments,
933-
AggregateColumnAssignmentHandler aggregateColumnAssignmentHandler,
934-
SqlTuple sqlTuple) {
935-
final List<? extends Expression> expressions = sqlTuple.getExpressions();
936-
assert targetColumnReferences.size() == expressions.size();
937-
for ( int i = 0; i < targetColumnReferences.size(); i++ ) {
938-
final ColumnReference columnReference = (ColumnReference) targetColumnReferences.get( i );
939-
addAssignment( assignments, aggregateColumnAssignmentHandler, columnReference, expressions.get( i ) );
940-
}
936+
AggregateColumnAssignmentHandler aggregateColumnAssignmentHandler) {
937+
checkAssignment( valueExpression, assignedPathInterpretation.getExpressionType() );
938+
addAssignment(
939+
assignments,
940+
aggregateColumnAssignmentHandler,
941+
(Assignable) assignedPathInterpretation,
942+
targetColumnReferences,
943+
valueExpression
944+
);
941945
}
942946

943947
private void checkAssignment(Expression valueExpression, ModelPart assignedPathType) {
@@ -958,12 +962,15 @@ private void checkAssignment(Expression valueExpression, ModelPart assignedPathT
958962
private void addAssignment(
959963
List<Assignment> assignments,
960964
AggregateColumnAssignmentHandler aggregateColumnAssignmentHandler,
961-
Expression columnReference,
965+
Assignable assignable,
966+
List<ColumnReference> targetColumnReferences,
962967
Expression valueExpression) {
963968
if ( aggregateColumnAssignmentHandler != null ) {
964-
aggregateColumnAssignmentHandler.addAssignment( assignments.size(), (ColumnReference) columnReference );
969+
for ( ColumnReference targetColumnReference : targetColumnReferences ) {
970+
aggregateColumnAssignmentHandler.addAssignment( assignments.size(), targetColumnReference );
971+
}
965972
}
966-
assignments.add( new Assignment( (ColumnReference) columnReference, valueExpression ) );
973+
assignments.add( new Assignment( assignable, valueExpression ) );
967974
}
968975

969976
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

‎hibernate-core/src/main/java/org/hibernate/query/sqm/sql/internal/BasicValuedPathInterpretation.java‎

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010

1111
import org.hibernate.metamodel.MappingMetamodel;
1212
import org.hibernate.metamodel.mapping.BasicValuedModelPart;
13+
import org.hibernate.metamodel.mapping.EntityAssociationMapping;
1314
import org.hibernate.metamodel.mapping.EntityMappingType;
1415
import org.hibernate.metamodel.mapping.MappingType;
1516
import org.hibernate.metamodel.mapping.ModelPart;
@@ -31,6 +32,8 @@
3132
import org.hibernate.sql.ast.tree.from.TableReference;
3233
import org.hibernate.sql.ast.tree.update.Assignable;
3334

35+
import org.checkerframework.checker.nullness.qual.Nullable;
36+
3437
import static jakarta.persistence.metamodel.Type.PersistenceType.ENTITY;
3538
import static org.hibernate.internal.util.NullnessUtil.castNonNull;
3639
import static org.hibernate.query.sqm.internal.SqmUtil.getTargetMappingIfNeeded;
@@ -124,15 +127,37 @@ else if ( expression instanceof SqlSelectionExpression ) {
124127
}
125128

126129
private final ColumnReference columnReference;
130+
private final @Nullable String affectedTableName;
127131

128132
public BasicValuedPathInterpretation(
129133
ColumnReference columnReference,
130134
NavigablePath navigablePath,
131135
BasicValuedModelPart mapping,
132136
TableGroup tableGroup) {
137+
this( columnReference, navigablePath, mapping, tableGroup, determineAffectedTableName( tableGroup, mapping ) );
138+
}
139+
140+
private static @Nullable String determineAffectedTableName(TableGroup tableGroup, BasicValuedModelPart mapping) {
141+
final ModelPartContainer modelPart = tableGroup.getModelPart();
142+
if ( modelPart instanceof EntityAssociationMapping ) {
143+
final EntityAssociationMapping associationMapping = (EntityAssociationMapping) modelPart;
144+
if ( !associationMapping.containsTableReference( mapping.getContainingTableExpression() ) ) {
145+
return associationMapping.getAssociatedEntityMappingType().getMappedTableDetails().getTableName();
146+
}
147+
}
148+
return null;
149+
}
150+
151+
public BasicValuedPathInterpretation(
152+
ColumnReference columnReference,
153+
NavigablePath navigablePath,
154+
BasicValuedModelPart mapping,
155+
TableGroup tableGroup,
156+
@Nullable String affectedTableName) {
133157
super( navigablePath, mapping, tableGroup );
134158
assert columnReference != null;
135159
this.columnReference = columnReference;
160+
this.affectedTableName = affectedTableName;
136161
}
137162

138163
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -143,6 +168,11 @@ public Expression getSqlExpression() {
143168
return columnReference;
144169
}
145170

171+
@Override
172+
public @Nullable String getAffectedTableName() {
173+
return affectedTableName;
174+
}
175+
146176
@Override
147177
public ColumnReference getColumnReference() {
148178
return columnReference;

0 commit comments

Comments
 (0)