从 exec.Command 写入文件

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《从 exec.Command 写入文件》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在尝试将 bash 命令中的文件写入 go 中的文件中。 请注意，这里使用 go 而不是 bash 有几个原因：我有一些更多的逻辑，例如解析配置文件，我想并行运行多个数据库的代码，最后执行一些更复杂的数据操作。

dumpStr := fmt.Sprintf("pg_dump -U %s -h %s %s | gzip", DbUserName, DbHost, DbName)
    cmd := exec.Command("bash", "-c", dumpStr)
    cmd.Env = append(cmd.Env, "PGPASSWORD="+DbPassword)

    outfile, err := os.Create(DbName + ".gz")
    if err != nil {
        panic(err)
    }
    outfile = cmd.Stdout
    defer outfile.Close()
    err = cmd.Start()
    if err != nil {
        panic(err)
    }
    cmd.Wait()

但是，我得到的结果是空的。 如果我从 cli 执行 dumpstr 但不是从该代码执行，我会获取数据... 我错过了什么？

解决方案

正如 flimzy 所说，您没有捕获 pg_dump 的输出。您可以使用 go 来完成此操作，也可以使用 pg_dump -s --file。它还可以使用 --compress 进行压缩，因此无需通过管道传输到 gzip。这样就不需要 bash 并且可以避免 shell 引用问题。

cmd := exec.command(
    "pg_dump",
    "--compress=9",
    "--file="+dbname + ".gz",
    "-u"+dbusername,
    "-h"+dbhost,
    dbname,
)

log.print("running pg_dump...")

if err := cmd.run(); err != nil {
    log.fatal(err)
}

更简单、更安全。

为了便于说明，这里介绍了如何在 go 中完成这一切。

使用 Cmd.StdoutPipe 获取 pg_dump 的标准输出的开放 io 读取器。然后使用 io.Copy 从 stdout 复制到您打开的文件。

@peter 指出，从 Cmd.Stdout is an io.Reader 开始，将打开的文件分配给 cmd.stdout 并让 cmd 直接写入会更简单。

// Same as above, but no --file.
cmd := exec.Command(
    "pg_dump",
    "--compress=9",
    "-U"+DbUserName,
    "-h"+DbHost,
    DbName,
)

// Open the output file
outfile, err := os.Create(DbName + ".gz")
if err != nil {
    log.Fatal(err)
}
defer outfile.Close()

// Send stdout to the outfile. cmd.Stdout will take any io.Writer.
cmd.Stdout = outfile

// Start the command
if err = cmd.Start(); err != nil {
    log.Fatal(err)
}

log.Print("Waiting for command to finish...")

// Wait for the command to finish.
if err = cmd.Wait(); err != nil {
    log.Fatal(err)
}

此外，您只检查命令是否启动，而不检查命令是否成功运行。

来自 Cmd.Start 的文档。

start 启动指定的命令，但不等待其完成。

一旦命令退出，wait 方法将返回退出代码并释放相关资源。

您正在检查 cmd.start 是否有错误，但没有检查 cmd.wait。检查 cmd.start 的错误仅意味着命令已启动。如果程序运行时出现错误，您将不知道错误是什么。

您需要实际使用命令的输出。你没有那样做。为此，请使用 StdoutPipe 方法，然后您可以将程序中的标准输出复制到您的文件中。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。