【golang】常用的文件操作

golang常用的文件操作（错误处理没写，根据需求完善）

package main

import (
	"io"
	"os"
	"path/filepath"
	"strings"
)

// 判断是否是目录
func IsDir(name string) bool {
	if info, err := os.Stat(name); err == nil {
		return info.IsDir()
	}
	return false
}

// 判断是否是文件
func IsFile(name string) bool {
	if info, err := os.Stat(name); err == nil {
		return !info.IsDir()
	}
	return false
}

// 判断文件是否存在
func FileIsExists(path string) bool {
	_, err := os.Stat(path)
	return err == nil
}

// 获取文件目录
func GetFileDir(path string) string {
	return filepath.Dir(path)
}

// 获取文件绝对路径
func GetFileAbsDir(path string) string {
	path, _ = filepath.Abs(path)
	return filepath.Dir(path)
}

// 获取文件名
func GetFileName(path string) string {
	baseName := filepath.Base(path)
	// 去掉文件名的后缀
	baseName = baseName[:len(baseName)-len(filepath.Ext(baseName))]
	return baseName
}

// 获取文件后缀
func GetFileSuffix(path string) string {
	baseName := filepath.Base(path)
	return filepath.Ext(baseName)
}

// 获取文件名带后缀
func GetFileNameWithSuffix(path string) string {
	return filepath.Base(path)
}

//  获取目录下所有文件
func GetDirFile(root string, getDir bool, level int) []string {
	var rst []string
	if strings.HasSuffix(root, `\`) || strings.HasSuffix(root, `/`) {
		root = root[:len(root)-1]
	}
	level = level + strings.Count(root, `/`) + strings.Count(root, `\`)
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			panic(err)
		}
		// 如果是目录的情况下并且不允许采集目录则直接返回
		if !getDir && info.IsDir() {
			return nil
		}
		// 如果是一级目录的情况下并且不允许采集一级目录则直接返回
		if strings.Count(path, `/`)+strings.Count(path, `\`) > level {
			return nil
		}
		// 其他情况则采集
		rst = append(rst, path)
		return nil
	})
	if err != nil {
		panic(err)
	}
	return rst
}

// 获取一级目录下的文件列表
func GetFileListLeval1(root string, getDir ...bool) []string {
	if len(getDir) > 0 && getDir[0] {
		return GetDirFile(root, true, 1)
	}
	return GetDirFile(root, false, 1)
}

// 文件复制
func CopyFile(src string, dst ...string) {
	srcFile, err := os.Open(src)
	if err != nil {
		panic(err)
	}
	defer srcFile.Close()
	copyFile := GetFileDir(src) + "/" + GetFileName(src) + ".copy" + GetFileSuffix(src)
	if len(dst) > 0 && FileIsExists(dst[0]) {
		copyFile = dst[0]
	}
	dstFile, err := os.Create(copyFile)
	if err != nil {
		panic(err)
	}
	defer dstFile.Close()
	_, err = io.Copy(dstFile, srcFile)
	if err != nil {
		panic(err)
	}
	return
}

// 文件重命名
func RenameFile(old, new string) {
	// 替换为实际的旧文件名和新文件名
	err := os.Rename(old, new)
	if err != nil {
		panic(err)
	}
}

// 创建文件夹
func MakeDirs(dir string) {
	if !FileIsExists(dir) {
		if err := os.MkdirAll(dir, 0777); err != nil {
			panic(err)
		}
	}
}