我想在Go中編寫一個tar_gz工具。輸入就像linux命令：$tar czvf targetFileName inputDirectoryPath假設我的inputDirectory結構如下：test [dir]-- 0.txt -- 1 [sub dir] -- 1.txt例如：使用命令：$tar czvf test.tar.gz test/我們可以tar和gzip整個測試目錄。我的問題是我可以編寫tar和gz路由來遞歸地迭代測試目錄中的所有文件，并將該文件寫入test.tar.gz文件。但是我不知道如何將目錄寫入test.tar.gz。運行程序后，test.tar.gz文件中的結構為：0.txt1.txt誰能告訴我如何將目錄遞歸寫入輸出tar.gz文件。非常感謝。 package main import ( "fmt" "os" "io" "log" "strings" "archive/tar" "compress/gzip" ) func handleError( _e error ) { if _e != nil { log.Fatal( _e ) } } func TarGzWrite( _path string, tw *tar.Writer, fi os.FileInfo ) { fr, err := os.Open( _path ) handleError( err ) defer fr.Close() h := new( tar.Header ) h.Name = fi.Name() h.Size = fi.Size() h.Mode = int64( fi.Mode() ) h.ModTime = fi.ModTime() err = tw.WriteHeader( h ) handleError( err ) _, err = io.Copy( tw, fr ) handleError( err ) } func IterDirectory( dirPath string, tw *tar.Writer ) { dir, err := os.Open( dirPath ) handleError( err ) defer dir.Close() fis, err := dir.Readdir( 0 ) handleError( err ) for _, fi := range fis { curPath := dirPath + "/" + fi.Name() if fi.IsDir() { //TarGzWrite( curPath, tw, fi ) IterDirectory( curPath, tw ) } else { fmt.Printf( "adding... %s\n", curPath ) TarGzWrite( curPath, tw, fi ) } } } func TarGz( outFilePath string, inPath string ) { // file write fw, err := os.Create( outFilePath ) handleError( err ) defer fw.Close() // gzip write gw := gzip.NewWriter( fw ) defer gw.Close()
2 回答
森林海
TA貢獻2011條經驗 獲得超2個贊
一種替代方法是使用內置的filepath.Walk函數
// root_directory has been set further up
walkFn := func(path string, info os.FileInfo, err error) error {
if info.Mode().IsDir() {
return nil
}
// Because of scoping we can reference the external root_directory variable
new_path := path[len(root_directory):]
if len(new_path) == 0 {
return nil
}
fr, err := os.Open(path)
if err != nil {
return err
}
defer fr.Close()
if h, err := tar.FileInfoHeader(info, new_path); err != nil {
log.Fatalln(err)
} else {
h.Name = new_path
if err = tw.WriteHeader(h); err != nil {
log.Fatalln(err)
}
}
if length, err := io.Copy( tw, fr ); err != nil {
log.Fatalln(err)
} else {
fmt.Println(length)
}
return nil
}
if err = filepath.Walk(root_directory, walkFn); err != nil {
return err
}
- 2 回答
- 0 關注
- 264 瀏覽
添加回答
舉報
0/150
提交
取消