diff options
author | Ala Luszczak <ala@databricks.com> | 2017-02-18 07:51:41 -0800 |
---|---|---|
committer | Reynold Xin <rxin@databricks.com> | 2017-02-18 07:51:41 -0800 |
commit | b486ffc86d8ad6c303321dcf8514afee723f61f8 (patch) | |
tree | 090b1eeb158c80cd51e6670d997351516bf22e15 /sql/catalyst | |
parent | 729ce3703257aa34c00c5c8253e6971faf6a0c8d (diff) | |
download | spark-b486ffc86d8ad6c303321dcf8514afee723f61f8.tar.gz spark-b486ffc86d8ad6c303321dcf8514afee723f61f8.tar.bz2 spark-b486ffc86d8ad6c303321dcf8514afee723f61f8.zip |
[SPARK-19447] Make Range operator generate "recordsRead" metric
## What changes were proposed in this pull request?
The Range was modified to produce "recordsRead" metric instead of "generated rows". The tests were updated and partially moved to SQLMetricsSuite.
## How was this patch tested?
Unit tests.
Author: Ala Luszczak <ala@databricks.com>
Closes #16960 from ala/range-records-read.
Diffstat (limited to 'sql/catalyst')
-rw-r--r-- | sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala index 87932e09a1..760ead42c7 100644 --- a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala +++ b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/CodeGenerator.scala @@ -31,6 +31,7 @@ import org.codehaus.janino.{ByteArrayClassLoader, ClassBodyEvaluator, SimpleComp import org.codehaus.janino.util.ClassFile import org.apache.spark.{SparkEnv, TaskContext, TaskKilledException} +import org.apache.spark.executor.InputMetrics import org.apache.spark.internal.Logging import org.apache.spark.metrics.source.CodegenMetrics import org.apache.spark.sql.catalyst.InternalRow @@ -933,7 +934,8 @@ object CodeGenerator extends Logging { classOf[UnsafeMapData].getName, classOf[Expression].getName, classOf[TaskContext].getName, - classOf[TaskKilledException].getName + classOf[TaskKilledException].getName, + classOf[InputMetrics].getName )) evaluator.setExtendedClass(classOf[GeneratedClass]) |