在Scala中调用外部命令作为子进程

时间:2011-06-12 04:05:36

标签: scala subprocess

在Python中,如果我想将外部命令作为子进程调用,我会执行以下操作:

from subprocess import Popen, PIPE
cmd = ['cat', '-be']
out, err = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE).communicate("some input")

在Scala中执行相同操作的标准方法是什么?使用Java的ProcessBuilder我提出了以下内容,但它非常难看:

def communicate(cmd: List[String], input: Option[String] = None): (String, String) = {

    val command = new java.util.ArrayList[String]()
    cmd.foreach(command.add(_))

    val builder = new ProcessBuilder(command)
    val process = builder.start()

    val stdinWriter = new java.io.PrintWriter((new java.io.OutputStreamWriter(new java.io.BufferedOutputStream(process.getOutputStream()))), true);
    val stdoutReader = new java.io.BufferedReader(new java.io.InputStreamReader(process.getInputStream()))
    val stderrReader = new java.io.BufferedReader(new java.io.InputStreamReader(process.getErrorStream()))

    input.foreach(stdinWriter.write(_))
    stdinWriter.close()

    def read(reader: java.io.BufferedReader): String = {
        val out = new ListBuffer[String]
        var line: String = reader.readLine()
        while (line != null) {
            out += line
            line = reader.readLine()
        }
        return out.result.mkString("\n")
    }

    val stdout = read(stdoutReader)
    val stderr = read(stderrReader)

    stdoutReader.close()
    stderrReader.close()

    return (stdout, stderr)

}

val (catout, caterr) = communicate(List("cat", "-be"), Some("some input"))
val (pwdout, pwderr) = communicate(List("pwd"))

Scala中是否有更好的替代品?

3 个答案:

答案 0 :(得分:9)

这里的答案How does the “scala.sys.process” from Scala 2.9 work?显示了如何使用新的Scala 2.9 scala.sys.process.Process。如果你不使用2.9,你可以使用sbt的过程部分; Scala流程源自哪个。

答案 1 :(得分:0)

你看过Apache commons exec http://commons.apache.org/exec/tutorial.html吗?

几年前我成功地从Scala中使用过它。挖掘了一些代码:

def quickRun(command: String, allowedExitCodes: Array[Int]): String = {
  val executor = new DefaultExecutor()
  val outputStream = new ByteArrayOutputStream()
  val errorStream = new ByteArrayOutputStream()

  val pumpStreamHandler = new PumpStreamHandler(outputStream, errorStream)
  executor.setStreamHandler(pumpStreamHandler)
  executor.setExitValues(allowedExitCodes)

  lazy val errorMsg = "Couldn't execute command: \"" + command + "\", errorStream: " + errorStream.toString()
  try {
    val exitValue = executor.execute(CommandLine.parse(command))
    if (executor.isFailure(exitValue)) {
      log.error(errorMsg)
      throw new CommandLineException(errorMsg)
    } else {
      return outputStream.toString()
    }
  } catch {
    case e: ExecuteException =>
      log.error(errorMsg)
      throw new CommandLineException(errorMsg)
    case e: IOException => throw new CommandLineException(errorMsg)
  }
}

但是scala.sys.process看起来更好,我认为你应该尽可能使用它......

答案 2 :(得分:0)

使用OS-Lib库,您的代码可以编写为以下Scala:

@ val res = os.proc("cat", "-be").call(stdin = "some input")
res: os.CommandResult = CommandResult(0, List(Left(     1   some input)))

@ res.out.string
res3: String = "     1\tsome input"

@ res.err.string
res4: String = ""