Question

我需要能够运行外部应用程序并与之交互，就像我从命令行手动运行它一样。我发现的所有示例只涉及运行程序和捕获输出。

下面是一个非常简单的例子，我希望说明我想要实现的目标。

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...

}

我试图调整我已经发现的各种例子，但没有成功。似乎即使＆＃39; rm＆＃39;正在等待回复，Go关闭了这个过程。

我们非常感谢您提供的任何示例，文章或建议。非常感谢提前。

Answer 1

你有两种可能性。首先是使用ReadLine（），但仅当应用程序输出为实线时才有效，您可以等待\ n。这不是rm的情况，因此您必须为Scanner开发自定义SplitFunction。两个版本都可以在下面找到。

请注意，您无法使用CombinedOutput，因为无法扫描。你必须使用管道。

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
}

这是扫描仪的第二个版本，它同时使用\ n和？作为行分隔符：

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
}

从代码内与外部应用程序交互

1 个答案: