diff options
author | baishuo <vc_java@hotmail.com> | 2014-12-02 12:12:03 -0800 |
---|---|---|
committer | Michael Armbrust <michael@databricks.com> | 2014-12-02 12:38:51 -0800 |
commit | aa3d369a6bf77a00939da020d823ab90c9fe3cab (patch) | |
tree | a57c3f894390fcb80353a7f585affa6499d72db7 /sql/core | |
parent | 1850d90b9bbfb973e13c2c2334ba817e623de46b (diff) | |
download | spark-aa3d369a6bf77a00939da020d823ab90c9fe3cab.tar.gz spark-aa3d369a6bf77a00939da020d823ab90c9fe3cab.tar.bz2 spark-aa3d369a6bf77a00939da020d823ab90c9fe3cab.zip |
[SPARK-4663][sql]add finally to avoid resource leak
Author: baishuo <vc_java@hotmail.com>
Closes #3526 from baishuo/master-trycatch and squashes the following commits:
d446e14 [baishuo] correct the code style
b36bf96 [baishuo] correct the code style
ae0e447 [baishuo] add finally to avoid resource leak
(cherry picked from commit 69b6fed206565ecb0173d3757bcb5110422887c3)
Signed-off-by: Michael Armbrust <michael@databricks.com>
Diffstat (limited to 'sql/core')
-rw-r--r-- | sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableOperations.scala | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableOperations.scala b/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableOperations.scala index 0e36852ddd..232ef90b01 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableOperations.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/parquet/ParquetTableOperations.scala @@ -302,11 +302,14 @@ case class InsertIntoParquetTable( val committer = format.getOutputCommitter(hadoopContext) committer.setupTask(hadoopContext) val writer = format.getRecordWriter(hadoopContext) - while (iter.hasNext) { - val row = iter.next() - writer.write(null, row) + try { + while (iter.hasNext) { + val row = iter.next() + writer.write(null, row) + } + } finally { + writer.close(hadoopContext) } - writer.close(hadoopContext) committer.commitTask(hadoopContext) 1 } |