shouldStop is called outside consume().

apache · Aug 13, 2017 · e0e7a6e · e0e7a6e
1 parent 05274e7
commit e0e7a6e
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/WholeStageCodegenExec.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/WholeStageCodegenExec.scala
@@ -183,16 +183,7 @@ trait CodegenSupport extends SparkPlan {
          | }
        """.stripMargin)
 
-    if (isShouldStopRequired) {
-      // Because the processing logic is enclosed in a function, `shouldStop` call in the function
-      // don't be affect outside loop, we need to check it and stop the loop.
-      s"""
-         | $doConsumeFuncName($callingParams);
-         | if (shouldStop()) return;
-       """.stripMargin
-    } else {
-      s"$doConsumeFuncName($callingParams);"
-    }
+    s"$doConsumeFuncName($callingParams);"
   }
 
   /**