Skip to content

Commit 8cfe79b

Browse files
committed
[SPARK-32524][SQL][TESTS] SharedSparkSession should clean up InMemoryRelation.ser
1 parent f3b10f5 commit 8cfe79b

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

sql/core/src/main/scala/org/apache/spark/sql/execution/columnar/InMemoryRelation.scala

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -277,6 +277,9 @@ object InMemoryRelation {
277277
ser.get
278278
}
279279

280+
/* Visible for testing */
281+
private[spark] def clearSerializer(): Unit = synchronized { ser = None }
282+
280283
def convertToColumnarIfPossible(plan: SparkPlan): SparkPlan = plan match {
281284
case gen: WholeStageCodegenExec => gen.child match {
282285
case c2r: ColumnarToRowTransition => c2r.child match {

sql/core/src/test/scala/org/apache/spark/sql/test/SharedSparkSession.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ import org.apache.spark.{DebugFilesystem, SparkConf}
2626
import org.apache.spark.internal.config.UNSAFE_EXCEPTION_ON_MEMORY_LEAK
2727
import org.apache.spark.sql.{SparkSession, SQLContext}
2828
import org.apache.spark.sql.catalyst.optimizer.ConvertToLocalRelation
29+
import org.apache.spark.sql.execution.columnar.InMemoryRelation.clearSerializer
2930
import org.apache.spark.sql.internal.{SQLConf, StaticSQLConf}
3031

3132
trait SharedSparkSession extends SQLTestUtils with SharedSparkSessionBase {
@@ -63,6 +64,7 @@ trait SharedSparkSessionBase
6364
with Eventually { self: Suite =>
6465

6566
protected def sparkConf = {
67+
clearSerializer()
6668
val conf = new SparkConf()
6769
.set("spark.hadoop.fs.file.impl", classOf[DebugFilesystem].getName)
6870
.set(UNSAFE_EXCEPTION_ON_MEMORY_LEAK, true)

0 commit comments

Comments
 (0)