From f883b0a7d81505b23d48f3d22842b11dd4660b90 Mon Sep 17 00:00:00 2001 From: YGXB_net Date: Wed, 10 Sep 2025 21:26:24 +0800 Subject: [PATCH] =?UTF-8?q?refactor(tool):=20=E4=BC=98=E5=8C=96=E6=96=87?= =?UTF-8?q?=E4=BB=B6=E5=A4=84=E7=90=86=E9=80=BB=E8=BE=91=E5=92=8C=E6=80=A7?= =?UTF-8?q?=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tool/update-dypid.go | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/tool/update-dypid.go b/tool/update-dypid.go index a3fe7fe..61a84bd 100644 --- a/tool/update-dypid.go +++ b/tool/update-dypid.go @@ -48,7 +48,6 @@ func main() { //检测./upload fmt.Println("程序启动成功,正在检测txt文件") - //os.Mkdir("./upload", os.ModePerm) for { files, err := getTxtFiles("./") if err != nil { @@ -62,16 +61,19 @@ func main() { wg.Add(1) go func() { processFile(filePath) - os.Remove(filePath) + err := os.Truncate(filePath, 0) + if err != nil { + fmt.Println("清空文件失败:", err) + } wg.Done() }() } wg.Wait() - } - time.Sleep(2 * time.Second) + time.Sleep(time.Minute) } } + func uploadDataToServer(data string) error { params := url.Values{} params.Set("token", viper.GetString("token")) @@ -88,10 +90,8 @@ func uploadDataToServer(data string) error { } // 获取目录中的所有txt文件 -func getTxtFiles(dir string) ([]string, error) { - var txtFiles []string - - err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error { +func getTxtFiles(dir string) (txtFiles []string, err error) { + err = filepath.Walk(dir, func(path string, info os.FileInfo, err error) error { if err != nil { return err } @@ -151,6 +151,7 @@ func processFile(filePath string) { if err := scanner.Err(); err != nil { fmt.Printf("读取文件 %s 错误: %v\n", filePath, err) + return } fmt.Printf("文件【%s】处理完成,共处理 %d 行数据\n", filePath, lineCount)