使用go语言提取目录层次结构

Question

我正在尝试使用go语言将文件夹的目录层次结构提取到数据结构中.filepath.Walk似乎是要走的路,但我到目前为止所能做的就是打印文件和文件夹的名称.这是我正在使用的:

func main() {
    visit := func(path string, info os.FileInfo, err error) error {
        if info.IsDir() {
            fmt.Println("dir:  ", path)
        } else {
            fmt.Println("file: ", path)
        }
        return nil
    }

    err := filepath.Walk("./", visit)
    if err != nil {
        log.Fatal(err)
    }
}

这会打印文件夹的名称,如:

dir:   folder1
file:  folder1/file1.txt
file:  folder1/file2.txt
file:  folder1/file3.txt
file:  folder1/file4.txt
dir:   folder1/folder2
file:  folder1/folder2/file5.txt
file:  folder1/folder2/file6.txt
file:  folder1/folder2/file7.txt
file:  folder1/folder2/file8.txt
file:  folder1/folder2/file9.txt

对于树结构我想过使用类似的东西:

type File struct {
    Name string
    Content string
}

type Folder struct {
    Name    string
    Files   []File
    Folders []Folder
}

但当然欢迎任何建议.

如何在go中将其转换为树结构？有更简单的方法吗？

Answer 1

AFAIK 在 Go 标准库中没有现成的东西。

树结构非常适合递归方法。我在您的文件和文件夹类型上定义了方法addFile。addFolder从根文件夹开始，您可以在 Walk 中调用这些方法。如果您得到 a/b/c，我们将致电root.addFile(a, b, c), a.addFile(b, c), b.addFile(c)。

我还将Folder.Folders更改为映射，因为filepath.Walk总是为我们提供完整路径，因此我们可以拆分它们并在文件夹映射中查找它们的组件。

这是一些快速而肮脏的代码，可能有错误并且没有进行完整的错误检查。它仅适用于当前目录，但这应该很容易修复。

我还在Folder 上添加了一个String() 方法，该方法可以被编译器识别，并将在打印该类型的实例时使用。

package main

import (
    "log"
    "os"
    "path/filepath"
    "strings"
)

type File struct {
    Name string
}

type Folder struct {
    Name    string
    Files   []File
    Folders map[string]*Folder
}

func newFolder(name string) *Folder {
    return &Folder{name, []File{}, make(map[string]*Folder)}
}

func (f *Folder) getFolder(name string) *Folder {
    if nextF, ok := f.Folders[name]; ok {
        return nextF
    } else {
        log.Fatalf("Expected nested folder %v in %v\n", name, f.Name)
    }
    return &Folder{} // cannot happen
}

func (f *Folder) addFolder(path []string) {
    for i, segment := range path {
        if i == len(path)-1 { // last segment == new folder
            f.Folders[segment] = newFolder(segment)
        } else {
            f.getFolder(segment).addFolder(path[1:])
        }
    }
}

func (f *Folder) addFile(path []string) {
    for i, segment := range path {
        if i == len(path)-1 { // last segment == file
            f.Files = append(f.Files, File{segment})
        } else {
            f.getFolder(segment).addFile(path[1:])
            return
        }
    }
}

func (f *Folder) String() string {
    var str string
    for _, file := range f.Files {
        str += f.Name + string(filepath.Separator) + file.Name + "\n"
    }
    for _, folder := range f.Folders {
        str += folder.String()
    }
    return str
}

func main() {
    startPath := "."
    rootFolder := newFolder(startPath)

    visit := func(path string, info os.FileInfo, err error) error {
        segments := strings.Split(path, string(filepath.Separator))
        if info.IsDir() {
            if path != startPath {
                rootFolder.addFolder(segments)
            }
        } else {
            rootFolder.addFile(segments)
        }
        return nil
    }

    err := filepath.Walk(startPath, visit)
    if err != nil {
        log.Fatal(err)
    }

    log.Printf("%v\n", rootFolder)
}