使用换行符来执行命令

一分耕耘，一分收获！既然打开了这篇文章《使用换行符来执行命令》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

我正在尝试使用 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())
}

解决方案

这里有一个问题是您使用 multireader 时遇到的问题：

cmd.stdin = io.multireader(bufin, os.stdin)

来自 docs for MultiReader（已添加重点）：

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

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

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

所以你的程序中发生的事情是这样的：cmd 的 stdin 读取 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)
    }
}

今天关于《使用换行符来执行命令》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！