Set parent for MERGE insert and update clause

ZhengguanLi · Apr 11, 2024 · bb4e77b · bb4e77b
1 parent 34bd5f1
commit bb4e77b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLMergeStatement.java b/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLMergeStatement.java
@@ -90,6 +90,9 @@ public MergeUpdateClause getUpdateClause() {
     }
 
     public void setUpdateClause(MergeUpdateClause updateClause) {
+        if (updateClause != null) {
+            updateClause.setParent(this);
+        }
         this.updateClause = updateClause;
     }
 
@@ -98,6 +101,9 @@ public MergeInsertClause getInsertClause() {
     }
 
     public void setInsertClause(MergeInsertClause insertClause) {
+        if (insertClause != null) {
+            insertClause.setParent(this);
+        }
         this.insertClause = insertClause;
     }