explicitly close streams on java.lang.Process to avoid descriptor leaks

This commit is contained in:
Mark Harrah 2012-01-14 21:09:11 -05:00
parent a9ccd74eb8
commit e23abdfce3
1 changed files with 18 additions and 10 deletions

View File

@ -55,6 +55,11 @@ object BasicIO
final val BufferSize = 8192
final val Newline = System.getProperty("line.separator")
def closeProcessStreams(p: JProcess): Unit =
Seq(p.getOutputStream, p.getInputStream, p.getErrorStream) foreach { s =>
if(s ne null) close(s)
}
def close(c: java.io.Closeable) = try { c.close() } catch { case _: java.io.IOException => () }
def processFully(buffer: Appendable): InputStream => Unit = processFully(appendLine(buffer))
def processFully(processLine: String => Unit): InputStream => Unit =
@ -364,8 +369,6 @@ private[sbt] class DummyProcessBuilder(override val toString: String, exitValue
override def run(io: ProcessIO): Process = new DummyProcess(exitValue)
override def canPipeTo = true
}
/** A thin wrapper around a java.lang.Process. `ioThreads` are the Threads created to do I/O.
* The implementation of `exitValue` waits until these threads die before returning. */
private class DummyProcess(action: => Int) extends Process
{
private[this] val exitCode = Future(action)
@ -399,18 +402,23 @@ private[sbt] class SimpleProcessBuilder(p: JProcessBuilder) extends AbstractProc
* returning. */
private class SimpleProcess(p: JProcess, inputThread: Thread, outputThreads: List[Thread]) extends Process
{
override def exitValue() =
override def exitValue(): Int =
{
try { p.waitFor() }// wait for the process to terminate
finally { inputThread.interrupt() } // we interrupt the input thread to notify it that it can terminate
outputThreads.foreach(_.join()) // this ensures that all output is complete before returning (waitFor does not ensure this)
andCleanup { p.waitFor() }// wait for the process to terminate
p.exitValue()
}
override def destroy() =
{
try { p.destroy() }
finally { inputThread.interrupt() }
}
andCleanup { p.destroy() }
private[this] def andCleanup[T](action: => Unit): Unit =
try
{
try { action }
finally { inputThread.interrupt() } // we interrupt the input thread to notify it that it can terminate
outputThreads.foreach(_.join()) // this ensures that all output is complete before returning (waitFor does not ensure this)
}
finally
BasicIO.closeProcessStreams(p)
}
private class FileOutput(file: File, append: Boolean) extends OutputStreamBuilder(new FileOutputStream(file, append), file.getAbsolutePath)