今天拿 Go 完成一个小操作,把一个多层的目录下的 jpg 文件全部检索出来,但是发现 go 原生的文件 api 做起来很麻烦
go 的文件可能需要用到以下的包
- io
- io/ioutils
- os
- path
- path/filepath
而且go 很蛋疼的地方在于,没有一个结构体用来统一的描述 文件/文件夹,所以当想要完成 “读取这个文件夹下的所有文件”,甚至递归遍历所有文件这样的逻辑的时候,会特别无措
回想起之前使用java 这种面向对象的语言的时候,只需要通过一个 File 类就可以实现这个逻辑了
所以决定自己封装一个 go 的 File 类(File 结构体)
封装之后,使用一个 File 结构体来统一描述文件和文件夹,上面的操作只需要使用以下代码就可以完成了
package main
import (
"fmt"
"github.com/Ericwyn/GoTools/file"
)
var jpgArr = make([]file.File, 0)
func main() {
// 使用 OpenFile 创建一个 File 对象
rootDir := file.OpenFile("./photoDir")
// 调用递归函数
searchAllJpg(rootDir)
// 打印所有的 jpg 文件的绝对路径
for _,jpgFile := range jpgArr {
fmt.Println(jpgFile.AbsPath())
}
}
func searchAllJpg(rootDir file.File) {
// 获取 rootDir 下的全部文件和文件夹
childrenFiles := rootDir.Children()
for _,fileTemp := range childrenFiles {
if fileTemp.IsFile() && fileTemp.Ext() == ".jpg" {
// 如果这个子 File 是一个文件对象,并且后缀名是 .jpg
jpgArr = append(jpgArr, fileTemp)
} else if fileTemp.IsDir() {
// 如果这个子 File 是一个文件夹,那么递归遍历
searchAllJpg(fileTemp)
}
}
}
对比起 Go 原生的 io API,真的是简单方便又养眼!
该工具主要有以下的功能
- 封装一个一个 struct 用来统一描述 文件/文件夹
- 获取路径/绝对路径/父文件夹路径
- 获取/文件(夹)名/后缀名
- 文件(夹)的创建/重命名/移动/删除
- 文件的读取/文本文件的单行读取
- 文件的写入
- 文件(夹)的存在判断
完整代码在: https://github.com/Ericwyn/GoTools/tree/master/file
有疑问加站长微信联系(非本文作者)