feat: add file utils

Author: zarttic

utils/file.go

@@ -0,0 +1,21 @@
+package utils
+
+import (
+	"io/fs"
+	"path/filepath"
+)
+
+// DirSize 获取一个目录的大小
+func DirSize(dirPath string) (int64, error) {
+	var size int64
+	err := filepath.Walk(dirPath, func(path string, info fs.FileInfo, err error) error {
+		if err != nil {
+			return err
+		}
+		if !info.IsDir() {
+			size += info.Size()
+		}
+		return nil
+	})
+	return size, err
+}

utils/file_test.go

@@ -0,0 +1,14 @@
+package utils
+
+import (
+	"github.com/stretchr/testify/assert"
+	"os"
+	"testing"
+)
+
+func TestDirSize(t *testing.T) {
+	dir, _ := os.Getwd()
+	dirSize, err := DirSize(dir)
+	assert.Nil(t, err)
+	assert.True(t, dirSize > 0)
+}