From 7462740cf92583c7ea6e86a10dcda53c4bb36e6b Mon Sep 17 00:00:00 2001
From: yanweidong <david.yan>
Date: Wed, 4 Feb 2026 00:37:05 +0800
Subject: [PATCH] fix bug

---
 internal/logic/strategy/boot.go | 63 +++++++++++++++++++++------------
 1 file changed, 40 insertions(+), 23 deletions(-)

diff --git a/internal/logic/strategy/boot.go b/internal/logic/strategy/boot.go
index da0e3b5..6a2f795 100644
--- a/internal/logic/strategy/boot.go
+++ b/internal/logic/strategy/boot.go
@@ -1,6 +1,7 @@
 package strategy
 
 import (
+	"fmt"
 	"log"
 	"sync"
 
@@ -12,38 +13,54 @@ func Boot() {
 	InitCacheByAll()
 }
 
-func BootAiStart(key string, ymd int) {
+// 启动 AI 分析任务
+func BootAiStart(key string, ymd int) error {
 	var datas []models.StratModel
 	err := impl.DBService.Where("strat_key=? and ymd=? and ai_score=0", key, ymd).Find(&datas).Error
 	if err != nil {
-		panic(err)
+		log.Printf("Failed to query data: %v", err)
+		return fmt.Errorf("query failed: %w", err)
 	}
 
-	// Create a buffered channel with a capacity of 20 to act as a semaphore
-	semaphore := make(chan struct{}, 20)
-
-	var wg sync.WaitGroup
+	// 构造任务列表
+	var tasks []func()
 	for _, row := range datas {
+		row := row // 避免闭包捕获循环变量
+		tasks = append(tasks, func() {
+			BootAiTask(row.ID, row.Code)
+		})
+	}
+
+	// 使用并发控制执行任务
+	runWithLimit(20, tasks)
+	return nil
+}
+
+// 执行单个 AI 分析任务
+func BootAiTask(id uint, code string) {
+	result, err := AiAnalysis(code)
+	if err != nil {
+		log.Printf("ERROR BootAiTask - ID: %d, Code: %s, Error: %v", id, code, err)
+		return
+	}
+	log.Printf("SUCCESS BootAiTask - ID: %d, Code: %s", id, code)
+	impl.DBService.Model(&models.StratModel{}).Where("id=?", id).Updates(result)
+}
+
+// 通用并发控制函数
+func runWithLimit(limit int, tasks []func()) {
+	semaphore := make(chan struct{}, limit)
+	var wg sync.WaitGroup
+
+	for _, task := range tasks {
 		wg.Add(1)
-		// Acquire a slot in the semaphore
-		semaphore <- struct{}{}
-		go func(row models.StratModel) {
+		semaphore <- struct{}{} // 获取信号量
+		go func(t func()) {
 			defer wg.Done()
-			// Release the slot in the semaphore when done
-			defer func() { <-semaphore }()
-			BootAiTask(row.ID, row.Code, &wg)
-		}(row)
+			defer func() { <-semaphore }() // 释放信号量
+			t()
+		}(task)
 	}
 
 	wg.Wait()
 }
-
-func BootAiTask(id uint, code string, wg *sync.WaitGroup) {
-	defer wg.Done()
-	result, err := AiAnalysis(code)
-	if err != nil {
-		log.Println("ERROR BootAiTask", err.Error())
-		return
-	}
-	impl.DBService.Model(&models.StratModel{}).Where("id=?", id).Updates(result)
-}