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

423 阅读1分钟

操作示例:

./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
}