add: file export for table
This commit is contained in:
27
lib/file/file.go
Normal file
27
lib/file/file.go
Normal file
@@ -0,0 +1,27 @@
|
||||
package file
|
||||
|
||||
import (
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
func GetFileAndDirCount(dir string) (int, int, error) {
|
||||
var fileCount, dirCount int
|
||||
|
||||
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if path == dir {
|
||||
return nil // 跳过当前目录
|
||||
}
|
||||
if info.IsDir() {
|
||||
dirCount++
|
||||
} else {
|
||||
fileCount++
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
return fileCount, dirCount, err
|
||||
}
|
||||
@@ -61,24 +61,3 @@ func GetFileInfo(dir, suffix string) ([]FileInfo, error) {
|
||||
}
|
||||
return files, nil
|
||||
}
|
||||
|
||||
func GetFileAndDirCount(dir string) (int, int, error) {
|
||||
var fileCount, dirCount int
|
||||
|
||||
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if path == dir {
|
||||
return nil // 跳过当前目录
|
||||
}
|
||||
if info.IsDir() {
|
||||
dirCount++
|
||||
} else {
|
||||
fileCount++
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
return fileCount, dirCount, err
|
||||
}
|
||||
|
||||
@@ -59,24 +59,3 @@ func GetFileInfo(dir, suffix string) ([]FileInfo, error) {
|
||||
}
|
||||
return files, nil
|
||||
}
|
||||
|
||||
func GetFileAndDirCount(dir string) (int, int, error) {
|
||||
var fileCount, dirCount int
|
||||
|
||||
err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if path == dir {
|
||||
return nil // 跳过当前目录
|
||||
}
|
||||
if info.IsDir() {
|
||||
dirCount++
|
||||
} else {
|
||||
fileCount++
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
return fileCount, dirCount, err
|
||||
}
|
||||
|
||||
@@ -13,7 +13,7 @@ func DataResp(data any) map[string]any {
|
||||
return map[string]any{"code": CODE_SUCC, "data": data}
|
||||
}
|
||||
|
||||
func SuccResp() map[string]any {
|
||||
func SuccMessageResp() map[string]any {
|
||||
return map[string]any{"code": CODE_SUCC, "message": "success"}
|
||||
}
|
||||
|
||||
@@ -21,6 +21,15 @@ func TotalResp(total int64) map[string]any {
|
||||
return map[string]any{"code": CODE_SUCC, "total": total}
|
||||
}
|
||||
|
||||
func TotalDataResp(data any, total int64) map[string]any {
|
||||
func TotalDataResp(data any, total any) map[string]any {
|
||||
return map[string]any{"code": CODE_SUCC, "data": data, "total": total}
|
||||
}
|
||||
|
||||
func SuccResp(va map[string]any) map[string]any {
|
||||
resp := make(map[string]any)
|
||||
resp["code"] = CODE_SUCC
|
||||
for k, v := range va {
|
||||
resp[k] = v
|
||||
}
|
||||
return resp
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user