如何在 Go 中复制文件？

Question

我有以下函数将文件（实际上是 io.Reader）复制到目标字符串位置。然而，似乎只有文件的一部分被实际复制，导致文件损坏。我做错了什么？

func CopyFile(in io.Reader, dst string) (err error) {

    // Does file already exist? Skip
    if _, err := os.Stat(dst); err == nil {
        return nil
    }

    err = nil

    out, err := os.Create(dst)
    if err != nil {
        fmt.Println("Error creating file", err)
        return
    }

    defer func() {
        cerr := out.Close()
        if err == nil {
            err = cerr
        }
    }()


    var bytes int64
    if bytes, err = io.Copy(out, in); err != nil {
        fmt.Println("io.Copy error")
        return
    }
    fmt.Println(bytes)

    err = out.Sync()
    return
}

我将其与

filepath.Walk(dir, visit)

方法一起使用来处理目录中的文件。

// Process each matching file on our walk down the filesystem
func visit(path string, f os.FileInfo, err error) error {

    if reader, err := os.Open(path); err == nil {
        defer reader.Close()

        // http://golang.org/pkg/os/#FileInfo
        statinfo, err := reader.Stat()

        if err != nil {
            fmt.Println(err)
            return nil
        }

        fmt.Println()
        fmt.Println(statinfo.Size())

        // Directory exists and is writable
        err = CopyFile(reader, "/tmp/foo/"+f.Name())

        if err != nil {
            fmt.Println(err)
        }

    } else {
        fmt.Println("Impossible to open the file:", err)
    }
}

当前的最接近的问题我可以有一个可接受的答案，建议使用硬/软链接，并且如果文件已经存在则不会中止。

Answer 1

package main

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

func main() {
    srcFile, err := os.Open("test.txt")
    check(err)
    defer srcFile.Close()

    destFile, err := os.Create("test_copy.txt") // creates if file doesn't exist
    check(err)
    defer destFile.Close()

    _, err = io.Copy(destFile, srcFile) // check first var for number of bytes copied
    check(err)

    err = destFile.Sync()
    check(err)
}

func check(err error) {
    if err != nil {
        fmt.Println("Error : %s", err.Error())
        os.Exit(1)
    }
}

这段代码对我有用。请使用

io.Copy

的返回值检查复制的字节数。

Answer 2

另一个选项是

ReadFrom

:

package main
import "os"

func copyFile(in, out string) (int64, error) {
   i, e := os.Open(in)
   if e != nil { return 0, e }
   defer i.Close()
   o, e := os.Create(out)
   if e != nil { return 0, e }
   defer o.Close()
   return o.ReadFrom(i)
}

func main() {
   _, e := copyFile("in.txt", "out.txt")
   if e != nil {
      panic(e)
   }
}

https://golang.org/pkg/os#File.ReadFrom

Answer 3

您可以使用https://github.com/otiai10/copy

package main

import (
    "fmt"
    "log"

    "github.com/otiai10/copy"
)

func main() {
    src := "source_file.txt"
    dst := "destination_file.txt"

    err := copy.Copy(src, dst)
    if err != nil {
        log.Fatalf("Error copying file: %v", err)
    }

    fmt.Println("File copied successfully!")
}

Answer 4

复制 golang 最简单的方法是 - http://golang.org/pkg/io/#Copy