GO语言如何实现列出目录和遍历目录的方法

本文实例讲述了GO语言实现列出目录和遍历目录的方法。分享给大家供大家参考。具体如下：

GO语言获取目录列表用 ioutil.ReadDir()，遍历目录用 filepath.Walk()，使用方法课参考本文示例。

具体示例代码如下：

package main

import (

"fmt"

"io/ioutil"

"os"

"path/filepath"

"strings"

)
//获取指定目录下的所有文件，不进入下一级目录搜索，可以匹配后缀过滤。

func ListDir(dirPth string, suffix string) (files []string, err error) {

files = make([]string, 0, 10)
dir, err := ioutil.ReadDir(dirPth)

if err != nil {

return nil, err

}
PthSep := string(os.PathSeparator)

suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
for _, fi := range dir {

if fi.IsDir() { // 忽略目录

continue

}

if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) { //匹配文件

files = append(files, dirPth+PthSep+fi.Name())

}

}
return files, nil

}
//获取指定目录及所有子目录下的所有文件，可以匹配后缀过滤。

func WalkDir(dirPth, suffix string) (files []string, err error) {

files = make([]string, 0, 30)

suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
err = filepath.Walk(dirPth, func(filename string, fi os.FileInfo, err error) error { //遍历目录

//if err != nil { //忽略错误

//return err

//}
if fi.IsDir() { // 忽略目录

return nil

}
if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {

files = append(files, filename)

}
return nil

})
return files, err

}
func main() {

files, err := ListDir("D:\\Go", ".txt")

fmt.Println(files, err)
files, err = WalkDir("E:\\Study", ".pdf")

fmt.Println(files, err)

}

希望本文所述对大家的GO语言程序设计有所帮助。

到此这篇关于GO语言如何实现列出目录和遍历目录的方法就介绍到这了。坚强的意志与决心可以战胜一切困难，执着的信念与无畏的心灵才是最强大的武器。更多相关GO语言如何实现列出目录和遍历目录的方法内容请查看相关栏目，小编编辑不易，再次感谢大家的支持！