GO语言学习笔记（三） - 递归查找目录及子目录下的文件

递归查找目录及子目录下的文件
递归查找文件夹及子文件夹下的文件

代码

package main import ( "fmt" "io/ioutil" "os" "strings" ) // 查找目录及子目录下的文件 // 添加参数判断 // path : 目标目录 // files : 容纳所有文件路径的结果数组 // targetType : 目标文件类型 // ignoreFile : 忽略文件（文件名，包括扩展名） // ignorePath : 忽略目录 // ignoreType : 忽略文件类型 func GetAllFile(path string, files *[]string, targetType *[]string, ignoreFile *[]string, ignorePath *[]string, ignoreType *[]string) (err error) { if !isAllEmpty(targetType) && !isAllEmpty(ignoreType) { fmt.Printf("WARNGING: 目标文件类型已指定, 忽略文件类型无须指定。后续处理中忽略文件类型作为空处理\n") } err = getAllFileRecursion(path, files, targetType, ignoreFile, ignorePath, ignoreType) return err; } // 递归查找目录及子目录下的文件 // path : 目标目录 // files : 容纳所有文件路径的结果数组 // targetType : 目标文件类型 // ignoreFile : 忽略文件（文件名，包括扩展名） // ignorePath : 忽略目录 // ignoreType : 忽略文件类型 func getAllFileRecursion(path string, files *[]string, targetType *[]string, ignoreFile *[]string, ignorePath *[]string, ignoreType *[]string) (err error) { l, err := ioutil.ReadDir(path) if err != nil { return err } separator := string(os.PathSeparator) for _, f := range l { tmp := string(path + separator + f.Name()) if f.IsDir() { // 过滤被忽略的文件夹（文件夹名完全相同） if !isInArray(ignorePath, f.Name()) { err = getAllFileRecursion(tmp, files, targetType, ignoreFile, ignorePath, ignoreType) if err != nil { return err } } } else { // 目标文件类型被指定 if !isAllEmpty(targetType) { // 属于目标文件类型 if isInSuffix(targetType, f.Name()) { // 忽略文件为空 或者 目标文件中不含有指定忽略文件 if isAllEmpty(ignoreFile) || !isInArray(ignoreFile, f.Name()) { *files = append(*files, tmp) } } } else { // 目标文件类型为空 // 忽略文件类型被指定 if !isAllEmpty(ignoreType) { // 不属于忽略文件类型 if !isInSuffix(ignoreType, f.Name()) { // 忽略文件为空 或者 目标文件中不含有指定忽略文件 if isAllEmpty(ignoreFile) || !isInArray(ignoreFile, f.Name()) { *files = append(*files, tmp) } } } else { // 忽略文件类型为空 // 忽略文件为空 或者 目标文件中不含有指定忽略文件 if isAllEmpty(ignoreFile) || !isInArray(ignoreFile, f.Name()) { *files = append(*files, tmp) } } } } } return nil } // 判断目标字符串是否是在数组中 func isInArray(list *[]string, s string) (isIn bool) { if len(*list) == 0 { return false } isIn = false for _, f := range *list { if f == s { isIn = true break } } return isIn } // 判断目标字符串的末尾是否含有数组中指定的字符串 func isInSuffix(list *[]string, s string) (isIn bool) { isIn = false for _, f := range *list { if strings.TrimSpace(f) != "" && strings.HasSuffix(s, f) { isIn = true break } } return isIn } // 判断数组各元素是否是空字符串或空格 func isAllEmpty(list *[]string) (isEmpty bool) { if len(*list) == 0 { return true } isEmpty = true for _, f := range *list { if strings.TrimSpace(f) != "" { isEmpty = false break } } return isEmpty } func main() { fmt.Printf("Info: Start\n") targetType := []string{".js", "", ""} ignoreFile := []string{"index.js"} ignorePath := []string{".git"} ignoreType := []string{".gitignore", ".exe"} var files []string var path = "C:\\Users\\Administrator\\go\\src\\github.com\\bettersun\\hellogo" err := GetAllFile(path, &files, &targetType, &ignoreFile, &ignorePath, &ignoreType) if err != nil { fmt.Printf(err.Error() + "\n") } fmt.Printf("文件名（全路径）列表:\n") for _, file := range files { fmt.Printf(" [%s]\n", file) } fmt.Printf("Info: End\n") }