aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/FileFormatWriter.scala14
1 files changed, 10 insertions, 4 deletions
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/FileFormatWriter.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/FileFormatWriter.scala
index bda64d4b91..4ec09bff42 100644
--- a/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/FileFormatWriter.scala
+++ b/sql/core/src/main/scala/org/apache/spark/sql/execution/datasources/FileFormatWriter.scala
@@ -324,8 +324,11 @@ object FileFormatWriter extends Logging {
override def releaseResources(): Unit = {
if (currentWriter != null) {
- currentWriter.close()
- currentWriter = null
+ try {
+ currentWriter.close()
+ } finally {
+ currentWriter = null
+ }
}
}
}
@@ -459,8 +462,11 @@ object FileFormatWriter extends Logging {
override def releaseResources(): Unit = {
if (currentWriter != null) {
- currentWriter.close()
- currentWriter = null
+ try {
+ currentWriter.close()
+ } finally {
+ currentWriter = null
+ }
}
}
}