代码示例:
package main
import (
"fmt"
"io/ioutil"
"os"
"strings"
)
//获取指定目录下的所有文件和目录
func GetFilesAndDirs(dirPth string) (files []string, dirs []string, err error) {
dir, err := ioutil.ReadDir(dirPth)
if err != nil {
return nil, nil, err
}
PthSep := string(os.PathSeparator)
//suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写
for _, fi := range dir {
if fi.IsDir() { // 目录, 递归遍历
dirs = append(dirs, dirPth+PthSep+fi.Name())
GetFilesAndDirs(dirPth + PthSep + fi.Name())
} else {
files = append(files, dirPth+PthSep+fi.Name())
// 过滤指定格式
// ok := strings.HasSuffix(fi.Name(), ".go")
// if ok {
// files = append(files, dirPth+PthSep+fi.Name())
// }
}
}
return files, dirs, nil
}
//获取指定目录下的所有文件,包含子目录下的文件
func GetAllFiles(dirPth string) (files []string, err error) {
var dirs []string
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() { // 目录, 递归遍历
dirs = append(dirs, dirPth+PthSep+fi.Name())
GetAllFiles(dirPth + PthSep + fi.Name())
} else {
// 过滤指定格式
ok := strings.HasSuffix(fi.Name(), ".go")
if ok {
files = append(files, dirPth+PthSep+fi.Name())
}
}
}
// 读取子目录下文件
for _, table := range dirs {
temp, _ := GetAllFiles(table)
for _, temp1 := range temp {
files = append(files, temp1)
}
}
return files, nil
}
func main() {
files, dirs, _ := GetFilesAndDirs("./simplemath")
for _, dir := range dirs {
fmt.Printf("获取的文件夹为[%s]\n", dir)
}
for _, table := range dirs {
temp, _, _ := GetFilesAndDirs(table)
for _, temp1 := range temp {
files = append(files, temp1)
}
}
for _, table1 := range files {
fmt.Printf("获取的文件为[%s]\n", table1)
}
fmt.Printf("=======================================\n")
xfiles, _ := GetAllFiles("./simplemath")
for _, file := range xfiles {
fmt.Printf("获取的文件为[%s]\n", file)
}
}
输出结果:
获取的文件夹为[./simplemath\mid]
获取的文件夹为[./simplemath\sub]
获取的文件为[./simplemath\add.go]
获取的文件为[./simplemath\mid\mid.go]
获取的文件为[./simplemath\sub\sub.go]
获取的文件为[./simplemath\add.go]
获取的文件为[./simplemath\mid\mid.go]
获取的文件为[./simplemath\sub\sub.go]