From 8e1b05fb425c32fcfb29ff0b440ed34fab294778 Mon Sep 17 00:00:00 2001
From: yanweidong <david.yan>
Date: Tue, 3 Feb 2026 22:42:47 +0800
Subject: [PATCH] fix bug

---
 internal/logic/strategy/boot.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/internal/logic/strategy/boot.go b/internal/logic/strategy/boot.go
index 41295a4..5fbb075 100644
--- a/internal/logic/strategy/boot.go
+++ b/internal/logic/strategy/boot.go
@@ -2,6 +2,7 @@ package strategy
 
 import (
 	"log"
+	"sync"
 
 	"git.apinb.com/quant/gostock/internal/impl"
 	"git.apinb.com/quant/gostock/internal/models"
@@ -17,12 +18,19 @@ func BootAiStart(key string, ymd int) {
 	if err != nil {
 		panic(err)
 	}
+
+	var wg sync.WaitGroup
+
 	for _, row := range datas {
-		go BootAiTask(row.ID, row.Code)
+		wg.Add(1)
+		go BootAiTask(row.ID, row.Code, &wg)
 	}
+
+	wg.Wait()
 }
 
-func BootAiTask(id uint, code string) {
+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())