GO語言實現簡單的目錄複寫功能_Golang

本文執行個體講述了GO語言實現簡單的目錄複寫功能。分享給大家供大家參考。具體實現方法如下：

建立一個獨立的 goroutine 遍曆檔案，主進程負責寫入資料。程式會複製空目錄，也可以設定只複製以 ".xx" 結尾的檔案。
嚴格來說這不是複製檔案，而是寫入新檔案。因為這個程式是建立新檔案，然後寫入複製資料的。我們一般的 copy 命令是不會修改檔案的 ctime(change time) 狀態的。

代碼如下：

複製代碼 代碼如下:

// 一個簡單的目錄複寫程式：一個獨立的 goroutine 遍曆目錄，主進程負責將資料寫入新目錄。
// 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
}

//複製檔案資料
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) //遍曆完成，關閉通道
}

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

   if fi.IsDir() != f.IsDir { //檢查檔案名稱被目錄名佔用衝突
    log.Println("[E]", "filename conflict:", f.RelPath)

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

希望本文所述對大家的GO語言程式設計有所協助。