如何在 Go 中读取写入临时文件的数据？

在 Go 中读取写入临时文件的数据

在 Go 中，创建和读取临时文件可能会带来挑战。考虑以下简化的测试代码：

package main

import (
    "bufio"
    "fmt"
    "io/ioutil"
    "log"
    "os"
    "path/filepath"
)

func main() {

    tmpFile, err := ioutil.TempFile("", fmt.Sprintf("%s-", filepath.Base(os.Args[0])))
    if err != nil {
        log.Fatal("Could not create temporary file", err)
    }
    fmt.Println("Created temp file:", tmpFile.Name())
    defer tmpFile.Close()

    fmt.Println("Writing some data to the temp file")
    if _, err = tmpFile.WriteString("test data"); err != nil {
        log.Fatal("Unable to write to temporary file", err)
    } else {
        fmt.Println("Data should have been written")
    }

    fmt.Println("Trying to read the temp file now")
    s := bufio.NewScanner(tmpFile)
    for s.Scan() {
        fmt.Println(s.Text())
    }
    err = s.Err()
    if err != nil {
        log.Fatal("error reading temp file", err)
    }
}

虽然代码正确创建并写入临时文件，但尝试读取空输出中的结果。这是因为写操作将指针移动到文件末尾。要读取数据，我们需要回溯到开头。

解决方案：

要解决此问题，请添加 tmpFile.Seek(0, 0) 移动在尝试读取之前将指针返回到文件的开头：

    tmpFile.Seek(0, 0)
    s := bufio.NewScanner(tmpFile)
    for s.Scan() {
        fmt.Println(s.Text())
    }

通过此修改，代码可以正确读取和打印数据。请记住在退出之前使用 defer tmpFile.Close() 关闭文件，以确保正确的资源管理。

以上是如何在 Go 中读取写入临时文件的数据？的详细内容。更多信息请关注PHP中文网其他相关文章！