[Go] 递归获取目录下的文件

操作示例：

./scan /Document/dir

代码：

// 定义递归文件树结构体
type treeList struct {
    Path string `json:"path"`
    File     []string  `json:"file"`
    Children []*treeList `json:"children"`
}

func main() {
    // 获取命令行输入的参数
    var dir string
    arg_num := len(os.Args)
    if arg_num < 2 {
        fmt.Println("Waring:请输入路径的名称")
        return
    }
    dir = os.Args[1] + "/"

    // 初始化tree结构体
    treelist := new(treeList)

    // 递归文件目录下的所有文件
    recursion(dir, treelist)

    // 将递归结构体转为json
    bytes, _ := json.Marshal(treelist)
    // fmt.Printf("%s",bytes)
    fmt.Println(string(bytes))
}

// 递归函数
func recursion(dir string, treelist *treeList) *treeList {
    // 读取文件夹下的所有文件
    files, err := ioutil.ReadDir(dir)
    if err != nil {
        fmt.Printf("Dir(%s) scan error,err:%v\n", dir, err)
    }

    // 定义存放文件的切片
    var fileSlice []string

    // 得到文件名次或文件夹名次
    for _, f := range files {
        filename := f.Name()
        if f.IsDir() { // 如果是文件夹，则继续递归
            fpath := dir + filename + "/"
            children := recursion(fpath,new(treeList))
            treelist.Children = append(treelist.Children,children)
        } else{ // 如果是文件
            fileSlice = append(fileSlice,filename)
        }
    }

    // 写入查找路径路径
    treelist.Path = dir

    // 当前目录的文件给到 treelist.File
    treelist.File = fileSlice

    return treelist
}