Revert "[SPARK-13363][SQL] support Aggregator in RelationalGroupedDataset"
This reverts commit 12854464c4
.
This commit is contained in:
parent
12854464c4
commit
527c780bb0
|
@ -208,11 +208,7 @@ class RelationalGroupedDataset protected[sql](
|
|||
*/
|
||||
@scala.annotation.varargs
|
||||
def agg(expr: Column, exprs: Column*): DataFrame = {
|
||||
toDF((expr +: exprs).map {
|
||||
case typed: TypedColumn[_, _] =>
|
||||
typed.withInputType(df.resolvedTEncoder, df.logicalPlan.output).expr
|
||||
case c => c.expr
|
||||
})
|
||||
toDF((expr +: exprs).map(_.expr))
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -19,6 +19,7 @@ package org.apache.spark.sql
|
|||
|
||||
import scala.language.postfixOps
|
||||
|
||||
import org.apache.spark.sql.catalyst.encoders.ExpressionEncoder
|
||||
import org.apache.spark.sql.expressions.Aggregator
|
||||
import org.apache.spark.sql.expressions.scala.typed
|
||||
import org.apache.spark.sql.functions._
|
||||
|
@ -84,15 +85,6 @@ class ParameterizedTypeSum[IN, OUT : Numeric : Encoder](f: IN => OUT)
|
|||
override def outputEncoder: Encoder[OUT] = implicitly[Encoder[OUT]]
|
||||
}
|
||||
|
||||
object RowAgg extends Aggregator[Row, Int, Int] {
|
||||
def zero: Int = 0
|
||||
def reduce(b: Int, a: Row): Int = a.getInt(0) + b
|
||||
def merge(b1: Int, b2: Int): Int = b1 + b2
|
||||
def finish(r: Int): Int = r
|
||||
override def bufferEncoder: Encoder[Int] = Encoders.scalaInt
|
||||
override def outputEncoder: Encoder[Int] = Encoders.scalaInt
|
||||
}
|
||||
|
||||
|
||||
class DatasetAggregatorSuite extends QueryTest with SharedSQLContext {
|
||||
|
||||
|
@ -208,8 +200,4 @@ class DatasetAggregatorSuite extends QueryTest with SharedSQLContext {
|
|||
(1279869254, "Some String"))
|
||||
}
|
||||
|
||||
test("aggregator in DataFrame/Dataset[Row]") {
|
||||
val df = Seq(1 -> "a", 2 -> "b", 3 -> "b").toDF("i", "j")
|
||||
checkAnswer(df.groupBy($"j").agg(RowAgg.toColumn), Row("a", 1) :: Row("b", 5) :: Nil)
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue