**操作示例:**
```text
./scan /Document/dir
```
**代码:**
```go
// 定义递归文件树结构体
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
}
```
有疑问加站长微信联系(非本文作者))