huangzj
379 lines
8.7 KiB
Go
379 lines
8.7 KiB
Go
/*
|
||
* @Author : huangzj
|
||
* @Time : 2020/4/26 14:56
|
||
* @Description： 文件工具,主要处理权限、文件创建等操作
|
||
*
|
||
*
|
||
* 打开文件的权限相关：
|
||
* os.O_RDONLY // 只读
|
||
* os.O_WRONLY // 只写
|
||
* os.O_RDWR // 读写
|
||
* os.O_APPEND // 往文件中添建（Append）
|
||
* os.O_CREATE // 如果文件不存在则先创建
|
||
* os.O_TRUNC // 文件打开时裁剪文件
|
||
* os.O_EXCL // 和O_CREATE一起使用，文件不能存在
|
||
* os.O_SYNC // 以同步I/O的方式打开
|
||
*
|
||
*/
|
||
|
||
package file
|
||
|
||
import (
|
||
err2 "Go-Tool/err"
|
||
"fmt"
|
||
"io/ioutil"
|
||
"os"
|
||
)
|
||
|
||
/*
|
||
* @param path 文件夹目录
|
||
* @return err 如果报错返回错误信息，否则返回nil
|
||
* @return bool 返回是否文件夹结果
|
||
* @return os.FileInfo 返回FileInfo对象，如果没有的话返回nil
|
||
* @description
|
||
*/
|
||
func IsDirExist(path string) (*err2.FileError, bool, os.FileInfo) {
|
||
dir, err := os.Stat(path)
|
||
//如果路仅错误直接返回错误
|
||
if err != nil {
|
||
return err2.ENewFileError(err), false, nil
|
||
}
|
||
//如果是文件夹直接返回
|
||
if dir.IsDir() {
|
||
return nil, true, dir
|
||
}
|
||
return err2.NewFileError("非文件夹"), false, nil
|
||
}
|
||
|
||
func IsFileExist(path string) (*err2.FileError, bool, os.FileInfo) {
|
||
file, err := os.Stat(path)
|
||
//如果路仅错误直接返回错误
|
||
if err != nil {
|
||
return err2.ENewFileError(err), false, nil
|
||
}
|
||
|
||
if file.IsDir() {
|
||
return err2.NewFileError("非文件"), false, nil
|
||
}
|
||
|
||
return nil, true, file
|
||
}
|
||
|
||
/*
|
||
* @param filepath 文件目录
|
||
* @return err 如果报错返回错误信息，否则返回nil
|
||
* @return bool 是否有读权限
|
||
* @return *File 返回文件对象，如果没有返回nil
|
||
* @description 这边需要注意的是，在打开文件的时候，如果报错，分两种情况，一个是没权限，一个是文件有问题
|
||
*/
|
||
func HasReadPermission(filePath string) (*err2.FileError, bool, *os.File) {
|
||
eErr, exist, _ := IsFileExist(filePath)
|
||
if eErr != nil || !exist {
|
||
return err2.ENewFileError(eErr), false, nil
|
||
}
|
||
file, err := os.OpenFile(filePath, os.O_RDONLY, 0666)
|
||
if err != nil {
|
||
//如果是没权限的错误返回
|
||
if os.IsPermission(err) {
|
||
return err2.ENewFileError(err), false, nil
|
||
}
|
||
//如果有权限但是发生错误的话，返回错误和有权限的标识
|
||
return err2.ENewFileError(err), true, nil
|
||
}
|
||
|
||
return nil, true, file
|
||
}
|
||
|
||
/*
|
||
* @param filepath 文件目录
|
||
* @return err 如果报错返回错误信息，否则返回nil
|
||
* @return bool 是否有写权限
|
||
* @return *File 返回文件对象，如果没有返回nil
|
||
* @description 这边需要注意的是，在打开文件的时候，如果报错，分两种情况，一个是没权限，一个是文件有问题
|
||
*/
|
||
func HasWritePermission(filePath string) (*err2.FileError, bool, *os.File) {
|
||
eErr, exist, _ := IsFileExist(filePath)
|
||
if eErr != nil || !exist {
|
||
return err2.ENewFileError(eErr), false, nil
|
||
}
|
||
file, err := os.OpenFile(filePath, os.O_WRONLY, 0666)
|
||
if err != nil {
|
||
//如果是没权限的错误返回
|
||
if os.IsPermission(err) {
|
||
return err2.ENewFileError(err), false, nil
|
||
}
|
||
//如果有权限但是发生错误的话，返回错误和有权限的标识
|
||
return err2.ENewFileError(err), true, nil
|
||
}
|
||
|
||
return nil, true, file
|
||
}
|
||
|
||
/*
|
||
* @param filepath 文件目录
|
||
* @return err 如果报错返回错误信息，否则返回nil
|
||
* @return bool 是否有写权限
|
||
* @return *File 返回文件对象，如果没有返回nil
|
||
* @description 判断对文件是否有读写权限.
|
||
*/
|
||
func HasReadWritePermission(filePath string) (*err2.FileError, bool, *os.File) {
|
||
rErr, hasR, rFile := HasReadPermission(filePath)
|
||
wErr, hasW, _ := HasWritePermission(filePath)
|
||
if hasR && hasW {
|
||
return nil, true, rFile
|
||
}
|
||
|
||
if rErr != nil {
|
||
return err2.ENewFileError(rErr), false, nil
|
||
}
|
||
|
||
return err2.ENewFileError(wErr), false, nil
|
||
}
|
||
|
||
/*
|
||
* @param filePath 文件路径
|
||
* @return FileError 文件错误
|
||
* @return File文件指针对象
|
||
* @description 创建文件，如果文件存在不进行覆盖
|
||
*/
|
||
func CreateFile(filePath string) (*err2.FileError, *os.File) {
|
||
file, err := os.Create(filePath)
|
||
if err != nil {
|
||
return err2.ENewFileError(err), nil
|
||
}
|
||
return nil, file
|
||
}
|
||
|
||
/*
|
||
* @param filePath 文件路径
|
||
* @return FileError 文件错误
|
||
* @return File文件指针对象
|
||
* @description 创建文件，如果文件存在进行覆盖
|
||
*/
|
||
func CreateOrCoverFile(filePath string) (*err2.FileError, *os.File) {
|
||
_, has, _ := IsFileExist(filePath)
|
||
|
||
if has {
|
||
rErr := os.Remove(filePath)
|
||
if rErr != nil {
|
||
return err2.ENewFileError(rErr), nil
|
||
}
|
||
return CreateFile(filePath)
|
||
}
|
||
return CreateFile(filePath)
|
||
}
|
||
|
||
/*
|
||
* @param filePath 文件路径
|
||
* @return File文件指针对象
|
||
* @description 获取文件夹下面的所有文件对象数组
|
||
*/
|
||
func GetAllFileFromDir(filePath string) []*os.File {
|
||
return getFiles(filePath)
|
||
}
|
||
|
||
func GetAllFileNameFromDir(filePath string) []string {
|
||
return getFileNames(filePath)
|
||
}
|
||
|
||
/*
|
||
* @param dir 文件夹地址
|
||
* @return 所有空文件夹数组
|
||
* @description 获取文件夹下面所有的空文件夹数组
|
||
*/
|
||
func GetAllEmptyDir(dir string) (*err2.FileError, []string) {
|
||
err, is, _ := IsDirExist(dir)
|
||
if err != nil {
|
||
return err2.ENewFileError(err), nil
|
||
}
|
||
if !is {
|
||
return err2.NewFileError("非文件夹"), nil
|
||
}
|
||
|
||
list := make([]string, 0)
|
||
errE, paths := GetAllDir(dir)
|
||
if errE != nil {
|
||
return errE, nil
|
||
}
|
||
for _, row := range paths {
|
||
err, is := isEmptyDir(row)
|
||
if err != nil {
|
||
return err, nil
|
||
}
|
||
if is {
|
||
list = append(list, row)
|
||
}
|
||
}
|
||
return nil, list
|
||
}
|
||
|
||
/*
|
||
* 获取所有非空文件夹的路径集合
|
||
*/
|
||
func GetAllNotEmptyDir(dir string) (*err2.FileError, []string) {
|
||
err, is, _ := IsDirExist(dir)
|
||
if err != nil {
|
||
return err2.ENewFileError(err), nil
|
||
}
|
||
if !is {
|
||
return err2.NewFileError("非文件夹"), nil
|
||
}
|
||
files := make([]string, 0)
|
||
err1, aFiles := GetAllDir(dir)
|
||
if err1 != nil {
|
||
return err1, nil
|
||
}
|
||
err3, eFiles := GetAllEmptyDir(dir)
|
||
if err3 != nil {
|
||
return err3, nil
|
||
}
|
||
for _, a := range aFiles {
|
||
isIn := false
|
||
for _, e := range eFiles {
|
||
if a == e {
|
||
isIn = true
|
||
break
|
||
}
|
||
}
|
||
if !isIn {
|
||
files = append(files, a)
|
||
}
|
||
}
|
||
return nil, files
|
||
}
|
||
|
||
/*
|
||
* 获取目录下的所有文件夹路径集合
|
||
*/
|
||
func GetAllDir(dir string) (*err2.FileError, []string) {
|
||
err, is, _ := IsDirExist(dir)
|
||
if err != nil {
|
||
return err2.ENewFileError(err), nil
|
||
}
|
||
if !is {
|
||
return err2.NewFileError("非文件夹"), nil
|
||
}
|
||
return nil, getAllDir(dir)
|
||
}
|
||
|
||
/*
|
||
* @param
|
||
* @return
|
||
* @description 删除文件夹下面所有的空文件夹
|
||
*/
|
||
func DeleteEmptyDir(filePath string) *err2.FileError {
|
||
err, is, _ := IsDirExist(filePath)
|
||
if err != nil {
|
||
return err2.ENewFileError(err)
|
||
}
|
||
if !is {
|
||
return err2.NewFileError("非文件夹")
|
||
}
|
||
|
||
//拿到所有的空文件夹
|
||
err, emptyDirList := GetAllEmptyDir(filePath)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
for _, row := range emptyDirList {
|
||
err := os.RemoveAll(row)
|
||
if err != nil {
|
||
return err2.ENewFileError(err)
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
func getAllDir(dir string) []string {
|
||
|
||
dirList := make([]string, 0)
|
||
info, err := ioutil.ReadDir(dir)
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
for _, f := range info {
|
||
if f.IsDir() {
|
||
list := getAllDir(fmt.Sprint(dir, "\\", f.Name()))
|
||
dirList = append(dirList, fmt.Sprint(dir, "\\", f.Name()))
|
||
if list != nil {
|
||
for _, l := range list {
|
||
dirList = append(dirList, l)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
return dirList
|
||
}
|
||
|
||
func isEmptyDir(dir string) (*err2.FileError, bool) {
|
||
_, err := ioutil.ReadDir(dir)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
file := GetAllFileFromDir(dir)
|
||
for _, f := range file {
|
||
info, err := f.Stat()
|
||
if err != nil {
|
||
return err2.ENewFileError(err), false
|
||
}
|
||
if !info.IsDir() {
|
||
return nil, false
|
||
}
|
||
}
|
||
|
||
return nil, true
|
||
}
|
||
|
||
func getFiles(filePath string) []*os.File {
|
||
fileList := make([]*os.File, 0)
|
||
info, err := ioutil.ReadDir(filePath)
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
|
||
for _, f := range info {
|
||
//处理文件夹的部分
|
||
if f.IsDir() {
|
||
list := getFiles(fmt.Sprint(filePath, "\\", f.Name()))
|
||
if list != nil {
|
||
for _, row := range list {
|
||
fileList = append(fileList, row)
|
||
}
|
||
}
|
||
} else {
|
||
//处理文件的部分 -- 使用读写方法打开文件
|
||
thisFile, _ := os.OpenFile(fmt.Sprint(filePath, "\\", f.Name()), os.O_RDWR, 0666)
|
||
fileList = append(fileList, thisFile)
|
||
}
|
||
}
|
||
|
||
return fileList
|
||
}
|
||
|
||
func getFileNames(filePath string) []string {
|
||
nameList := make([]string, 0)
|
||
info, err := ioutil.ReadDir(filePath)
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
|
||
for _, f := range info {
|
||
//处理文件夹的部分
|
||
if f.IsDir() {
|
||
list := getFileNames(fmt.Sprint(filePath, "\\", f.Name()))
|
||
if list != nil {
|
||
for _, row := range list {
|
||
nameList = append(nameList, row)
|
||
}
|
||
}
|
||
} else {
|
||
//处理文件的部分
|
||
nameList = append(nameList, fmt.Sprint(filePath, "\\", f.Name()))
|
||
}
|
||
}
|
||
|
||
return nameList
|
||
}
|