从代码中与外部应用程序进行交互

huangapple go评论82阅读模式
英文:

Interact with external application from within code

问题

我需要能够运行一个外部应用程序,并与其进行交互,就像我从命令行手动运行它一样。我找到的所有示例都只涉及运行程序并捕获输出。

下面是一个非常简单的示例,我希望能够说明我想要实现的目标。

package main

import (
	"fmt"
	"log"
	"os/exec"
)

func main() {

  cmd := exec.Command("rm", "-i", "somefile.txt")
  out, err := cmd.CombinedOutput()
  if err != nil {
    log.Fatal(err)
  }
  if string(out) == "Remove file 'somefile.txt'?" {
    // send the response 'y' back to the rm process
  }

  // 程序正常完成...

}

我尝试过修改各种示例来实现这一点,但都没有成功。似乎即使 'rm' 等待响应,Go 也会关闭该进程。

如果您能提供任何示例、文章或建议,我将非常感激。非常感谢!

英文:

I need to be able to run an external application and interact with it as though I was manually running it from the command-line. All the examples I find only deal with running the program and capturing the output.

Below is a very simple example that I hope illustrates what I am trying to accomplish.

package main

import (
	"fmt"
	"log"
	"os/exec"
)

func main() {

  cmd := exec.Command("rm", "-i", "somefile.txt")
  out, err := cmd.CombinedOutput()
  if err != nil {
    log.Fatal(err)
  }
  if string(out) == "Remove file 'somefile.txt'?" {
    // send the response 'y' back to the rm process
  }

  // program completes normally...

}

I've tried to tweak various examples that I've found to accomplish this with zero success. It seems that even though 'rm' is waiting for a response, Go closes the process.

Any examples, articles, or advice you can provide would be greatly appreciated. Many thanks in advance.

答案1

得分: 2

你有两种可能性。第一种是使用ReadLine(),但只有在应用程序输出是完整行时才有效,并且你可以等待\n。这在rm命令中不适用,所以你需要为Scanner开发一个自定义的SplitFunction。下面可以找到两个版本。

请注意,你不能使用CombinedOutput,因为它无法被扫描。你必须使用管道。

package main

import (
	"bufio"
	"log"
	"os/exec"
)

func main() {

	cmd := exec.Command("rm", "-i", "somefile.txt")

	// Stdout + stderr
	out, err := cmd.StderrPipe() // rm将提示写入err
	if err != nil {
		log.Fatal(err)
	}
	r := bufio.NewReader(out)

	// Stdin
	in, err := cmd.StdinPipe()
	if err != nil {
		log.Fatal(err)
	}
	defer in.Close()

	// 启动命令!
	err = cmd.Start()
	if err != nil {
		log.Fatal(err)
	}

	line, _, err := r.ReadLine()

	for err != nil {
		if string(line) == "Remove file 'somefile.txt'?" {
			in.Write([]byte("y\n"))
		}
		line, _, err = r.ReadLine()
	}

	// 程序正常完成...
}

这是第二个版本,它使用\n和?作为行分隔符:

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"log"
	"os/exec"
)

// 丑陋的hack,这是bufio.ScanLines添加了?作为另一个分隔符:D
func new_scanner(data []byte, atEOF bool) (advance int, token []byte, err error) {
	if atEOF && len(data) == 0 {
		return 0, nil, nil
	}
	if i := bytes.IndexByte(data, '\n'); i >= 0 {
		// 我们有一个完整的以换行符结尾的行。
		fmt.Printf("nn\n")
		return i + 1, data[0:i], nil
	}
	if i := bytes.IndexByte(data, '?'); i >= 0 {
		// 我们有一个完整的以?结尾的行。
		return i + 1, data[0:i], nil
	}
	// 如果我们在EOF处,我们有一行最后一个没有结束符。返回它。
	if atEOF {
		return len(data), data, nil
	}
	// 请求更多数据。
	return 0, nil, nil
}

func main() {

	cmd := exec.Command("rm", "-i", "somefile.txt")

	// Stdout + stderr
	out, err := cmd.StderrPipe() // 再次,rm将提示写入stderr
	if err != nil {
		log.Fatal(err)
	}

	scanner := bufio.NewScanner(out)
	scanner.Split(new_scanner)

	// Stdin
	in, err := cmd.StdinPipe()
	if err != nil {
		log.Fatal(err)
	}
	defer in.Close()

	// 启动命令!
	err = cmd.Start()
	if err != nil {
		log.Fatal(err)
	}

	// 开始扫描
	for scanner.Scan() {
		line := scanner.Text()
		if line == "rm: remove regular empty file ‘somefile.txt’" {
			in.Write([]byte("y\n"))
		}
	}
	// 报告scanner的错误
	if err := scanner.Err(); err != nil {
		log.Fatal(err)
	}

	// 程序正常完成...
}
英文:

You have two possibilities. First is to use ReadLine() but that works only if application output is full lines, and you can wait for \n. This is not the case with rm, so you have to develop a custom SplitFunction for Scanner. Both versions can be found below.

Please note that you can not use CombinedOutput, as it can not be Scanned. You have to use the pipes.

package main
import (
"bufio"
//"fmt"
"log"
"os/exec"
)
func main() {
cmd := exec.Command("rm", "-i", "somefile.txt")
// Stdout + stderr
out, err := cmd.StderrPipe() // rm writes the prompt to err
if err != nil {
log.Fatal(err)
}
r := bufio.NewReader(out)
// Stdin
in, err := cmd.StdinPipe()
if err != nil {
log.Fatal(err)
}
defer in.Close()
// Start the command!
err = cmd.Start()
if err != nil {
log.Fatal(err)
}
line, _, err := r.ReadLine()
for err != nil {
if string(line) == "Remove file 'somefile.txt'?" {
in.Write([]byte("y\n"))
}
line, _, err = r.ReadLine()
}
// program completes normally...s
}

This is a second version with the scanner, and it uses both \n and ? as line delimiters:

package main
import (
"bufio"
"bytes"
"fmt"
"log"
"os/exec"
)
// Ugly hack, this is bufio.ScanLines with ? added as an other delimiter :D
func new_scanner(data []byte, atEOF bool) (advance int, token []byte, err error) {
if atEOF && len(data) == 0 {
return 0, nil, nil
}
if i := bytes.IndexByte(data, '\n'); i >= 0 {
// We have a full newline-terminated line.
fmt.Printf("nn\n")
return i + 1, data[0:i], nil
}
if i := bytes.IndexByte(data, '?'); i >= 0 {
// We have a full ?-terminated line.
return i + 1, data[0:i], nil
}
// If we're at EOF, we have a final, non-terminated line. Return it.
if atEOF {
return len(data), data, nil
}
// Request more data.
return 0, nil, nil
}
func main() {
cmd := exec.Command("rm", "-i", "somefile.txt")
// Stdout + stderr
out, err := cmd.StderrPipe() // Again, rm writes prompts to stderr
if err != nil {
log.Fatal(err)
}
scanner := bufio.NewScanner(out)
scanner.Split(new_scanner)
// Stdin
in, err := cmd.StdinPipe()
if err != nil {
log.Fatal(err)
}
defer in.Close()
// Start the command!
err = cmd.Start()
if err != nil {
log.Fatal(err)
}
// Start scanning
for scanner.Scan() {
line := scanner.Text()
if line == "rm: remove regular empty file ‘somefile.txt’" {
in.Write([]byte("y\n"))
}
}
// Report scanner's errors
if err := scanner.Err(); err != nil {
log.Fatal(err)
}
// program completes normally...s
}

huangapple
  • 本文由 发表于 2014年12月6日 02:49:55
  • 转载请务必保留本文链接:https://go.coder-hub.com/27322722.html
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定