本文實(shí)例講述了GO語言實(shí)現(xiàn)列出目錄和遍歷目錄的方法。分享給大家供大家參考。具體如下：
GO語言獲取目錄列表用 ioutil.ReadDir()，遍歷目錄用 filepath.Walk()，使用方法課參考本文示例。
具體示例代碼如下：
import (
"fmt"
"io/ioutil"
"os"
"path/filepath"
"strings"
)
//獲取指定目錄下的所有文件，不進(jìn)入下一級(jí)目錄搜索，可以匹配后綴過濾。
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 { //忽略錯(cuò)誤
// 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)
}
希望本文所述對(duì)大家的GO語言程序設(shè)計(jì)有所幫助。
