快捷導(dǎo)航

GO語言實(shí)現(xiàn)簡單的目錄復(fù)制功能

更新時間：2014年12月03日 15:26:54 投稿：shichen2014

這篇文章主要介紹了GO語言實(shí)現(xiàn)簡單的目錄復(fù)制功能,通過新建及復(fù)制內(nèi)容等操作最終實(shí)現(xiàn)復(fù)制目錄的功能效果,具有一定的參考借鑒價值,需要的朋友可以參考下

本文實(shí)例講述了GO語言實(shí)現(xiàn)簡單的目錄復(fù)制功能。分享給大家供大家參考。具體實(shí)現(xiàn)方法如下：

創(chuàng)建一個獨(dú)立的 goroutine 遍歷文件，主進(jìn)程負(fù)責(zé)寫入數(shù)據(jù)。程序會復(fù)制空目錄，也可以設(shè)置只復(fù)制以 ".xx" 結(jié)尾的文件。
嚴(yán)格來說這不是復(fù)制文件，而是寫入新文件。因?yàn)檫@個程序是創(chuàng)建新文件，然后寫入復(fù)制數(shù)據(jù)的。我們一般的 copy 命令是不會修改文件的 ctime(change time) 狀態(tài)的。

代碼如下：

復(fù)制代碼代碼如下:

// 一個簡單的目錄復(fù)制程序：一個獨(dú)立的 goroutine 遍歷目錄，主進(jìn)程負(fù)責(zé)將數(shù)據(jù)寫入新目錄。
// 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) //相對路徑
    fileInfo.Size = fi.Size()
    fileInfo.IsDir = fi.IsDir()
   }

c <- fileInfo
}
})
close(c) //遍歷完成，關(guān)閉通道
}

//寫目標(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)存在，而且源不是一個目錄

if fi.IsDir() != f.IsDir { //檢查文件名被目錄名占用沖突
log.Println("[E]", "filename conflict:", f.RelPath)

   } else if fi.Size() != f.Size { //源和目標(biāo)的大小不一致時才重寫
    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) //在一個獨(dú)立的 goroutine 中遍歷文件
WriteFiles("E:\\study.bak", files_ch)
}

希望本文所述對大家的GO語言程序設(shè)計(jì)有所幫助。

您可能感興趣的文章: