go exec 对不同 shell 命令的不同行为

Question

我正在尝试对控制台 go 应用程序使用不同的 shell 命令，由于某些原因，以下交互式 shell 的行为有所不同。

此代码打印 mongoDB 查询的结果：

cmd := exec.Command("sh", "-c", "mongo --quiet --host=localhost blog")
stdout, _ := cmd.StdoutPipe()

stdin, _ := cmd.StdinPipe()
stdoutScanner := bufio.NewScanner(stdout)

go func() {
    for stdoutScanner.Scan() {
        println(stdoutScanner.Text())
    }
}()

cmd.Start()
io.WriteString(stdin, "db.getCollection('posts').find({status:'ACTIVE'}).itcount()\n")

//can't finish command, need to reuse it for other queries
//stdin.Close()
//cmd.Wait()

time.Sleep(2 * time.Second)

但是 Neo4J shell 的相同代码不会打印任何内容：

cmd := exec.Command("sh", "-c", "cypher-shell -u neo4j -p 121314 --format plain")
stdout, _ := cmd.StdoutPipe()

stdin, _ := cmd.StdinPipe()
stdoutScanner := bufio.NewScanner(stdout)

go func() {
    for stdoutScanner.Scan() {
        println(stdoutScanner.Text())
    }
}()

cmd.Start()
io.WriteString(stdin, "match (n) return count(n);\n")

//can't finish the command, need to reuse it for other queries
//stdin.Close()
//cmd.Wait()
time.Sleep(2 * time.Second)

有什么区别？我怎样才能让第二个工作？（不关闭命令）

P.S 当我直接打印到 os.Stdout:

时，Neo4J 工作正常

cmd := exec.Command("sh", "-c", "cypher-shell -u neo4j -p 121314 --format plain")

cmd.Stdout = os.Stdout

stdin, _ := cmd.StdinPipe()

cmd.Start()
io.WriteString(stdin, "match (n) return count(n);\n")

//stdin.Close()
//cmd.Wait()
time.Sleep(2 * time.Second)

Answer 1

当 cypher-shell 的输入不是一个（交互式）终端时，它期望读取整个输入和将其作为单个脚本执行。 “整个输入”是指“EOF 之前的所有内容”。这是典型的 REPL 程序：例如，python 的行为也是如此。

所以你的 Cypher 代码直到你 stdin.Close() 才开始执行。您的 cmd.Stdout = os.Stdout 示例似乎有效，因为当您的 Go 程序退出时 stdin 被隐式关闭，并且只有 then 会 cypher-shell 执行您的代码并打印到stdout，它仍然连接到您的终端。

您或许应该以不同的方式构建流程。例如，你不能为每个查询运行一个新的 cypher-shell 吗？

但是，如果所有其他方法都失败了，您可以通过欺骗 cypher-shell 认为它的标准输入 is a terminal. This is called a “pty”, and you can do it in Go with github.com/kr/pty 来解决这个问题。要点是这也会使 cypher-shell 打印提示并回显您的输入，如果您希望以编程方式处理输出，则必须检测并丢弃这些输入。

cmd := exec.Command("sh", "-c", "cypher-shell -u neo4j -p 121314 --format plain")
f, _ := pty.Start(cmd)
stdoutScanner := bufio.NewScanner(f)
cmd.Start()

// Give it some time to start, then read and discard the startup banner.
time.Sleep(2 * time.Second)
f.Read(make([]byte, 4096))

go func() {
    for stdoutScanner.Scan() {
        println(stdoutScanner.Text())
    }
}()

io.WriteString(f, "match (n) return count(n);\n")
time.Sleep(2 * time.Second)

io.WriteString(f, "match (n) return count(n) + 123;\n")
time.Sleep(2 * time.Second)

旁白 1: 在您的示例中，您不需要 sh -c，因为您没有使用 shell 的任何功能。您可以通过运行ning cypher-shell 直接避免额外的 shell 过程的开销：

cmd := exec.Command("cypher-shell", "-u", "neo4j", "-p", "121314", "--format", "plain")

旁白 2： 不要丢弃生产代码中返回的 error 值。

go exec 对不同 shell 命令的不同行为

Different behavior of go exec for different shell commands

cmd

go

interactive-shell