[SPARK-10612] [SQL] Add prepare to LocalNode.

The idea is that we should separate the function call that does memory reservation (i.e. prepare) from the function call that consumes the input (e.g. open()), so all operators can be a chance to reserve memory before they are all consumed.

Author: Reynold Xin <[email protected]>

Closes #8761 from rxin/SPARK-10612.

Author: rxin

sql/core/src/main/scala/org/apache/spark/sql/execution/local/LocalNode.scala

@@ -45,6 +45,14 @@ abstract class LocalNode(conf: SQLConf) extends TreeNode[LocalNode] with Logging
 
   def output: Seq[Attribute]
 
+  /**
+   * Called before open(). Prepare can be used to reserve memory needed. It must NOT consume
+   * any input data.
+   *
+   * Implementations of this must also call the `prepare()` function of its children.
+   */
+  def prepare(): Unit = children.foreach(_.prepare())
+
   /**
    * Initializes the iterator state. Must be called before calling `next()`.
    *