[nereids] pull up join from union all rule

apache · Dec 20, 2023 · 93fb58a · 93fb58a
1 parent 39817c6
commit 93fb58a
Showing 1 changed file with 20 additions and 10 deletions.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/nereids/rules/rewrite/PullUpJoinFromUnionAll.java b/fe/fe-core/src/main/java/org/apache/doris/nereids/rules/rewrite/PullUpJoinFromUnionAll.java
@@ -344,23 +344,33 @@ private boolean checkJoinConditionOnPk(LogicalJoin joinRoot, LogicalCatalogRelat
     }
 
     private Set<String> getPkInfoFromConstraint(LogicalCatalogRelation table) {
-        for (Map.Entry<String, Constraint> constraintMap : table.getTable().getConstraintsMap().entrySet()) {
-            Constraint constraint = constraintMap.getValue();
-            if (constraint instanceof PrimaryKeyConstraint) {
-                return ((PrimaryKeyConstraint) constraint).getPrimaryKeyNames();
+        table.getTable().readLock();
+        try {
+            for (Map.Entry<String, Constraint> constraintMap : table.getTable().getConstraintsMap().entrySet()) {
+                Constraint constraint = constraintMap.getValue();
+                if (constraint instanceof PrimaryKeyConstraint) {
+                    return ((PrimaryKeyConstraint) constraint).getPrimaryKeyNames();
+                }
             }
+            return null;
+        } finally {
+            table.getTable().readUnlock();
         }
-        return null;
     }
 
     private Set<String> getUkInfoFromConstraint(LogicalCatalogRelation table) {
-        for (Map.Entry<String, Constraint> constraintMap : table.getTable().getConstraintsMap().entrySet()) {
-            Constraint constraint = constraintMap.getValue();
-            if (constraint instanceof UniqueConstraint) {
-                return ((UniqueConstraint) constraint).getUniqueColumnNames();
+        table.getTable().readLock();
+        try {
+            for (Map.Entry<String, Constraint> constraintMap : table.getTable().getConstraintsMap().entrySet()) {
+                Constraint constraint = constraintMap.getValue();
+                if (constraint instanceof UniqueConstraint) {
+                    return ((UniqueConstraint) constraint).getUniqueColumnNames();
+                }
             }
+            return null;
+        } finally {
+            table.getTable().readUnlock();
         }
-        return null;
     }
 
     private boolean checkJoinRoot(LogicalJoin joinRoot) {