同时读写 Golang 中的 os.OpenFile？

在Golang实战开发的过程中，我们经常会遇到一些这样那样的问题，然后要卡好半天，等问题解决了才发现原来一些细节知识点还是没有掌握好。今天golang学习网就整理分享《同时读写 Golang 中的 os.OpenFile？》，聊聊，希望可以帮助到正在努力赚钱的你。

我有在执行系统命令时写入日志文件的代码。例如

    logfile, err := os.OpenFile(THIS_LOG_FILE, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0600)
    if err != nil {
        return err
    }
    cmd.Stderr = logfile
    cmd.Stdout = logfile

    go func() {
        err := cmd.Run()
        if err != nil {
            // WANT TO LOG ERROR HERE
        }
    }()

在“// want to log”行，除了先前分配的日志文件目标之外，我还想将内容输出到标准记录器。有没有办法将其记录在内存中？或者我应该将所有内容写入内存缓冲区并在最后刷新？

为了澄清，在捕获内存中命令的输出时，我可以解析它并在运行的程序中采取行动（处理错误/等）。当我写入日志文件时，该信息就会丢失。

我的问题是，理论上，我可以从我刚刚编写的文件中读回该内容，但这似乎很浪费（如果命令失败，则很容易失败）。

正确答案

如果我理解正确的话，你想在执行 shell 命令时将 stdout/stderror 的内容写入文件。

由于stdout和stderror实现了readcloser接口，因此可以通过io.multireader合并它们，并执行io.copy从源到目标的操作。

以下代码片段实现了管道

package main

import (
    "io"
    "log"
    "os"
    "os/exec"
)

func main() {
    // prepare the command
    cmd := exec.Command("your-shell-command.sh")

    // get the stdout and stderr stream
    erc, err := cmd.StderrPipe()
    if err != nil {
        log.Fatalln("Failed to get stderr reader: ", err)
    }
    orc, err := cmd.StdoutPipe()
    if err != nil {
        log.Fatalln("Failed to get stdout reader: ", err)
    }

    // combine stdout and stderror ReadCloser
    rc := io.MultiReader(erc, orc)

    // Prepare the writer
    f, err := os.Create("output.log")
    if err != nil {
        log.Fatalln("Failed to create file")
    }
    defer f.Close()
    // Command.Start starts a new go routine
    if err := cmd.Start(); err != nil {
        log.Println("Failed to start the command")
    }

    // add the TeeReader.
    var buf bytes.Buffer 
    tr := io.TeeReader(rc, &buf) 
    if _, err := io.Copy(f, tr); err != nil { 
        logger.Fatalf("Failed to stream to file: %s", err) 
    }


    if err := cmd.Wait(); err != nil {
        log.Println("Failed to wait the command to execute: ", err)
    }
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《同时读写 Golang 中的 os.OpenFile？》文章吧，也可关注golang学习网公众号了解相关技术文章。