浅析go语言怎么提取目录层次结构

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《浅析go语言怎么提取目录层次结构》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到golang等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

问题内容

我正在尝试将文件夹的目录层次结构提取到 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 中将其转换为树结构？ 有没有更简单的方法来做到这一点？

正确答案

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 %vn", 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("%vn", rootFolder)
}

终于介绍完啦！小伙伴们，这篇关于《浅析go语言怎么提取目录层次结构》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！