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

概述

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

代码

package zdpgo_file

import (

"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 + "/" + fname

err = 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_file

import "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 []string

files, 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 []string

files, 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)

}

}