Golang实现递归复制文件夹

代码

go 复制代码
package zdpgo_file

import (
	"errors"
	"os"
	"path/filepath"
	"strings"
)

// CopyDir 复制文件夹
// @param srcPath 源文件夹
// @param desPath 目标文件夹
// @return error 错误信息
func CopyDir(srcPath, desPath string) error {
	// 检查目录是否正确
	if !IsDir(srcPath) {
		return errors.New("要复制的文件夹不存在")
	}

	// 源文件夹和目标文件夹不能相同
	if strings.TrimSpace(srcPath) == strings.TrimSpace(desPath) {
		return errors.New("源文件夹和目标文件夹不能相同")
	}

	// 创建目标文件夹
	if !IsDir(desPath) {
		err := AddDir(desPath)
		if err != nil {
			return err
		}
	}

	// 路径规范
	srcPath = strings.Replace(srcPath, "\\", "/", -1)
	desPath = strings.Replace(desPath, "\\", "/", -1)

	// 遍历源文件夹
	err := filepath.Walk(srcPath, func(path string, f os.FileInfo, err error) error {
		if f == nil {
			return err
		}

		// 复制目录是将源目录中的子目录复制到目标路径中,不包含源目录本身
		if path == srcPath {
			return nil
		}

		// 生成新路径
		destNewPath := desPath + "/" + path
		destNewPath = strings.Replace(destNewPath, "\\", "/", -1)

		// 如果发生了递归,则退出
		if strings.Count(destNewPath, desPath) > 1 {
			return nil
		}

		// 如果是文件:复制文件
		// 如果是文件夹:不存在就创建新文件夹
		if !f.IsDir() {
			err = CopyFile(path, destNewPath)
			if err != nil {
				return err
			}
		} else {
			if !IsExist(destNewPath) {
				return AddDir(destNewPath)
			}
		}

		return nil
	})

	// 返回
	return err
}

测试

go 复制代码
package zdpgo_file

import "testing"

func TestCopyDir(t *testing.T) {
	if err := AddDir("test"); err != nil {
		t.Fatal(err)
	}

	data := []struct {
		Src string
		Dst string
	}{
		{".", "test/test2"},
		{"test/test2", "test/test1"},
	}

	for _, tt := range data {
		v := CopyDir(tt.Src, tt.Dst)
		if v != nil {
			t.Error(v)
		}
	}
}

测试结果

相关推荐
sniper_fandc1 小时前
SpringBoot系列—入门
java·spring boot·后端
好开心啊没烦恼1 小时前
Python 数据分析:计算,分组统计1,df.groupby()。听故事学知识点怎么这么容易?
开发语言·python·数据挖掘·数据分析·pandas
lljss20202 小时前
Python11中创建虚拟环境、安装 TensorFlow
开发语言·python·tensorflow
Python×CATIA工业智造5 小时前
Frida RPC高级应用:动态模拟执行Android so文件实战指南
开发语言·python·pycharm
我叫小白菜6 小时前
【Java_EE】单例模式、阻塞队列、线程池、定时器
java·开发语言
狐凄6 小时前
Python实例题:基于 Python 的简单聊天机器人
开发语言·python
weixin_446122467 小时前
JAVA内存区域划分
java·开发语言·redis
悦悦子a啊7 小时前
Python之--基本知识
开发语言·前端·python
Piper蛋窝7 小时前
深入 Go 语言垃圾回收:从原理到内建类型 Slice、Map 的陷阱以及为何需要 strings.Builder
后端·go
QuantumStack8 小时前
【C++ 真题】P1104 生日
开发语言·c++·算法