Golang程序如何将输出保存到文件中

如果你从控制台运行它，为什么不直接将输出重定向到文件呢？

来源：stackoverflow.com

如何用Bash将stdout和stderr都重定向并追加到一个文件？

标签： linux, bash, redirect, stream

在Go中，将程序输出保存到文件有多种方式，以下是几种常用方法：

1. 使用os.Create和fmt.Fprintf

package main

import (
    "fmt"
    "os"
)

func main() {
    // 创建或截断文件
    file, err := os.Create("output.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    // 将输出写入文件
    fmt.Fprintf(file, "程序开始执行\n")
    fmt.Fprintf(file, "计算结果: %d\n", 42)
    fmt.Fprintf(file, "字符串: %s\n", "Hello, World!")
    
    fmt.Println("输出已保存到output.txt")
}

2. 使用os.OpenFile进行更精细的控制

package main

import (
    "os"
    "io"
)

func main() {
    // 以追加模式打开文件（如果不存在则创建）
    file, err := os.OpenFile("output.log", 
        os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    // 写入数据
    data := []byte("日志条目: 程序启动成功\n")
    if _, err := file.Write(data); err != nil {
        panic(err)
    }
    
    // 写入多行内容
    multiLine := "第一行\n第二行\n第三行\n"
    if _, err := io.WriteString(file, multiLine); err != nil {
        panic(err)
    }
}

3. 同时输出到控制台和文件

package main

import (
    "fmt"
    "io"
    "os"
)

func main() {
    // 创建文件
    file, err := os.Create("output.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    // 创建多写入器
    multiWriter := io.MultiWriter(os.Stdout, file)
    
    // 同时输出到控制台和文件
    fmt.Fprintln(multiWriter, "这行会显示在控制台并保存到文件")
    fmt.Fprintf(multiWriter, "格式化输出: %v\n", []int{1, 2, 3})
    
    // 直接使用Write方法
    multiWriter.Write([]byte("直接写入字节数据\n"))
}

4. 使用bufio提高写入性能（适合大量数据）

package main

import (
    "bufio"
    "fmt"
    "os"
)

func main() {
    file, err := os.Create("large_output.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    writer := bufio.NewWriter(file)
    defer writer.Flush() // 确保所有缓冲数据写入文件
    
    // 写入大量数据
    for i := 0; i < 1000; i++ {
        fmt.Fprintf(writer, "行号: %d, 数据: %d\n", i, i*i)
    }
    
    fmt.Println("1000行数据已写入文件")
}

5. 重定向标准输出到文件

package main

import (
    "fmt"
    "os"
)

func main() {
    // 保存原始标准输出
    originalStdout := os.Stdout
    
    // 重定向标准输出到文件
    file, err := os.Create("stdout.txt")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    os.Stdout = file
    
    // 所有fmt.Print/Println/Printf都会写入文件
    fmt.Println("这行会写入文件而不是控制台")
    fmt.Printf("格式化: %s\n", "文件输出")
    
    // 恢复原始标准输出
    os.Stdout = originalStdout
    fmt.Println("这行会显示在控制台")
}

6. 结构化数据写入（JSON格式）

package main

import (
    "encoding/json"
    "os"
)

type Data struct {
    Name    string `json:"name"`
    Value   int    `json:"value"`
    Items   []string `json:"items"`
}

func main() {
    data := Data{
        Name:  "示例",
        Value: 100,
        Items: []string{"A", "B", "C"},
    }
    
    file, err := os.Create("data.json")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    
    encoder := json.NewEncoder(file)
    encoder.SetIndent("", "  ") // 美化输出
    
    if err := encoder.Encode(data); err != nil {
        panic(err)
    }
}

选择哪种方法取决于具体需求：简单文本输出用fmt.Fprintf，大量数据用bufio，需要同时输出到控制台和文件用io.MultiWriter，结构化数据用相应的编码器。