命令需要换行才能完成

Question

我正在尝试使用 Go 将 sqlite3 作为进程运行。我想将cmd.Stdin 与bytes.Buffer 和os.Stdin 结合起来。然而，当我在标准输入字节缓冲区上写.quit 命令时，程序并没有直接退出，而是在等待来自os.stdin 的换行符。当它收到来自os.stdin 的换行符时，它就退出了。

我尝试调用os.Stdin.Write([]byte("\n"))，但没有成功。如何在.quit 命令之后直接退出而不与os.Stdin 进行任何交互？

func main() {
    cmd := exec.Command("/usr/bin/sqlite3")

    bufOut, bufErr, bufIn := &bytes.Buffer{}, &bytes.Buffer{}, &bytes.Buffer{}

    cmd.Stdout = io.MultiWriter(bufOut, os.Stdout)
    cmd.Stderr = io.MultiWriter(bufErr, os.Stderr)
    cmd.Stdin = io.MultiReader(bufIn, os.Stdin)

    if err := cmd.Start(); err != nil {
        log.Fatal(err)
    }

    // Execute .help command on sqlite3
    if _, err := bufIn.Write([]byte(".help\n")); err != nil {
        log.Fatal(err)
    }

    // Execute .quit command on sqlite3 (should exit here)
    if _, err := bufIn.Write([]byte(".quit\n")); err != nil {
        log.Fatal(err)
    }
    // Nevertheless, it requires a '\n' from os.Stdin before exiting

    if err := cmd.Wait(); err != nil {
        log.Fatal(err)
    }

    fmt.Println("out:", bufOut.String(), "err:", bufErr.String())
}

Answer 1

一个问题是您在这里使用 MultiReader：

cmd.Stdin = io.MultiReader(bufIn, os.Stdin)

来自docs for MultiReader（强调添加）：

MultiReader 返回一个 Reader，它是提供的输入阅读器的逻辑串联。 它们是按顺序读取的。

其他相关信息来自the documentation for Cmd.Stdin。它说如果Stdin 不是nil 或*os.File，那么

在命令执行期间，一个单独的 goroutine 从标准输入读取数据并通过管道将该数据传递给命令。在这种情况下，直到 goroutine 停止复制，Wait 才会完成，因为它已经到达 Stdin 的末尾（EOF 或读取错误），或者因为写入管道返回了错误。

所以在你的程序中发生的事情是这样的：cmd 的Stdin-reading goroutine 去读取 MultiReader bufIn。首先它会读取您写入bufIn 的数据，然后它会尝试读取更多数据。 MultiReader 看到bufIn 已用尽，并移动到Read 其下一个参数os.Stdin。当您调用cmd.Wait 时，它会阻塞等待MultiReader 发送EOF，但MultiReader 本身会阻塞尝试从os.Stdin 读取。在您发送 EOF（例如，按 ctrl-D）或按 Enter 之前，什么都不会发生。 （我不太确定为什么按 enter 有效，因为这通常不会导致 EOF，但如果我弄清楚细节，我会更新。）

另一个问题是竞争条件，如Cerise Limón notes - goroutine 完全有可能在您写入之前从您的bufIn 读取，在这种情况下，您随后写入bufIn 的任何内容都将是忽略。

与其使用缓冲区与您的进程通信，不如使用cmd.StdoutPipe 和cmd.StdinPipe。但是，这并不能解决整个问题 - 您不能使用阻塞 IO 以交互方式与子进程通信，因为您最终将等待从等待您编写命令的进程中读取。可能你最好的选择是使用 goroutine 从命令中读取并使用带有超时的 select 来读取输出，直到可以合理地确定没有更多的输出。

这是一个实现。它至少需要一些充实，错误处理，但它确实有效。 startScanner 设置 goroutine 从命令输出中读取行并将它们写入通道。 readLinesFromChannelWithTimeout 从给定通道读取，直到给定超时过去且没有数据。请注意，您必须在调用cmd.Wait() 之前调用cmdIn.Close()，否则后者将无限期挂起。

package main

import (
    "bufio"
    "fmt"
    "io"
    "log"
    "os/exec"
    "time"
)

func readLinesFromChannelWithTimeout(ch chan string, timeout time.Duration) []string {
    var lines []string
    for {
        select {
        case line, ok := <-ch:
            if !ok {
                return lines
            } else {
                lines = append(lines, line)
            }
        case <-time.After(timeout):
            return lines
        }
    }
}

func startScanner(cmdOut io.ReadCloser) chan string {
    ch := make(chan string)
    go func(ch chan string) {
        defer close(ch)
        scanner := bufio.NewScanner(cmdOut)
        for scanner.Scan() {
            ch <- scanner.Text()
        }
    }(ch)
    return ch
}

func main() {
    cmd := exec.Command("/usr/bin/sqlite3")
    cmdIn, _ := cmd.StdinPipe()
    cmdOut, _ := cmd.StdoutPipe()
    cmd.Start()

    ch := startScanner(cmdOut)

    var lines []string

    io.WriteString(cmdIn, ".help\n")
    lines = readLinesFromChannelWithTimeout(ch, time.Millisecond*100)
    fmt.Printf("Got %d lines from .help\n", len(lines))

    io.WriteString(cmdIn, ".show\n")
    lines = readLinesFromChannelWithTimeout(ch, time.Millisecond*100)
    fmt.Printf("Got %d lines from .show\n", len(lines))

    cmdIn.Close() // vital! Wait() will hang otherwise
    if err := cmd.Wait(); err != nil {
        log.Fatal(err)
    }
}