package main
import (
"fmt"
"io/ioutil"
"os"
"strings"
)
var fileList []string
func getExtFile(path string, ext string ) ([]string, error) {
rd, err := ioutil.ReadDir(path)
if err !=nil {
fmt.Println("read dir fail: ", err)
return fileList, err
}
for _, file := range rd {
filePath := path + string(os.PathSeparator) + file.Name()
if file.IsDir() {
_, _ = getExtFile(filePath, ext)
} else {
if ok:= strings.HasSuffix(filePath, ext); ok {
fileList = append(fileList, filePath)
}
}
}
return fileList,err
}
func main() {
/*
实现遍历目录与子目录,抓取.txt 文件
*/
f,_ := getExtFile("d://tmp/", ".yaml")
for _,val := range f {
fmt.Println(val)
}
}