golang File 对象封装

今天拿 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

你可能感兴趣的:(golang File 对象封装)