本文實(shí)例講述了GO語(yǔ)言實(shí)現(xiàn)簡(jiǎn)單的目錄復(fù)制功能。分享給大家供大家參考。具體實(shí)現(xiàn)方法如下：
創(chuàng)建一個(gè)獨(dú)立的 goroutine 遍歷文件，主進(jìn)程負(fù)責(zé)寫(xiě)入數(shù)據(jù)。程序會(huì)復(fù)制空目錄，也可以設(shè)置只復(fù)制以 ".xx" 結(jié)尾的文件。
嚴(yán)格來(lái)說(shuō)這不是復(fù)制文件，而是寫(xiě)入新文件。因?yàn)檫@個(gè)程序是創(chuàng)建新文件，然后寫(xiě)入復(fù)制數(shù)據(jù)的。我們一般的 copy 命令是不會(huì)修改文件的 ctime(change time) 狀態(tài)的。
// 一個(gè)簡(jiǎn)單的目錄復(fù)制程序：一個(gè)獨(dú)立的 goroutine 遍歷目錄，主進(jìn)程負(fù)責(zé)將數(shù)據(jù)寫(xiě)入新目錄。
// 2014-11-02 Bing.L
package main
import (
"io"
"log"
"os"
"path/filepath"
"strings"
)
type FileInfo struct {
RelPath string
Size int64
IsDir bool
Handle *os.File
}
//復(fù)制文件數(shù)據(jù)
func ioCopy(srcHandle *os.File, dstPth string) (err error) {
dstHandle, err := os.OpenFile(dstPth, os.O_CREATE|os.O_WRONLY, os.ModePerm)
if err != nil {
return err
}
defer srcHandle.Close()
defer dstHandle.Close()
_, err = io.Copy(dstHandle, srcHandle)
return err
}
//遍歷目錄，將文件信息傳入通道
func WalkFiles(srcDir, suffix string, c chan- *FileInfo) {
suffix = strings.ToUpper(suffix)
filepath.Walk(srcDir, func(f string, fi os.FileInfo, err error) error { //遍歷目錄
if err != nil {
log.Println("[E]", err)
}
fileInfo := FileInfo{}
if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) { //匹配文件
if fh, err := os.OpenFile(f, os.O_RDONLY, os.ModePerm); err != nil {
log.Println("[E]", err)
} else {
fileInfo.Handle = fh
fileInfo.RelPath, _ = filepath.Rel(srcDir, f) //相對(duì)路徑
fileInfo.Size = fi.Size()
fileInfo.IsDir = fi.IsDir()
}
c - fileInfo
}
})
close(c) //遍歷完成，關(guān)閉通道
}
//寫(xiě)目標(biāo)文件
func WriteFiles(dstDir string, c -chan *FileInfo) {
if err := os.Chdir(dstDir); err != nil { //切換工作路徑
log.Fatalln("[F]", err)
}
for f := range c {
if fi, err := os.Stat(f.RelPath); os.IsNotExist(err) { //目標(biāo)不存在
if f.IsDir {
if err := os.MkdirAll(f.RelPath, os.ModeDir); err != nil {
log.Println("[E]", err)
}
} else {
if err := ioCopy(f.Handle, f.RelPath); err != nil {
log.Println("[E]", err)
} else {
log.Println("[I] CP:", f.RelPath)
}
}
} else if !f.IsDir { //目標(biāo)存在，而且源不是一個(gè)目錄
if fi.IsDir() != f.IsDir { //檢查文件名被目錄名占用沖突
log.Println("[E]", "filename conflict:", f.RelPath)
} else if fi.Size() != f.Size { //源和目標(biāo)的大小不一致時(shí)才重寫(xiě)
if err := ioCopy(f.Handle, f.RelPath); err != nil {
log.Println("[E]", err)
} else {
log.Println("[I] CP:", f.RelPath)
}
}
}
}
}
func main() {
files_ch := make(chan *FileInfo, 100)
go WalkFiles("E:\\study", ".doc", files_ch) //在一個(gè)獨(dú)立的 goroutine 中遍歷文件
WriteFiles("E:\\study.bak", files_ch)
}
