Josh Rosen created SPARK-19091:
----------------------------------
Summary: Implement more accurate statistics for LogicalRDD when
child is ParallelCollectionRDD
Key: SPARK-19091
URL: https://issues.apache.org/jira/browse/SPARK-19091
Project: Spark
Issue Type: Improvement
Components: SQL
Reporter: Josh Rosen
The Catalyst optimizer uses LogicalRDD to represent scans from existing RDDs.
In general, it's hard to produce size estimates for arbitrary RDDs, which is
why the current implementation simply estimates these relations size using the
default size (see the TODO at (for Datasets created via
spark.createDataFrame()))
In the special case where data has been parallelized with {{sc.parallelize()}}
we'll wind up with a ParallelCollectionRDD whose number of elements is known.
When we construct a LogicalRDD plan node in {{SparkSession.createDataFrame()}}
we'll probably be using an RDD which is a one-to-one transformation of a
parallel collection RDD (e.g. mapping an encoder to convert case classes to
internal rows). Thus we can have LogicalRDD's statistics method pattern-match
on cases where we have a MappedPartitionsRDD stacked immediately on top of a
ParallelCollectionRDD, then use walk up the RDD parent chain to determine the
number of elements and we can combine this with the schema and a conservative
"fudge factor" to produce an over-estimate of the LogicalRDD's size which will
be more accurate than the default size.
I believe that this will help us to avoid falling into pathologically bad plans
when joining tiny parallelize()d data sets against huge tables and have one of
my own production use-cases which would have benefitted directly from such an
optimization.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
