golang中怎么利用遞歸構(gòu)建一個(gè)無(wú)限級(jí)樹(shù)狀目錄

golang中怎么利用遞歸構(gòu)建一個(gè)無(wú)限級(jí)樹(shù)狀目錄，很多新手對(duì)此不是很清楚，為了幫助大家解決這個(gè)難題，下面小編將為大家詳細(xì)講解，有這方面需求的人可以來(lái)學(xué)習(xí)下，希望你能有所收獲。

package main

import (
	"encoding/json"
	"fmt"
	"os"
	"path/filepath"
	"sort"
)

func main() {
	rootpath := "D:\\projects"

	root := FileNode{"projects", rootpath, []*FileNode{}}
	fileInfo, _ := os.Lstat(rootpath)

	walk(rootpath, fileInfo, &root)

	data, _ := json.Marshal(root)

	fmt.Printf("%s", data)
}

type FileNode struct {
	Name      string      `json:"name"`
	Path      string      `json:"path"`
	FileNodes []*FileNode `json:"children"`
}

func walk(path string, info os.FileInfo, node *FileNode) {
	// 列出當(dāng)前目錄下的所有目錄、文件
	files := listFiles(path)

	// 遍歷這些文件
	for _, filename := range files {
		// 拼接全路徑
		fpath := filepath.Join(path, filename)

		// 構(gòu)造文件結(jié)構(gòu)
		fio, _ := os.Lstat(fpath)

		// 將當(dāng)前文件作為子節(jié)點(diǎn)添加到目錄下
		child := FileNode{filename, fpath, []*FileNode{}}
		node.FileNodes = append(node.FileNodes, &child)

		// 如果遍歷的當(dāng)前文件是個(gè)目錄，則進(jìn)入該目錄進(jìn)行遞歸
		if fio.IsDir() {
			walk(fpath, fio, &child)
		}
	}

	return
}

func listFiles(dirname string) []string {
	f, _ := os.Open(dirname)

	names, _ := f.Readdirnames(-1)
	f.Close()

	sort.Strings(names)

	return names
}

看完上述內(nèi)容是否對(duì)您有幫助呢？如果還想對(duì)相關(guān)知識(shí)有進(jìn)一步的了解或閱讀更多相關(guān)文章，請(qǐng)關(guān)注億速云行業(yè)資訊頻道，感謝您對(duì)億速云的支持。