parthchandra commented on a change in pull request #24865: [SPARK-27100][SQL]
dag-scheduler-event-loop" java.lang.StackOverflowError
URL: https://github.com/apache/spark/pull/24865#discussion_r295925099
##########
File path:
sql/core/src/test/scala/org/apache/spark/sql/sources/BucketedReadSuite.scala
##########
@@ -735,4 +746,97 @@ abstract class BucketedReadSuite extends QueryTest with
SQLTestUtils {
df1.groupBy("j").agg(max("k")))
}
}
+
+ // A test with a partition where the number of files in the partition is
+ // large. tests for the condition where the serialization of such a task
may result in a stack
+ // overflow if the files list is stored in a recursive data structure
+ test("SPARK-27100 stack overflow: read data with large partitions") {
+ // Need a large number of files in the partition for the overflow
+ val numFilesInPartition = 100000
+
+ def getInputRDD(sparkSession: SparkSession, isBucketed: Boolean):
RDD[InternalRow] = {
+ val partitionValues = InternalRow.apply(Array("a"))
+
+ val schema = new StructType()
+ val fakeHadoopFsRelation = new HadoopFsRelation(null, schema, schema,
null,
+ new ParquetFileFormat, null)(sparkSession)
+ val optionalBucketSet = null
+ val bucketSpec = new BucketSpec(1, Seq("a"), Seq("b"))
+ val files = (0 to numFilesInPartition - 1).toStream.map { i =>
+ new FileStatus(10, false, 1, 512, 1000, new Path(s"file${i}_0.zzz"))
+ }
+ val partitionDirectory = PartitionDirectory(partitionValues, files);
+
+ val fileSource = FileSourceScanExec(fakeHadoopFsRelation,
+ null,
+ schema,
+ null,
+ Option(optionalBucketSet),
+ Seq.empty,
+ Option(new TableIdentifier("stackOverflow")))
+
+ val inputRDD = if (isBucketed) {
+ // Create a Bucketed RDD. This is a private method so we need to call
this indirectly.
+ val createBucketedReadRDD =
PrivateMethod[RDD[InternalRow]]('createBucketedReadRDD)
+
+ fileSource.invokePrivate(createBucketedReadRDD(bucketSpec,
Review comment:
Assuming you mean the line `val bucketSpec = BucketSpec(1, Seq("a"),
Seq("b"))`. Done
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
