diff --git a/project/SparkBuild.scala b/project/SparkBuild.scala index 3d89af2aa7b4a..46f6bf130f4f2 100644 --- a/project/SparkBuild.scala +++ b/project/SparkBuild.scala @@ -160,16 +160,21 @@ object SparkBuild extends PomBuild { val replacements = Map( """customId="println" level="error"""" -> """customId="println" level="warn"""" ) - var contents = Source.fromFile(in).getLines.mkString("\n") - for ((k, v) <- replacements) { - require(contents.contains(k), s"Could not rewrite '$k' in original scalastyle config.") - contents = contents.replace(k, v) - } - new PrintWriter(out) { - write(contents) - close() + val source = Source.fromFile(in) + try { + var contents = source.getLines.mkString("\n") + for ((k, v) <- replacements) { + require(contents.contains(k), s"Could not rewrite '$k' in original scalastyle config.") + contents = contents.replace(k, v) + } + new PrintWriter(out) { + write(contents) + close() + } + out + } finally { + source.close() } - out } // Return a cached scalastyle task for a given configuration (usually Compile or Test)