From 133495d82672c3f34d40a6298cc80c31f91faf5c Mon Sep 17 00:00:00 2001
From: Cheng Hao <hao.cheng@intel.com>
Date: Tue, 24 Jun 2014 19:07:02 -0700
Subject: [PATCH] [SQL]Add base row updating methods for JoinedRow
This will be helpful in join operators.
Author: Cheng Hao <hao.cheng@intel.com>
Closes #1187 from chenghao-intel/joinedRow and squashes the following commits:
87c19e3 [Cheng Hao] Add base row set methods for JoinedRow
---
.../sql/catalyst/expressions/Projection.scala | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Projection.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Projection.scala
index c9b7cea6a3..a9e976c3ab 100644
--- a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Projection.scala
+++ b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/Projection.scala
@@ -81,6 +81,18 @@ class JoinedRow extends Row {
this
}
+ /** Updates this JoinedRow by updating its left base row. Returns itself. */
+ def withLeft(newLeft: Row): Row = {
+ row1 = newLeft
+ this
+ }
+
+ /** Updates this JoinedRow by updating its right base row. Returns itself. */
+ def withRight(newRight: Row): Row = {
+ row2 = newRight
+ this
+ }
+
def iterator = row1.iterator ++ row2.iterator
def length = row1.length + row2.length
@@ -124,4 +136,9 @@ class JoinedRow extends Row {
}
new GenericRow(copiedValues)
}
+
+ override def toString() = {
+ val row = (if (row1 != null) row1 else Seq[Any]()) ++ (if (row2 != null) row2 else Seq[Any]())
+ s"[${row.mkString(",")}]"
+ }
}
--
GitLab