在Golang中复制文件的简单方法

有没有简单/快速的方法可以在Go中复制文件？

我找不到文档中的快速方法，并且搜索互联网也无济于事。

警告：此答案主要是关于向文件添加第二个链接，而不是复制内容。

一个强大的和高效率副本的概念很简单，但并不容易实现，因为需要处理一些边缘情况和系统限制由目标操作系统强加的，它的配置。

如果您只想复制现有文件，则可以使用os.Link(srcName, dstName)。这样可以避免在应用程序中移动字节并节省磁盘空间。对于大文件，这可以节省大量的时间和空间。

但是，各种操作系统对硬链接的工作方式都有不同的限制。根据您的应用程序和目标系统配置的不同，Link()调用可能无法在所有情况下都起作用。

如果您想要一个通用，健壮和高效的复制功能，请更新Copy()至：

1. 进行检查以确保至少某种形式的复制将成功（访问权限，目录存在等）
2. 检查两个文件是否已经存在并且使用相同 os.SameFile，如果相同则返回成功
3. 尝试链接，成功则返回
4. 复制字节（所有有效方法均失败），返回结果

一种优化方法是在go例程中复制字节，以便调用者不会在字节复制上进行阻塞。这样做会给调用方带来额外的复杂性，以使其无法正确处理成功/错误情况。

如果两者都需要，我将有两个不同的复制功能：CopyFile(src, dst string) (error)用于阻塞复制，CopyFileAsync(src, dst string) (chan c, error)对于异步情况，它将信令通道传递回调用方。

package main

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

// CopyFile copies a file from src to dst. If src and dst files exist, and are
// the same, then return success. Otherise, attempt to create a hard link
// between the two files. If that fail, copy the file contents from src to dst.
func CopyFile(src, dst string) (err error) {
    sfi, err := os.Stat(src)
    if err != nil {
        return
    }
    if !sfi.Mode().IsRegular() {
        // cannot copy non-regular files (e.g., directories,
        // symlinks, devices, etc.)
        return fmt.Errorf("CopyFile: non-regular source file %s (%q)", sfi.Name(), sfi.Mode().String())
    }
    dfi, err := os.Stat(dst)
    if err != nil {
        if !os.IsNotExist(err) {
            return
        }
    } else {
        if !(dfi.Mode().IsRegular()) {
            return fmt.Errorf("CopyFile: non-regular destination file %s (%q)", dfi.Name(), dfi.Mode().String())
        }
        if os.SameFile(sfi, dfi) {
            return
        }
    }
    if err = os.Link(src, dst); err == nil {
        return
    }
    err = copyFileContents(src, dst)
    return
}

// copyFileContents copies the contents of the file named src to the file named
// by dst. The file will be created if it does not already exist. If the
// destination file exists, all it's contents will be replaced by the contents
// of the source file.
func copyFileContents(src, dst string) (err error) {
    in, err := os.Open(src)
    if err != nil {
        return
    }
    defer in.Close()
    out, err := os.Create(dst)
    if err != nil {
        return
    }
    defer func() {
        cerr := out.Close()
        if err == nil {
            err = cerr
        }
    }()
    if _, err = io.Copy(out, in); err != nil {
        return
    }
    err = out.Sync()
    return
}

func main() {
    fmt.Printf("Copying %s to %s\n", os.Args[1], os.Args[2])
    err := CopyFile(os.Args[1], os.Args[2])
    if err != nil {
        fmt.Printf("CopyFile failed %q\n", err)
    } else {
        fmt.Printf("CopyFile succeeded\n")
    }
}

您已经拥有了在标准库中编写此类功能所需的所有功能。这是显而易见的代码。

// Copy the src file to dst. Any existing file will be overwritten and will not
// copy file attributes.
func Copy(src, dst string) error {
    in, err := os.Open(src)
    if err != nil {
        return err
    }
    defer in.Close()

    out, err := os.Create(dst)
    if err != nil {
        return err
    }
    defer out.Close()

    _, err = io.Copy(out, in)
    if err != nil {
        return err
    }
    return out.Close()
}

import (
    "io/ioutil"
    "log"
)

func checkErr(err error) {
    if err != nil {
        log.Fatal(err)
    }
}

func copy(src string, dst string) {
    // Read all content of src to data
    data, err := ioutil.ReadFile(src)
    checkErr(err)
    // Write data to dst
    err = ioutil.WriteFile(dst, data, 0644)
    checkErr(err)
}

如果您在linux / mac中运行代码，则只需执行系统的cp命令。

srcFolder := "copy/from/path"
destFolder := "copy/to/path"
cpCmd := exec.Command("cp", "-rf", srcFolder, destFolder)
err := cpCmd.Run()

它的处理有点像脚本，但是可以完成工作。另外，您需要导入“ os / exec”

在这种情况下，有两个条件需要验证，我更喜欢非嵌套代码

func Copy(src, dst string) (int64, error) {
  src_file, err := os.Open(src)
  if err != nil {
    return 0, err
  }
  defer src_file.Close()

  src_file_stat, err := src_file.Stat()
  if err != nil {
    return 0, err
  }

  if !src_file_stat.Mode().IsRegular() {
    return 0, fmt.Errorf("%s is not a regular file", src)
  }

  dst_file, err := os.Create(dst)
  if err != nil {
    return 0, err
  }
  defer dst_file.Close()
  return io.Copy(dst_file, src_file)
}

从Go 1.15（2020年8月）开始，可以使用File.ReadFrom：

package main

import (
   "log"
   "os"
)

func main() {
   in_o, e := os.Open("a.go")
   if e != nil {
      log.Fatal(e)
   }
   out_o, e := os.Create("b.go")
   if e != nil {
      log.Fatal(e)
   }
   out_o.ReadFrom(in_o)
}

这是复制文件的一种明显方法：

package main
import (
    "os"
    "log"
    "io"
)

func main() {
    sFile, err := os.Open("test.txt")
    if err != nil {
        log.Fatal(err)
    }
    defer sFile.Close()

    eFile, err := os.Create("test_copy.txt")
    if err != nil {
        log.Fatal(err)
    }
    defer eFile.Close()

    _, err = io.Copy(eFile, sFile) // first var shows number of bytes
    if err != nil {
        log.Fatal(err)
    }

    err = eFile.Sync()
    if err != nil {
        log.Fatal(err)
    }
}

如果您在Windows上，则可以像这样包装CopyFileW：

package utils

import (
    "syscall"
    "unsafe"
)

var (
    modkernel32   = syscall.NewLazyDLL("kernel32.dll")
    procCopyFileW = modkernel32.NewProc("CopyFileW")
)

// CopyFile wraps windows function CopyFileW
func CopyFile(src, dst string, failIfExists bool) error {
    lpExistingFileName, err := syscall.UTF16PtrFromString(src)
    if err != nil {
        return err
    }

    lpNewFileName, err := syscall.UTF16PtrFromString(dst)
    if err != nil {
        return err
    }

    var bFailIfExists uint32
    if failIfExists {
        bFailIfExists = 1
    } else {
        bFailIfExists = 0
    }

    r1, _, err := syscall.Syscall(
        procCopyFileW.Addr(),
        3,
        uintptr(unsafe.Pointer(lpExistingFileName)),
        uintptr(unsafe.Pointer(lpNewFileName)),
        uintptr(bFailIfExists))

    if r1 == 0 {
        return err
    }
    return nil
}

代码是受包装器启发的 C:\Go\src\syscall\zsyscall_windows.go

您可以使用“ exec”。Windows的exec.Command（“ cmd”，“ / c”，“ copy”，“ fileToBeCopied destinationDirectory”）对于Windows，我已经使用了它并且工作正常。您可以参考手册以获取更多关于exec的信息。

看看go-shutil。

但是请注意，它不会复制元数据。还需要有人来实施诸如移动之类的事情。

仅使用exec可能值得。