You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.2 KiB
103 lines
2.2 KiB
package core
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"scagent/util"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// 传入的参数
|
|
// scope: all 所有文件
|
|
// scope: dir 目录下的文件
|
|
type Args struct {
|
|
FilePath, Scope string
|
|
}
|
|
|
|
// 返回的参数
|
|
type Reply struct {
|
|
Redata string
|
|
}
|
|
|
|
// 返回的json数据
|
|
type FileRpcService struct{}
|
|
|
|
// 返回的文件json结构体
|
|
type FloopJson struct {
|
|
Path string `json:"path"`
|
|
Size int64 `json:"size"`
|
|
Hash string `json:"hash"`
|
|
}
|
|
|
|
// 文件列表
|
|
type FlpList struct {
|
|
Flist []FloopJson `json:"list"`
|
|
}
|
|
|
|
// 获取文件路径
|
|
func (f *FileRpcService) GetFilePath(args *Args, replay *string) error {
|
|
|
|
// 启用日志
|
|
logger := util.NewProductionLogger()
|
|
defer logger.Sync()
|
|
// 并发处理
|
|
go scanFileUnderPath(args.FilePath, args.Scope, logger, replay)
|
|
runtime.Gosched()
|
|
//
|
|
return nil
|
|
}
|
|
|
|
// 扫描指定目录下的文件，可以全部，也可以局部
|
|
func scanFileUnderPath(rootDir string, scope string, logger *zap.Logger, replay *string) {
|
|
var flplist FlpList // 文件列表
|
|
|
|
// 遍历目录下的文件
|
|
var allFiles []string // 存储所有文件路径
|
|
|
|
// 递归遍历目录及其子目录
|
|
err := filepath.WalkDir(rootDir, func(path string, d os.DirEntry, err error) error {
|
|
// 处理遍历过程中的错误（如权限不足）
|
|
if err != nil {
|
|
logger.Error("访问路径出错", zap.Error(err))
|
|
// 判断扫描范围
|
|
if scope == "all" {
|
|
return filepath.SkipDir
|
|
} else {
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// 只收集文件（排除目录）
|
|
if !d.IsDir() {
|
|
allFiles = append(allFiles, path)
|
|
}
|
|
return nil
|
|
})
|
|
// 打印错误日志
|
|
if err != nil {
|
|
logger.Error("遍历目录出错", zap.Error(err))
|
|
}
|
|
|
|
//
|
|
// 打印所有文件路径
|
|
for _, file := range allFiles {
|
|
// 打印所有文件路径
|
|
// fmt.Printf("%d. %s\n", i+1, file)
|
|
d, _ := os.Stat(file)
|
|
// 加入到列表
|
|
hash := util.CalacHash(file)
|
|
flplist.Flist = append(flplist.Flist, FloopJson{Path: file, Size: d.Size(), Hash: hash})
|
|
}
|
|
|
|
// 将文件列表转换为json字符串
|
|
jsonStr, err := json.Marshal(flplist)
|
|
if err != nil {
|
|
logger.Error("转换为json字符串出错", zap.Error(err))
|
|
}
|
|
// 执行结果存到replay
|
|
*replay = string(jsonStr)
|
|
|
|
}
|
|
|