Question

我需要编写一个函数，当给定文件夹的路径时扫描以该文件夹为根的文件。
然后我需要在该文件夹中显示目录结构。

我知道如何做2（我将使用jstree在浏览器中显示它）。

请帮助我完成第1部分，比如开始写这样的内容功能在go。

Answer 1

编辑：足够的人仍然会回答这个问题，我认为我会为Go1 API更新它。这是filepath.Walk()的一个工作示例。原文如下。

package main

import (
  "path/filepath"
  "os"
  "flag"
  "fmt"
)

func visit(path string, f os.FileInfo, err error) error {
  fmt.Printf("Visited: %s\n", path)
  return nil
} 


func main() {
  flag.Parse()
  root := flag.Arg(0)
  err := filepath.Walk(root, visit)
  fmt.Printf("filepath.Walk() returned %v\n", err)
}

请注意，filepath.Walk以递归方式遍历目录树。

这是一个示例运行：

$ mkdir -p dir1/dir2
$ touch dir1/file1 dir1/dir2/file2
$ go run walk.go dir1
Visited: dir1
Visited: dir1/dir2
Visited: dir1/dir2/file2
Visited: dir1/file1
filepath.Walk() returned <nil>

原始答案：步行文件路径的界面已于每周更新.2011-09-16，请参阅http://groups.google.com/group/golang-nuts/msg/e304dd9cf196a218。以下代码在不久的将来不适用于GO的发行版本。

标准库中实际上只有一个函数：filepath.Walk。

package main

import (
    "path/filepath"
    "os"
    "flag"
)

type visitor int

// THIS CODE NO LONGER WORKS, PLEASE SEE ABOVE
func (v visitor) VisitDir(path string, f *os.FileInfo) bool {
    println(path)
    return true
} 

func (v visitor) VisitFile(path string, f *os.FileInfo) {
    println(path)
}

func main() {
    root := flag.Arg(0)
    filepath.Walk(root, visitor(0), nil)
}

Answer 2

这是获取目录中文件的文件信息的方法。

package main

import (
    "fmt"
    "os"
    "path/filepath"
)

func main() {
    dirname := "." + string(filepath.Separator)
    d, err := os.Open(dirname)
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    defer d.Close()
    fi, err := d.Readdir(-1)
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    for _, fi := range fi {
        if fi.Mode().IsRegular() {
            fmt.Println(fi.Name(), fi.Size(), "bytes")
        }
    }
}

Answer 3

这是一个循环遍历所有文件和目录的示例。请注意，如果您想知道您要追加的路径是否是目录，请选中“f.IsDir（）”。

package main

import (
    "fmt"
    "os"
    "path/filepath"
)

func main() {
    searchDir := "c:/path/to/dir"

    fileList := []string{}
    err := filepath.Walk(searchDir, func(path string, f os.FileInfo, err error) error {
        fileList = append(fileList, path)
        return nil
    })

    for _, file := range fileList {
        fmt.Println(file)
    }
}

Answer 4

包github.com/kr/fs为Walker提供了一个非常有趣的API。

Answer 5

Go标准包ioutil已内置函数用于此案例场景，请参阅下面的示例

func searchFiles(dir string) { // dir is the parent directory you what to search
    files, err := ioutil.ReadDir(dir)
    if err != nil {
        log.Fatal(err)
    }

    for _, file := range files {
        fmt.Println(file.Name())
    }
}

Answer 6

请注意“Walk不遵循符号链接”，因此如果您要编写执行该功能的功能，我建议使用ioutil.ReadDir。我自己的基准测试显示它比filepath.Glob更快，内存密集更少。

此外，ioutil.ReadDir使用基本字符串比较（strA > strB）按basename对文件进行排序。作为一个devops人，我通常通过进行反向数值比较（例如，最新的构建）来对dir名称进行排序。如果这也是您的情况，那么最好直接致电os.ReadDir（ioutil.ReadDir正在调用此内容）并自行进行排序。

以下是具有数字排序的ReadDir部分的示例：

// ReadDirNumSort - Same as ioutil/ReadDir but uses returns a Numerically
// Sorted file list.
//
// Taken from https://golang.org/src/io/ioutil/ioutil.go
// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
//
// Modified Sort method to use Numerically sorted names instead.
// It also allows reverse sorting.
func ReadDirNumSort(dirname string, reverse bool) ([]os.FileInfo, error) {
    f, err := os.Open(dirname)
    if err != nil {
        return nil, err
    }
    list, err := f.Readdir(-1)
    f.Close()
    if err != nil {
        return nil, err
    }
    if reverse {
        sort.Sort(sort.Reverse(byName(list)))
    } else {
        sort.Sort(byName(list))
    }
    return list, nil
}

// byName implements sort.Interface.
type byName []os.FileInfo

func (f byName) Len() int      { return len(f) }
func (f byName) Swap(i, j int) { f[i], f[j] = f[j], f[i] }
func (f byName) Less(i, j int) bool {
    nai, err := strconv.Atoi(f[i].Name())
    if err != nil {
        return f[i].Name() < f[j].Name()
    }
    naj, err := strconv.Atoi(f[j].Name())
    if err != nil {
        return f[i].Name() < f[j].Name()
    }
    return nai < naj
}

Answer 7

您可能希望在这里进行函数管理，以便能够充分利用搜索功能

func visit(files *[]string) filepath.WalkFunc {
    return func (path string, info os.FileInfo, err error) error {
               // maybe do this in some if block
               *files = append(*files, path)
               return nil
           }
}

如何在Golang中执行文件系统扫描？

7 个答案: