743 lines
18 KiB
Go
743 lines
18 KiB
Go
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"crypto/md5"
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// FileInfo 文件信息结构体
|
|
type FileInfo struct {
|
|
Name string `json:"name"`
|
|
Path string `json:"path"`
|
|
Size int64 `json:"size"`
|
|
ModTime time.Time `json:"mod_time"`
|
|
IsDir bool `json:"is_dir"`
|
|
Ext string `json:"ext"`
|
|
}
|
|
|
|
// FilteredFiles 包含按类型分类的文件结果
|
|
type FilteredFiles struct {
|
|
SupportedFiles []FileInfo // 支持的文件类型
|
|
OtherFiles []FileInfo // 其他文件类型
|
|
}
|
|
|
|
// fileUtilsStruct 用于封装文件工具方法
|
|
type fileUtilsStruct struct{}
|
|
|
|
// FileUtils 是对外暴露的文件工具对象
|
|
var FileUtils = fileUtilsStruct{}
|
|
|
|
// Exists 1. 检查文件或目录是否存在
|
|
func (fileUtilsStruct) Exists(path string) bool {
|
|
_, err := os.Stat(path)
|
|
return !os.IsNotExist(err)
|
|
}
|
|
|
|
// 2. 检查是否为目录
|
|
func (fileUtilsStruct) IsDir(path string) bool {
|
|
info, err := os.Stat(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return info.IsDir()
|
|
}
|
|
|
|
// 3. 检查是否为文件
|
|
func (fileUtilsStruct) IsFile(path string) bool {
|
|
info, err := os.Stat(path)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return !info.IsDir()
|
|
}
|
|
|
|
// 4. 获取指定目录下的所有文件夹
|
|
func (fileUtilsStruct) GetDirectories(dirPath string) ([]string, error) {
|
|
var dirs []string
|
|
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
dirs = append(dirs, entry.Name())
|
|
}
|
|
}
|
|
|
|
return dirs, nil
|
|
}
|
|
|
|
// 5. 获取指定目录下指定格式的文件
|
|
func (fileUtilsStruct) GetFilesByExtension(dirPath string, extensions ...string) ([]string, error) {
|
|
var files []string
|
|
|
|
// 标准化扩展名（确保以.开头）
|
|
for i, ext := range extensions {
|
|
if !strings.HasPrefix(ext, ".") {
|
|
extensions[i] = "." + ext
|
|
}
|
|
extensions[i] = strings.ToLower(extensions[i])
|
|
}
|
|
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
fileExt := strings.ToLower(filepath.Ext(entry.Name()))
|
|
for _, ext := range extensions {
|
|
if fileExt == ext {
|
|
files = append(files, entry.Name())
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return files, nil
|
|
}
|
|
|
|
// 6. 找到指定目录下第一个符合格式的文件
|
|
func (f fileUtilsStruct) GetFirstFileByExtension(dirPath string, extensions ...string) (string, error) {
|
|
files, err := f.GetFilesByExtension(dirPath, extensions...)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if len(files) == 0 {
|
|
return "", fmt.Errorf("未找到符合扩展名的文件")
|
|
}
|
|
|
|
return files[0], nil
|
|
}
|
|
|
|
// 7. 递归获取目录下所有文件
|
|
func (fileUtilsStruct) GetAllFiles(dirPath string, recursive bool) ([]FileInfo, error) {
|
|
var files []FileInfo
|
|
|
|
if recursive {
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
files = append(files, FileInfo{
|
|
Name: info.Name(),
|
|
Path: path,
|
|
Size: info.Size(),
|
|
ModTime: info.ModTime(),
|
|
IsDir: false,
|
|
Ext: filepath.Ext(info.Name()),
|
|
})
|
|
}
|
|
return nil
|
|
})
|
|
return files, err
|
|
} else {
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
info, err := entry.Info()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
files = append(files, FileInfo{
|
|
Name: info.Name(),
|
|
Path: filepath.Join(dirPath, info.Name()),
|
|
Size: info.Size(),
|
|
ModTime: info.ModTime(),
|
|
IsDir: false,
|
|
Ext: filepath.Ext(info.Name()),
|
|
})
|
|
}
|
|
}
|
|
return files, nil
|
|
}
|
|
}
|
|
|
|
// GetFilteredFiles 根据指定的文件类型递归获取文件，并按类型分类
|
|
func (fileUtilsStruct) GetFilteredFiles(dirPath string, recursive bool, supportedTypes []string) (*FilteredFiles, error) {
|
|
var result FilteredFiles
|
|
|
|
// 将支持的文件类型转换为map，便于快速查找（转为小写）
|
|
supportedMap := make(map[string]bool)
|
|
for _, ext := range supportedTypes {
|
|
supportedMap[strings.ToLower(ext)] = true
|
|
}
|
|
|
|
if recursive {
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
fileInfo := FileInfo{
|
|
Name: info.Name(),
|
|
Path: path,
|
|
Size: info.Size(),
|
|
ModTime: info.ModTime(),
|
|
IsDir: false,
|
|
Ext: filepath.Ext(info.Name()),
|
|
}
|
|
|
|
// 检查文件扩展名是否在支持列表中（不区分大小写）
|
|
ext := strings.ToLower(filepath.Ext(info.Name()))
|
|
if supportedMap[ext] {
|
|
result.SupportedFiles = append(result.SupportedFiles, fileInfo)
|
|
} else {
|
|
result.OtherFiles = append(result.OtherFiles, fileInfo)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
return &result, err
|
|
} else {
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
info, err := entry.Info()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
fileInfo := FileInfo{
|
|
Name: info.Name(),
|
|
Path: filepath.Join(dirPath, info.Name()),
|
|
Size: info.Size(),
|
|
ModTime: info.ModTime(),
|
|
IsDir: false,
|
|
Ext: filepath.Ext(info.Name()),
|
|
}
|
|
|
|
// 检查文件扩展名是否在支持列表中（不区分大小写）
|
|
ext := strings.ToLower(filepath.Ext(info.Name()))
|
|
if supportedMap[ext] {
|
|
result.SupportedFiles = append(result.SupportedFiles, fileInfo)
|
|
} else {
|
|
result.OtherFiles = append(result.OtherFiles, fileInfo)
|
|
}
|
|
}
|
|
}
|
|
return &result, nil
|
|
}
|
|
}
|
|
|
|
// 递归获取目录下所有文件夹
|
|
func (fileUtilsStruct) GetAllDirs(dirPath string, recursive bool) ([]FileInfo, error) {
|
|
var dirs []FileInfo
|
|
|
|
if recursive {
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 只收集目录，排除根目录本身
|
|
if info.IsDir() && path != dirPath {
|
|
dirs = append(dirs, FileInfo{
|
|
Name: info.Name(),
|
|
Path: path,
|
|
Size: 0, // 目录大小通常为0或不适用
|
|
ModTime: info.ModTime(),
|
|
IsDir: true,
|
|
Ext: "", // 目录没有扩展名
|
|
})
|
|
}
|
|
return nil
|
|
})
|
|
return dirs, err
|
|
} else {
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if entry.IsDir() {
|
|
info, err := entry.Info()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
dirs = append(dirs, FileInfo{
|
|
Name: info.Name(),
|
|
Path: filepath.Join(dirPath, info.Name()),
|
|
Size: 0, // 目录大小通常为0或不适用
|
|
ModTime: info.ModTime(),
|
|
IsDir: true,
|
|
Ext: "", // 目录没有扩展名
|
|
})
|
|
}
|
|
}
|
|
return dirs, nil
|
|
}
|
|
}
|
|
|
|
// 8. 删除文件或目录
|
|
func (f fileUtilsStruct) Delete(path string) error {
|
|
if !f.Exists(path) {
|
|
return fmt.Errorf("路径不存在: %s", path)
|
|
}
|
|
|
|
return os.RemoveAll(path)
|
|
}
|
|
|
|
// 9. 创建目录（递归创建）
|
|
func (fileUtilsStruct) CreateDir(dirPath string) error {
|
|
return os.MkdirAll(dirPath, 0755)
|
|
}
|
|
|
|
// 10. 复制文件
|
|
func (f fileUtilsStruct) CopyFile(src, dst string) error {
|
|
// 检查目标文件是否存在
|
|
if _, err := os.Stat(dst); err == nil {
|
|
// 目标文件存在，跳过
|
|
return os.ErrExist
|
|
}
|
|
|
|
sourceFile, err := os.Open(src)
|
|
if err != nil {
|
|
return fmt.Errorf("打开源文件失败: %w", err)
|
|
}
|
|
defer sourceFile.Close()
|
|
|
|
// 获取源文件信息
|
|
srcInfo, err := sourceFile.Stat()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 确保目标目录存在
|
|
dstDir := filepath.Dir(dst)
|
|
if err := f.CreateDir(dstDir); err != nil {
|
|
return fmt.Errorf("创建目标目录失败: %w", err)
|
|
}
|
|
|
|
destFile, err := os.Create(dst)
|
|
if err != nil {
|
|
return fmt.Errorf("创建目标文件失败: %w", err)
|
|
}
|
|
defer destFile.Close()
|
|
|
|
_, err = io.Copy(destFile, sourceFile)
|
|
if err != nil {
|
|
return fmt.Errorf("复制文件失败: %w", err)
|
|
}
|
|
|
|
// 设置文件权限
|
|
return os.Chmod(dst, srcInfo.Mode())
|
|
}
|
|
|
|
// CopyDir 递归复制整个目录
|
|
func (f fileUtilsStruct) CopyDir(src, dst string) error {
|
|
// 获取源目录信息
|
|
srcInfo, err := os.Stat(src)
|
|
if err != nil {
|
|
return fmt.Errorf("获取源目录信息失败: %w", err)
|
|
}
|
|
|
|
// 确保源路径是一个目录
|
|
if !srcInfo.IsDir() {
|
|
return fmt.Errorf("源路径不是目录: %s", src)
|
|
}
|
|
|
|
// 创建目标目录
|
|
if err := os.MkdirAll(dst, srcInfo.Mode()); err != nil {
|
|
return fmt.Errorf("创建目标目录失败: %w", err)
|
|
}
|
|
|
|
// 读取源目录内容
|
|
entries, err := os.ReadDir(src)
|
|
if err != nil {
|
|
return fmt.Errorf("读取源目录失败: %w", err)
|
|
}
|
|
|
|
// 遍历源目录中的每个条目
|
|
for _, entry := range entries {
|
|
srcPath := filepath.Join(src, entry.Name())
|
|
dstPath := filepath.Join(dst, entry.Name())
|
|
|
|
if entry.IsDir() {
|
|
// 如果是目录，递归复制
|
|
if err := f.CopyDir(srcPath, dstPath); err != nil {
|
|
// 子目录失败也继续复制其他内容
|
|
fmt.Printf("[WARN] 复制子目录 %s 失败: %v\n", entry.Name(), err)
|
|
continue
|
|
}
|
|
} else {
|
|
err := f.CopyFile(srcPath, dstPath)
|
|
if err != nil {
|
|
// 判断是否是文件已存在或拒绝访问
|
|
if os.IsPermission(err) || os.IsExist(err) {
|
|
fmt.Printf("[WARN] 复制文件 %s 被跳过: %v\n", dstPath, err)
|
|
continue
|
|
}
|
|
// 其他错误还是报错
|
|
return fmt.Errorf("复制文件 %s 失败: %w", entry.Name(), err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// CopyDirWithOptions 带选项的目录复制函数
|
|
type CopyOptions struct {
|
|
Overwrite bool // 是否覆盖已存在的文件
|
|
SkipSymlinks bool // 是否跳过符号链接
|
|
}
|
|
|
|
func (f fileUtilsStruct) CopyDirWithOptions(src, dst string, opts *CopyOptions) error {
|
|
if opts == nil {
|
|
opts = &CopyOptions{Overwrite: true, SkipSymlinks: false}
|
|
}
|
|
|
|
// 获取源目录信息
|
|
srcInfo, err := os.Stat(src)
|
|
if err != nil {
|
|
return fmt.Errorf("获取源目录信息失败: %w", err)
|
|
}
|
|
|
|
if !srcInfo.IsDir() {
|
|
return fmt.Errorf("源路径不是目录: %s", src)
|
|
}
|
|
|
|
// 检查目标目录是否已存在
|
|
if dstInfo, err := os.Stat(dst); err == nil {
|
|
if !dstInfo.IsDir() {
|
|
return fmt.Errorf("目标路径已存在但不是目录: %s", dst)
|
|
}
|
|
if !opts.Overwrite {
|
|
return fmt.Errorf("目标目录已存在: %s", dst)
|
|
}
|
|
} else if !os.IsNotExist(err) {
|
|
return fmt.Errorf("检查目标目录失败: %w", err)
|
|
}
|
|
|
|
// 创建目标目录
|
|
if err := os.MkdirAll(dst, srcInfo.Mode()); err != nil {
|
|
return fmt.Errorf("创建目标目录失败: %w", err)
|
|
}
|
|
|
|
// 读取源目录内容
|
|
entries, err := os.ReadDir(src)
|
|
if err != nil {
|
|
return fmt.Errorf("读取源目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
srcPath := filepath.Join(src, entry.Name())
|
|
dstPath := filepath.Join(dst, entry.Name())
|
|
|
|
// 处理符号链接
|
|
if entry.Type()&os.ModeSymlink != 0 {
|
|
if opts.SkipSymlinks {
|
|
continue
|
|
}
|
|
if err := f.copySymlink(srcPath, dstPath); err != nil {
|
|
return fmt.Errorf("复制符号链接 %s 失败: %w", entry.Name(), err)
|
|
}
|
|
continue
|
|
}
|
|
|
|
if entry.IsDir() {
|
|
// 递归复制子目录
|
|
if err := f.CopyDirWithOptions(srcPath, dstPath, opts); err != nil {
|
|
return fmt.Errorf("复制子目录 %s 失败: %w", entry.Name(), err)
|
|
}
|
|
} else {
|
|
// 检查文件是否已存在
|
|
if !opts.Overwrite {
|
|
if _, err := os.Stat(dstPath); err == nil {
|
|
return fmt.Errorf("目标文件已存在: %s", dstPath)
|
|
}
|
|
}
|
|
|
|
// 复制文件
|
|
if err := f.CopyFile(srcPath, dstPath); err != nil {
|
|
return fmt.Errorf("复制文件 %s 失败: %w", entry.Name(), err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// copySymlink 复制符号链接
|
|
func (fileUtilsStruct) copySymlink(src, dst string) error {
|
|
link, err := os.Readlink(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 删除可能已存在的目标文件/链接
|
|
os.Remove(dst)
|
|
|
|
return os.Symlink(link, dst)
|
|
}
|
|
|
|
// 11. 移动/重命名文件
|
|
func (f fileUtilsStruct) MoveFile(src, dst string) error {
|
|
// 确保目标目录存在
|
|
dstDir := filepath.Dir(dst)
|
|
if err := f.CreateDir(dstDir); err != nil {
|
|
return fmt.Errorf("创建目标目录失败: %w", err)
|
|
}
|
|
|
|
return os.Rename(src, dst)
|
|
}
|
|
|
|
// 12. 读取文件全部内容
|
|
func (fileUtilsStruct) ReadFile(filePath string) (string, error) {
|
|
content, err := os.ReadFile(filePath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("读取文件失败: %w", err)
|
|
}
|
|
return string(content), nil
|
|
}
|
|
|
|
// 13. 按行读取文件
|
|
func (fileUtilsStruct) ReadFileLines(filePath string) ([]string, error) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("打开文件失败: %w", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
var lines []string
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
lines = append(lines, scanner.Text())
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
return nil, fmt.Errorf("读取文件失败: %w", err)
|
|
}
|
|
|
|
return lines, nil
|
|
}
|
|
|
|
// 14. 写入文件内容
|
|
func (f fileUtilsStruct) WriteFile(filePath, content string) error {
|
|
// 确保目录存在
|
|
dir := filepath.Dir(filePath)
|
|
if err := f.CreateDir(dir); err != nil {
|
|
return fmt.Errorf("创建目录失败: %w", err)
|
|
}
|
|
|
|
return os.WriteFile(filePath, []byte(content), 0644)
|
|
}
|
|
|
|
// 15. 追加写入文件
|
|
func (f fileUtilsStruct) AppendFile(filePath, content string) error {
|
|
// 确保目录存在
|
|
dir := filepath.Dir(filePath)
|
|
if err := f.CreateDir(dir); err != nil {
|
|
return fmt.Errorf("创建目录失败: %w", err)
|
|
}
|
|
|
|
file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
if err != nil {
|
|
return fmt.Errorf("打开文件失败: %w", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = file.WriteString(content)
|
|
return err
|
|
}
|
|
|
|
// 16. 获取文件大小
|
|
func (fileUtilsStruct) GetFileSize(filePath string) (int64, error) {
|
|
info, err := os.Stat(filePath)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("获取文件信息失败: %w", err)
|
|
}
|
|
return info.Size(), nil
|
|
}
|
|
|
|
// 17. 获取文件修改时间
|
|
func (fileUtilsStruct) GetModTime(filePath string) (time.Time, error) {
|
|
info, err := os.Stat(filePath)
|
|
if err != nil {
|
|
return time.Time{}, fmt.Errorf("获取文件信息失败: %w", err)
|
|
}
|
|
return info.ModTime(), nil
|
|
}
|
|
|
|
// 18. 获取文件扩展名
|
|
func (fileUtilsStruct) GetExtension(filePath string) string {
|
|
return filepath.Ext(filePath)
|
|
}
|
|
|
|
// 19. 获取文件名（不含扩展名）
|
|
func (fileUtilsStruct) GetFileName(filePath string) string {
|
|
base := filepath.Base(filePath)
|
|
ext := filepath.Ext(base)
|
|
return strings.TrimSuffix(base, ext)
|
|
}
|
|
|
|
// 20. 获取文件基本名称（含扩展名）
|
|
func (fileUtilsStruct) GetBaseName(filePath string) string {
|
|
return filepath.Base(filePath)
|
|
}
|
|
|
|
// 21. 获取文件目录
|
|
func (fileUtilsStruct) GetDir(filePath string) string {
|
|
return filepath.Dir(filePath)
|
|
}
|
|
|
|
// 22. 计算文件MD5哈希值
|
|
func (fileUtilsStruct) GetFileMD5(filePath string) (string, error) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("打开文件失败: %w", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
hash := md5.New()
|
|
if _, err := io.Copy(hash, file); err != nil {
|
|
return "", fmt.Errorf("计算哈希失败: %w", err)
|
|
}
|
|
|
|
return fmt.Sprintf("%x", hash.Sum(nil)), nil
|
|
}
|
|
|
|
// 23. 计算文件SHA256哈希值
|
|
func (fileUtilsStruct) GetFileSHA256(filePath string) (string, error) {
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("打开文件失败: %w", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
hash := sha256.New()
|
|
if _, err := io.Copy(hash, file); err != nil {
|
|
return "", fmt.Errorf("计算哈希失败: %w", err)
|
|
}
|
|
|
|
return fmt.Sprintf("%x", hash.Sum(nil)), nil
|
|
}
|
|
|
|
// 24. 清空目录内容（保留目录本身）
|
|
func (fileUtilsStruct) CleanDir(dirPath string) error {
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
path := filepath.Join(dirPath, entry.Name())
|
|
if err := os.RemoveAll(path); err != nil {
|
|
return fmt.Errorf("删除 %s 失败: %w", path, err)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// 25. 获取目录大小
|
|
func (fileUtilsStruct) GetDirSize(dirPath string) (int64, error) {
|
|
var size int64
|
|
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !info.IsDir() {
|
|
size += info.Size()
|
|
}
|
|
return nil
|
|
})
|
|
|
|
return size, err
|
|
}
|
|
|
|
// 26. 格式化文件大小
|
|
func (fileUtilsStruct) FormatFileSize(bytes int64) string {
|
|
const unit = 1024
|
|
if bytes < unit {
|
|
return fmt.Sprintf("%d B", bytes)
|
|
}
|
|
div, exp := int64(unit), 0
|
|
for n := bytes / unit; n >= unit; n /= unit {
|
|
div *= unit
|
|
exp++
|
|
}
|
|
return fmt.Sprintf("%.1f %cB", float64(bytes)/float64(div), "KMGTPE"[exp])
|
|
}
|
|
|
|
// 27. 搜索文件（按文件名模式）
|
|
func (fileUtilsStruct) SearchFiles(dirPath, pattern string, recursive bool) ([]string, error) {
|
|
var matches []string
|
|
|
|
if recursive {
|
|
err := filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.IsDir() {
|
|
matched, err := filepath.Match(pattern, info.Name())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if matched {
|
|
matches = append(matches, path)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
return matches, err
|
|
} else {
|
|
entries, err := os.ReadDir(dirPath)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("读取目录失败: %w", err)
|
|
}
|
|
|
|
for _, entry := range entries {
|
|
if !entry.IsDir() {
|
|
matched, err := filepath.Match(pattern, entry.Name())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if matched {
|
|
matches = append(matches, filepath.Join(dirPath, entry.Name()))
|
|
}
|
|
}
|
|
}
|
|
return matches, nil
|
|
}
|
|
}
|
|
|
|
// 28. 获取当前工作目录
|
|
func (fileUtilsStruct) GetCurrentDir() (string, error) {
|
|
return os.Getwd()
|
|
}
|
|
|
|
// 29. 改变工作目录
|
|
func (fileUtilsStruct) ChangeDir(dirPath string) error {
|
|
return os.Chdir(dirPath)
|
|
}
|