Fix parent-child relationship

ZhengguanLi · Sep 5, 2024 · 783a54e · 783a54e
1 parent ab57847
commit 783a54e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLDeleteStatement.java b/core/src/main/java/com/alibaba/druid/sql/ast/statement/SQLDeleteStatement.java
@@ -209,6 +209,9 @@ public SQLTableSource getUsing() {
     }
 
     public void setUsing(SQLTableSource using) {
+        if (using != null) {
+            using.setParent(this);
+        }
         this.using = using;
     }
 

diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleDeleteStatement.java b/core/src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleDeleteStatement.java
@@ -38,6 +38,9 @@ public OracleReturningClause getReturning() {
     }
 
     public void setReturning(OracleReturningClause returning) {
+        if (returning != null) {
+            returning.setParent(this);
+        }
         this.returning = returning;
     }
 

diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleInsertStatement.java b/core/src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/stmt/OracleInsertStatement.java
@@ -64,6 +64,9 @@ public OracleReturningClause getReturning() {
     }
 
     public void setReturning(OracleReturningClause returning) {
+        if (returning != null) {
+            returning.setParent(this);
+        }
         this.returning = returning;
     }
 
@@ -72,6 +75,9 @@ public SQLErrorLoggingClause getErrorLogging() {
     }
 
     public void setErrorLogging(SQLErrorLoggingClause errorLogging) {
+        if (errorLogging != null) {
+            errorLogging.setParent(this);
+        }
         this.errorLogging = errorLogging;
     }