update logic of collecting files

This commit is contained in:
r0n1n7an 2023-07-26 00:07:53 +08:00
parent 79415a4819
commit 84adc85fe8
2 changed files with 27 additions and 3 deletions

Binary file not shown.

View File

@ -60,7 +60,7 @@ func main() {
} }
log.Printf("[MSG] Summary Loacal AODs ...\r\n") log.Printf("[MSG] Summary Loacal AODs ...\r\n")
aodInfoList, err = _SummaryFiles(aodPath, aodType) aodInfoList, err = _SummaryAOD(aodPath, aodType)
if err != nil { if err != nil {
log.Printf("[ERR] Get AOD List Failed: %s\r\n", err.Error()) log.Printf("[ERR] Get AOD List Failed: %s\r\n", err.Error())
os.Exit(1) os.Exit(1)
@ -68,7 +68,7 @@ func main() {
log.Printf("[MSG] Total Local AODs: %d\r\n", len(aodInfoList)) log.Printf("[MSG] Total Local AODs: %d\r\n", len(aodInfoList))
log.Printf("[MSG] Summary Local Modules ...\r\n") log.Printf("[MSG] Summary Local Modules ...\r\n")
moduleInfoList, err = _SummaryFiles(modPath, "*") moduleInfoList, err = _SummaryFiles(modPath)
if err != nil { if err != nil {
log.Printf("[ERR] Get Module List Failed: %s\r\n", err.Error()) log.Printf("[ERR] Get Module List Failed: %s\r\n", err.Error())
os.Exit(1) os.Exit(1)
@ -170,7 +170,7 @@ func main() {
_WriteFileFromStringSlice(".\\NeedlessModules.txt", needlessModuleList) _WriteFileFromStringSlice(".\\NeedlessModules.txt", needlessModuleList)
} }
func _SummaryFiles(basePath, filter string) ([]FileInfo, error) { func _SummaryAOD(basePath, filter string) ([]FileInfo, error) {
var err error var err error
sortInfoList := make([]FileInfo, 0) sortInfoList := make([]FileInfo, 0)
fileInfoList := make([]FileInfo, 0) fileInfoList := make([]FileInfo, 0)
@ -215,6 +215,30 @@ func _SummaryFiles(basePath, filter string) ([]FileInfo, error) {
return sortInfoList, nil return sortInfoList, nil
} }
func _SummaryFiles(basePath string) ([]FileInfo, error) {
var err error
fileInfoList := make([]FileInfo, 0)
err = filepath.Walk(basePath, func(path string, info os.FileInfo, err error) error {
if !info.IsDir() {
var fileInfo FileInfo
fileInfo.FileName = info.Name()
fileInfo.FilePath = path
fileInfo.FileSize = info.Size()
//fileInfo.ModTime = info.ModTime().Format("2006-01-02 15:04:05")
if path == basePath+"\\"+info.Name() {
fileInfoList = append(fileInfoList, fileInfo)
}
}
return nil
})
if err != nil {
log.Printf("[ERR] Get File List Failed: %s\r\n", err.Error())
return nil, err
}
return fileInfoList, nil
}
func _RemoveDuplicateElementOfSlice(src []string) []string { func _RemoveDuplicateElementOfSlice(src []string) []string {
dst := make([]string, 0, len(src)) dst := make([]string, 0, len(src))
tmp := map[string]struct{}{} tmp := map[string]struct{}{}