我正在使用 Scala scala.sys.process 库。
我知道我可以使用! 捕获退出代码并使用!! 捕获输出，但是如果我想同时捕获两者怎么办？
我看到这个答案https://stackoverflow.com/a/6013932/416338 看起来很有希望，但我想知道是否有一个班轮，我错过了什么。
【问题讨论】：
标签： scala
我有以下实用方法来运行命令：
import sys.process._
def runCommand(cmd: Seq[String]): (Int, String, String) = {
val stdoutStream = new ByteArrayOutputStream
val stderrStream = new ByteArrayOutputStream
val stdoutWriter = new PrintWriter(stdoutStream)
val stderrWriter = new PrintWriter(stderrStream)
val exitValue = cmd.!(ProcessLogger(stdoutWriter.println, stderrWriter.println))
stdoutWriter.close()
stderrWriter.close()
(exitValue, stdoutStream.toString, stderrStream.toString)
}
如您所见，它捕获标准输出、标准错误和结果代码。
【讨论】：
您可以使用ProcessIO。我在 Specs2 测试中需要类似的东西，我必须根据stdin 上的输入（in 和out 的类型为String）检查退出值以及进程的输出：
"the operation" should {
f"return '$out' on input '$in'" in {
var res = ""
val io = new ProcessIO(
stdin => { stdin.write(in.getBytes)
stdin.close() },
stdout => { res = convertStreamToString(stdout)
stdout.close() },
stderr => { stderr.close() })
val proc = f"$operation $file".run(io)
proc.exitValue() must be_==(0)
res must be_==(out)
}
}
我想这可能会对你有所帮助。在示例中，我忽略了来自 stderr 的内容。
【讨论】：
您可以指定捕获文本的输出流：
import sys.process._
val os = new java.io.ByteArrayOutputStream
val code = ("volname" #> os).!
os.close()
val opt = if (code == 0) Some(os.toString("UTF-8")) else None
【讨论】：
val code = ("volname" #> os).! 行就会挂起。
ByteArrayOutputStream 替换为 File，它会起作用
BasicIO 或 ProcessLogger 的单行使用很有吸引力。
scala> val sb = new StringBuffer
sb: StringBuffer =
scala> ("/bin/ls /tmp" run BasicIO(false, sb, None)).exitValue
res0: Int = 0
scala> sb
res1: StringBuffer = ...
或
scala> import collection.mutable.ListBuffer
import collection.mutable.ListBuffer
scala> val b = ListBuffer[String]()
b: scala.collection.mutable.ListBuffer[String] = ListBuffer()
scala> ("/bin/ls /tmp" run ProcessLogger(b append _)).exitValue
res4: Int = 0
scala> b mkString "\n"
res5: String = ...
根据捕获的含义，您可能对输出感兴趣，除非退出代码不为零。在这种情况下，处理异常。
scala> val re = "Nonzero exit value: (\\d+)".r.unanchored
re: scala.util.matching.UnanchoredRegex = Nonzero exit value: (\d+)
scala> Try ("./bomb.sh" !!) match {
| case Failure(f) => f.getMessage match {
| case re(x) => println(s"Bad exit $x")
| }
| case Success(s) => println(s)
| }
warning: there were 1 feature warning(s); re-run with -feature for details
Bad exit 3
【讨论】：
Nonzero exit value: (...)...
import sys.process._
“亚历克斯克鲁斯”在您的链接中提供的响应相当简洁，除非性能较差。
你可以扩展 sys.process.ProcessLogger 来管理
var out = List[String]()
var err = List[String]()
在内部，带有用于 out.reverse 和 err.reverse 结果的 getter。
【讨论】：
这是一个非常简单的 Scala 包装器，它允许您检索标准输出、标准错误和退出代码。
import scala.sys.process._
case class ProcessInfo(stdout: String, stderr: String, exitCode: Int)
object CommandRunner {
def runCommandAndGetOutput(command: String): ProcessInfo = {
val stdout = new StringBuilder
val stderr = new StringBuilder
val status = command ! ProcessLogger(stdout append _, stderr append _)
ProcessInfo(stdout.toString(), stderr.toString(), status)
}
}
【讨论】：
我结合了这些并想出了这个。预期的 RC 在那里，因为我有一个程序需要在一个项目中运行，它在工作时返回 1。这确实取决于异常的文本，但它仍然会做一些不匹配的合理事情。
private val ProcessErrorP: Regex = "(.*): error=(\\d+),(.*)".r.unanchored
case class ProcessInfo(stdout: String, stderr: String, exitCode: Int, private val expectedRd: Int) {
def succeeded: Boolean = exitCode == expectedRd
def failed: Boolean = !succeeded
def asOpt: Option[String] = if (succeeded) None else Some(stderr)
}
/**
* Run a simple command
* @param command -- what to run
* @return -- what happened
*/
def run(command: String, expectedRc: Int = 0): ProcessInfo = {
try {
val stdout = new StringBuilder
val stderr = new StringBuilder
val status = command ! ProcessLogger(stdout append _, stderr append _)
ProcessInfo(stdout.toString(), stderr.toString(), status, expectedRc)
} catch {
case io: IOException =>
val dm = io.getMessage
dm match {
case ProcessErrorP(message, code, reason) =>
ProcessInfo("", s"$message, $reason", code.toInt, expectedRc)
case m: String =>
ProcessInfo("", m, 999, expectedRc)
}
}
}
【讨论】：