Golang 递归复制目录结构 - 使用 copy 库

在 Go 中递归复制目录结构是一个常见需求，标准库中没有直接提供这样的功能，但我们可以使用一些第三方库来简化操作。下面我将介绍如何使用 github.com/otiai10/copy 这个流行的库来实现目录递归复制。

安装 copy 库

首先需要安装这个库：

go get github.com/otiai10/copy

基本使用示例

package main

import (
	"fmt"
	"log"
	
	"github.com/otiai10/copy"
)

func main() {
	// 源目录路径
	src := "/path/to/source/directory"
	// 目标目录路径
	dst := "/path/to/destination/directory"

	// 递归复制目录
	err := copy.Copy(src, dst)
	if err != nil {
		log.Fatalf("复制失败: %v", err)
	}
	
	fmt.Println("目录复制完成!")
}

高级选项

copy 库提供了一些有用的选项来控制复制行为：

1. 跳过特定文件

err := copy.Copy(src, dst, copy.Options{
	Skip: func(src string) (bool, error) {
		// 跳过所有 .git 目录和 .DS_Store 文件
		return strings.Contains(src, ".git") || strings.HasSuffix(src, ".DS_Store"), nil
	},
})

2. 添加权限

err := copy.Copy(src, dst, copy.Options{
	AddPermission: 0600, // 为所有文件添加读写权限
})

3. 同步复制（先清空目标目录）

err := copy.Copy(src, dst, copy.Options{
	Sync: true, // 先删除目标目录内容
})

4. 保留文件权限

err := copy.Copy(src, dst, copy.Options{
	PermissionControl: copy.AddPermission(0600), // 保留原始权限并添加新权限
})

完整示例

下面是一个更完整的示例，包含错误处理和进度反馈：

package main

import (
	"fmt"
	"log"
	"os"
	"path/filepath"
	"strings"
	
	"github.com/otiai10/copy"
)

func main() {
	if len(os.Args) < 3 {
		fmt.Println("用法: copydir <源目录> <目标目录>")
		os.Exit(1)
	}
	
	src := os.Args[1]
	dst := os.Args[2]
	
	// 检查源目录是否存在
	if _, err := os.Stat(src); os.IsNotExist(err) {
		log.Fatalf("源目录不存在: %s", src)
	}
	
	// 打印复制进度
	fmt.Printf("正在从 %s 复制到 %s...\n", src, dst)
	
	// 执行复制，跳过 .git 目录和临时文件
	err := copy.Copy(src, dst, copy.Options{
		Skip: func(src string) (bool, error) {
			filename := filepath.Base(src)
			return strings.HasPrefix(filename, ".") || strings.HasPrefix(filename, "~"), nil
		},
		OnSymlink: func(src string) copy.SymlinkAction {
			return copy.Shallow // 对符号链接进行浅拷贝
		},
		OnDirExists: func(src, dest string) copy.DirExistsAction {
			return copy.Replace // 如果目录已存在则替换
		},
	})
	
	if err != nil {
		log.Fatalf("复制失败: %v", err)
	}
	
	fmt.Println("目录复制完成!")
	
	// 打印复制统计信息
	srcInfo, _ := os.Stat(src)
	dstInfo, _ := os.Stat(dst)
	fmt.Printf("源目录大小: %d 字节\n", srcInfo.Size())
	fmt.Printf("目标目录大小: %d 字节\n", dstInfo.Size())
}

注意事项

1. 复制大目录时可能需要较长时间，考虑添加进度条
2. 符号链接处理需要特别注意，默认是深拷贝
3. 权限问题可能导致复制失败，特别是在不同用户之间复制时
4. 跨文件系统复制时某些特性可能无法保留

替代方案

如果你不想使用第三方库，也可以使用标准库实现递归复制：

func CopyDir(src string, dst string) error {
    return filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }
        
        relPath, _ := filepath.Rel(src, path)
        dstPath := filepath.Join(dst, relPath)
        
        if info.IsDir() {
            return os.MkdirAll(dstPath, info.Mode())
        }
        
        data, err := os.ReadFile(path)
        if err != nil {
            return err
        }
        
        return os.WriteFile(dstPath, data, info.Mode())
    })
}

不过使用 copy 库更加简洁且功能更全面，推荐在项目中使用。