[[377341]]

前言

有時(shí)候,我看著我的目錄,或者我想找個(gè)文件時(shí),尤其是目錄很多時(shí),總是有點(diǎn)難受。

比如這個(gè)目錄,是我的一個(gè)教程目錄。

但是我并不能一目了然的知道每個(gè)文件夾下都有啥,并且每次找一個(gè)文件也很費(fèi)事。

所以,基于上述的需求,使用Go語(yǔ)言,終于擼了個(gè)目錄生成器,感覺(jué)還行。

實(shí)現(xiàn)的效果

通過(guò)編寫(xiě)的腳本,可以將目錄整合成.txt文件,并且下級(jí)目錄使用4個(gè)空格縮進(jìn)。

代碼

先別管怎么實(shí)現(xiàn)的,先看代碼。

package main 
 
import ( 
    "bytes" 
    "flag" 
    "fmt" 
    "io/ioutil" 
    "os" 
    "path/filepath" 
) 
 
func GenderNSymbol(char byte, count int) string { 
    symbolSince := bytes.Repeat([]byte{char}, count) 
    var symbol = string(symbolSince) 
    return symbol 
} 
 
// @title    GenderDirTree 
// @description   生成目錄樹(shù) 
// @param     path        string         "需要生成的目錄" 
// @param     count        int           "生成相同字符的個(gè)數(shù)" 
// @param     char        byte           "生成相同字符的字符" 
// @param     current_tier        int            "當(dāng)前層數(shù)" 
// @param     end_tier        int            "終止層數(shù)" 
// @param     fileObj        int            "文件對(duì)象" 
// @return    無(wú)        無(wú)         "無(wú)" 
func GenderDirTree(path string, count int, char byte, current_tier int, end_tier int, fileObj *os.File) { 
    if !(current_tier < end_tier) && end_tier != 0 { 
        return 
} 
    current_tier++ 
    files, err := ioutil.ReadDir(path) 
    if err != nil { 
        fmt.Println("錯(cuò)誤:目錄錯(cuò)誤") 
        return 
} 
    for _, file := range files { 
        var name = file.Name() 
        //生成指定數(shù)目的相同符號(hào) 
        var plac = GenderNSymbol(char, count) 
        //符號(hào)+目錄 
        var sname = fmt.Sprintf("%s%s\n", plac, name) 
        //輸出當(dāng)前目錄 
        fileObj.WriteString(sname) 
        //fmt.Println(sname) 
        //判斷是否為目錄,如果是,繼續(xù)下次遞歸 
        var isDir = file.IsDir() 
        if isDir { 
            //拼接傳入的目錄和循環(huán)的當(dāng)前目錄 
            var nerPaht = filepath.Join(path, name) 
            GenderDirTree(nerPaht, count+4, char, current_tier, end_tier, fileObj) 
        } 
} 
} 
 
func main() { 
    //終止層數(shù),0表示無(wú)限層,>0表示指定層數(shù) 
    var end_tier int 
    //輸入的目錄 
    //var path = `D:\0_教程\易錦教程` 
    var path string 
    flag.StringVar(&path, "path", "", "目錄") 
    flag.IntVar(&end_tier, "tier", 0, "終止層數(shù)") 
    flag.Parse() 
    //文件對(duì)象 
    var wDirPath = filepath.Join(path, "目錄.txt") 
    fileObj, _ := os.OpenFile(wDirPath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) 
    defer fileObj.Close() 
    //生成目錄樹(shù) 
    GenderDirTree(path, 0, ' ', 0, end_tier, fileObj) 
} 

去掉注釋,其實(shí)也就30行代碼,就實(shí)現(xiàn)了這個(gè)功能。

實(shí)現(xiàn)的功能

• 自定義要生成的目錄。
• 自定義生成目錄終止的層數(shù)。

打包

go build main.go 

-help

采用flag包動(dòng)態(tài)控制參數(shù),參數(shù)如下。

d:>main.exe -h 
Usage of main.exe: 
  -path string 
        目錄 
  -tier int 
        終止層數(shù) 

運(yùn)行

語(yǔ)法

main.exe -path <要生成的目錄> -tier <終止層級(jí)> 
注意:-tier,終止層數(shù),0表示無(wú)限層,>0表示指定層數(shù),默認(rèn)為0 

示例

此處-tier指定的是0,表示無(wú)限制生成,如果指定是1,就表示生成一層。

結(jié)果

會(huì)在要生成的目錄下,多出一個(gè)目錄.txt。

打開(kāi)如下

主要代碼理解

flag

fileObj

判斷是否為目錄,遞歸

總結(jié)

本次更像是日常使用中,為了解決實(shí)際問(wèn)題的腳本,其實(shí)腳本使用Python更好。

但是Go有一個(gè)更大的優(yōu)勢(shì)就是可以編譯為exe。

這次的難點(diǎn)主要是理解遞歸,自己調(diào)自己!