refactor(tool): 优化文件处理逻辑和错误处理
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user