当前位置：首页 > news >正文

Golang实现根据文件后缀删除文件和递归删除文件

news 来源：原创 2024/7/5 4:43:34

概述

这个功能会非常强大，因为在日常工作中，我通常会遇到需要批量删除文件的场景，通过这个方法，再结合我的另一个命令行开发框架，能够很轻松的开发出这个功能。

代码

package zdpgo_fileimport ("errors""os""path/filepath""strings"
)// DeleteFile 删除文件
func DeleteFile(filePath string) error {if !IsExist(filePath) {return errors.New("要删除的文件不存在")}err := os.Remove(filePath)if err != nil {return err}return nil
}// DeleteFileBySuffix 根据文件后缀进行删除
func DeleteFileBySuffix(dir, suffix string) error {// 校验目录if !IsDir(dir) {return errors.New(dir + "不是一个有效的目录")}// 校验后缀，为了安全，后缀不能为空if suffix == "" {return errors.New("文件后缀不能为空")}// 遍历文件夹err := filepath.Walk(dir, func(fname string, fi os.FileInfo, err error) error {// 忽略目录if fi.IsDir() {return nil}// 找到了，进行删除if len(suffix) == 0 || strings.HasSuffix(strings.ToLower(fi.Name()), suffix) {err = DeleteFile(fname){if err != nil {return err}}}return nil})return err
}// DeleteFileBySuffixRecursion 递归根据文件后缀删除
func DeleteFileBySuffixRecursion(dir, suffix string) error {// 校验目录if !IsDir(dir) {return errors.New(dir + "不是一个有效的目录")}// 校验后缀，为了安全，后缀不能为空if suffix == "" {return errors.New("文件后缀不能为空")}// 遍历文件夹err := filepath.Walk(dir, func(fname string, fi os.FileInfo, err error) error {// 忽略目录if fi.IsDir() {newDir := dir + "/" + fnameerr = DeleteFileBySuffixRecursion(newDir, suffix)if err != nil {return err}}// 找到了，进行删除if len(suffix) == 0 || strings.HasSuffix(strings.ToLower(fi.Name()), suffix) {err = DeleteFile(fname){if err != nil {return err}}}return nil})return err
}

测试

package zdpgo_fileimport "testing"func TestDeleteFileBySuffix(t *testing.T) {var err error// 添加文件夹err = AddDir("test")if err != nil {t.Error(err)}// 添加文件err = AddFile("test/test.txt", []byte("hello world"))if err != nil {t.Error(err)}err = AddFile("test/test.txt2", []byte("hello world"))if err != nil {t.Error(err)}// 根据文件后缀删除文件err = DeleteFileBySuffix("test", ".txt")if err != nil {t.Error(err)}// 根据文件后缀查找文件var files []stringfiles, err = FindDirBySuffix("test", ".txt")if err != nil {t.Error(err)}if len(files) != 0 {t.Errorf("len(files) should be 0, but %d", len(files))}files, err = FindDirBySuffix("test", ".txt2")if err != nil {t.Error(err)}if len(files) != 1 {t.Errorf("len(files) should be 1, but %d", len(files))}// 删除文件夹err = DeleteDir("test")if err != nil {t.Error(err)}
}func TestDeleteFileBySuffixRecursion(t *testing.T) {var err error// 添加文件夹err = AddDir("test/a/b/c")if err != nil {t.Error(err)}// 添加文件err = AddFile("test/test.txt", []byte("hello world"))if err != nil {t.Error(err)}err = AddFile("test/a/test.txt", []byte("hello world"))if err != nil {t.Error(err)}err = AddFile("test/a/b/test.txt", []byte("hello world"))if err != nil {t.Error(err)}err = AddFile("test/a/b/c/test.txt", []byte("hello world"))if err != nil {t.Error(err)}// 根据文件后缀删除文件err = DeleteFileBySuffix("test", ".txt")if err != nil {t.Error(err)}// 根据文件后缀查找文件var files []stringfiles, err = FindDirBySuffix("test", ".txt")if err != nil {t.Error(err)}if len(files) != 0 {t.Errorf("len(files) should be 0, but %d", len(files))}// 删除文件夹err = DeleteDir("test")if err != nil {t.Error(err)}
}