手把手教你用Go语言开发一款简易目录生成器

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           "生成相同字符的个数"// @param     char        byte           "生成相同字符的字符"// @param     current_tier        int            "当前层数"// @param     end_tier        int            "终止层数"// @param     fileObj        int            "文件对象"// @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("错误:目录错误")        return}    for _, file := range files {        var name = file.Name()        //生成指定数目的相同符号        var plac = GenderNSymbol(char, count)        //符号+目录        var sname = fmt.Sprintf("%s%s\n", plac, name)        //输出当前目录        fileObj.WriteString(sname)        //fmt.Println(sname)        //判断是否为目录,如果是,继续下次递归        var isDir = file.IsDir()        if isDir {            //拼接传入的目录和循环的当前目录            var nerPaht = filepath.Join(path, name)            GenderDirTree(nerPaht, count+4, char, current_tier, end_tier, fileObj)        }}}
func main() {    //终止层数,0表示无限层,>0表示指定层数    var end_tier int    //输入的目录    //var path = `D:\0_教程\易锦教程`    var path string    flag.StringVar(&path, "path", "", "目录")    flag.IntVar(&end_tier, "tier", 0, "终止层数")    flag.Parse()    //文件对象    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        终止层数

main.exe -path <要生成的目录> -tier <终止层级>注意:-tier,终止层数,0表示无限层,>0表示指定层数,默认为0