一、背景介绍
在日常工作中,经常需要将一个目录下的文件复制到另一个目录或者将一个文件复制到另一个文件。GolangCopy就是基于Go语言实现的文件复制工具。
二、主要功能
GolangCopy实现了以下主要功能:
- 复制单个文件
- 复制整个目录
- 支持指定目标目录
- 支持文件覆盖
三、代码实现
以下为GolangCopy的主要代码实现,核心的文件复制函数是CopyFile,代码解释在注释中:
package main
import (
"fmt"
"io"
"io/ioutil"
"os"
"path/filepath"
)
// 复制单个文件
func CopyFile(src, dst string) (err error) {
// 打开源文件
fi, err := os.Open(src)
if err != nil {
return
}
defer fi.Close()
// 创建目标文件
fo, err := os.Create(dst)
if err != nil {
return
}
defer func() {
if errFo := fo.Close(); errFo != nil {
err = errFo
}
}()
// 复制文件内容
_, err = io.Copy(fo, fi)
if err != nil {
return
}
// 获取源文件信息
fiStat, err := fi.Stat()
if err != nil {
return
}
// 修改目标文件的权限和时间戳
err = os.Chmod(dst, fiStat.Mode())
if err != nil {
return
}
err = os.Chtimes(dst, fiStat.ModTime(), fiStat.ModTime())
if err != nil {
return
}
return
}
// 复制整个目录
func CopyDir(src, dst string) (err error) {
// 获取源目录信息
srcInfo, err := os.Stat(src)
if err != nil {
return
}
// 创建目标目录
err = os.MkdirAll(dst, srcInfo.Mode())
if err != nil {
return
}
// 获取源目录下的文件列表
files, err := ioutil.ReadDir(src)
if err != nil {
return
}
// 循环复制文件
for _, file := range files {
srcFile := filepath.Join(src, file.Name())
dstFile := filepath.Join(dst, file.Name())
if file.IsDir() {
// 递归复制子目录
err = CopyDir(srcFile, dstFile)
if err != nil {
return
}
} else {
// 复制文件
err = CopyFile(srcFile, dstFile)
if err != nil {
return
}
}
}
return
}
func main() {
// 复制单个文件
err := CopyFile("./src/test.txt", "./dst/test.txt")
if err != nil {
fmt.Println(err)
return
}
// 复制整个目录
err = CopyDir("./src", "./dst")
if err != nil {
fmt.Println(err)
return
}
}
四、总结
通过简单的代码实现,我们实现了一个基于Go语言的文件复制工具GolangCopy。这个工具可以帮助我们快速、高效地将文件复制到指定目录中。在实际应用中,我们可以根据具体需求对代码进行调整,以满足不同场景的需求。