refactor(tool): 优化文件处理逻辑和错误处理
All checks were successful
构建上传工具 / build-tool (push) Successful in 54s
部署开发环境 / deploy-dev (push) Successful in 1m13s

This commit is contained in:
2025-10-11 12:26:43 +08:00
parent f35fad442a
commit 1f0158dd76

View File

@@ -41,7 +41,7 @@ func main() {
files, err := getTxtFiles("./") files, err := getTxtFiles("./")
if err != nil { if err != nil {
fmt.Println(err) fmt.Println(err)
return continue
} }
if files != nil { if files != nil {
start := time.Now() start := time.Now()
@@ -49,23 +49,23 @@ func main() {
fileLines := make(map[string]int) fileLines := make(map[string]int)
fmt.Println("正在统计", len(files), "个文件行数") fmt.Println("正在统计", len(files), "个文件行数")
for _, filePath := range files { for _, filePath := range files {
file, err := os.Open(filePath) file, err := os.Open(filePath)
if err != nil { if err != nil {
fmt.Println("打开文件失败:", err) fmt.Println("打开文件失败:", err)
} }
defer file.Close()
// 使用 bufio.Scanner 逐行读取 // 使用 bufio.Scanner 逐行读取
scanner := bufio.NewScanner(file) scanner := bufio.NewScanner(file)
lineCount := 0 lineCount := 0
for scanner.Scan() { for scanner.Scan() {
lineCount++ lineCount++
} }
file.Close()
if lineCount == 0 { if lineCount == 0 {
return continue
} }
fileLines[filepath.Base(filePath)] = lineCount fileLines[filepath.Base(filePath)] = lineCount
fmt.Println(filepath.Base(filePath), "文件行数:", lineCount) fmt.Println(filepath.Base(filePath), "文件行数:", lineCount)
} }
var tasks []Task var tasks []Task