手把手教你用Go語言開發(fā)一款簡易目錄生成器

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   生成目錄樹// @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    無        無         "無"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表示無限層,>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()    //生成目錄樹    GenderDirTree(path, 0, ' ', 0, end_tier, fileObj)}

go build main.go

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

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