在go语言种实现复制目录 - 包括子目录和文件示例

Go  /  管理员 发布于 1年前   482

如何用Go复制目录的所有内容到另一个目录，复制过程将执行最低限度的正确性检查，并确保所有权限也被复制过来。

示例代码：

copydir.go

package main
 import (
   "os"
   "flag"
   "fmt"
   "io"
 )
 func CopyFile(source string, dest string) (err error) {
     sourcefile, err := os.Open(source)
     if err != nil {
         return err
     }
     defer sourcefile.Close()
     destfile, err := os.Create(dest)
     if err != nil {
         return err
     }
     defer destfile.Close()
     _, err = io.Copy(destfile, sourcefile)
     if err == nil {
         sourceinfo, err := os.Stat(source)
         if err != nil {
             err = os.Chmod(dest, sourceinfo.Mode())
         }
     }
     return
 }
 func CopyDir(source string, dest string) (err error) {
     // 获取源目录的属性
     sourceinfo, err := os.Stat(source)
     if err != nil {
         return err
     }
     // 创建目标目录
     err = os.MkdirAll(dest, sourceinfo.Mode())
     if err != nil {
         return err
     }
     directory, _ := os.Open(source)
     objects, err := directory.Readdir(-1)
     for _, obj := range objects {
         sourcefilepointer := source + "/" + obj.Name()
         destinationfilepointer := dest + "/" + obj.Name()
         if obj.IsDir() {
             // 创建子目录 - 递归
             err = CopyDir(sourcefilepointer, destinationfilepointer)
             if err != nil {
                 fmt.Println(err)
             }
         } else {
             // 进行复制
             err = CopyFile(sourcefilepointer, destinationfilepointer)
             if err != nil {
                 fmt.Println(err)
             }
         }
     }
     return
 }
 func main() {
   flag.Parse() // 获取源目录和目标目录
   source_dir := flag.Arg(0) // 从第一个参数中获取源目录
   dest_dir := flag.Arg(1) // 从第 2 个参数中获取目标目录
   fmt.Println("源 :" + source_dir)
   // 检查源目录是否存在
   src, err := os.Stat(source_dir)
   if err != nil {
      panic(err)
   }
   if !src.IsDir() {
     fmt.Println("源文件不是目录")
     os.Exit(1)
   }
   // 创建目标目录
   fmt.Println("目的地 :"+ dest_dir)
   _, err = os.Open(dest_dir)
   if !os.IsNotExist(err) {
     fmt.Println("目标目录已存在。终止！")
     os.Exit(1)
   }
   err = CopyDir(source_dir, dest_dir)
   if err != nil {
      fmt.Println(err)
   } else {
      fmt.Println("复制的目录")
   }
 }

输出：

通过执行以下来构建程序

>go build copydir.go 

并执行 

>/copydir /Users/admin/source_test /Users/admin/test

源 :/Users/admin/source_test
目的地 :/Users/admin/test
复制的目录

注意

我曾尝试使用 filepath.walk() 实现该功能，但在递归调用该函数时遇到了问题。

因此，在本文，将使用 readdir(-1)