gmy 5 месяцев назад
Родитель
Сommit
9c5989f348
1 измененных файлов с 6 добавлено и 5 удалено
  1. 6 5
      services/ruizide/data_processor.go

+ 6 - 5
services/ruizide/data_processor.go

@@ -51,6 +51,7 @@ func downloadData(ctx context.Context) error {
 		chromedp.Click(clientSearchLink, chromedp.ByQuery),
 		chromedp.WaitVisible(`input[class="ais-SearchBox-input rounded border py-2 px-3 shadow-sm font-size-14 w-100"]`, chromedp.ByQuery),
 		chromedp.SetValue(`input[class="ais-SearchBox-input rounded border py-2 px-3 shadow-sm font-size-14 w-100"]`, "oil demand signals weekly report", chromedp.ByQuery),
+		chromedp.Sleep(10*time.Second),
 		//chromedp.Click(`div.ais-InfiniteHits li a:has(img[src="/Static/img/icons/xls.png"])`, chromedp.ByQuery),
 	); err != nil {
 		return fmt.Errorf("下载 Analytics Library 数据错误: %v", err)
@@ -92,7 +93,7 @@ func downloadData(ctx context.Context) error {
 // 等待下载文件并重命名
 func waitAndRenameDownloadedFile(newFileName, targetDir string) error {
 	// 等待一段时间以确保文件下载完成
-	time.Sleep(60 * time.Second) // 可能需要根据实际情况调整
+	time.Sleep(100 * time.Second) // 可能需要根据实际情况调整
 
 	// 查找下载目录中的文件
 	files, err := filepath.Glob(filepath.Join(defaultDir, "*.xlsx"))
@@ -155,16 +156,16 @@ func moveFile(source, destination string) error {
 	time.Sleep(60 * time.Second)
 
 	// 删除源文件
-	if err := os.Remove(source); err != nil {
+	/*if err := os.Remove(source); err != nil {
 		return fmt.Errorf("删除源文件时出错: %v", err)
-	}
+	}*/
 
 	return nil
 }
 
 // 解析网页数据,下载文件
-// func main() {
-func resolverNet() {
+func main() {
+	//func resolverNet() {
 	// 创建下载目录
 	if err := os.MkdirAll(downloadDir, os.ModePerm); err != nil {
 		fmt.Printf("创建下载目录时出错: %v\n", err)