fix bug

2026-02-03 22:38:24 +08:00
parent 3f70d5ac01
commit 35874406b3
4 changed files with 53 additions and 32 deletions
--- a/internal/logic/strategy/data.go
+++ b/internal/logic/strategy/data.go
@@ -34,7 +34,12 @@ func GetFullData(code string) *StockData {
 }

 func GetStocks() (stocks []string) {
-	impl.DBService.Model(&models.StockBasic{}).Group("ts_code").Pluck("ts_code", &stocks)
+	impl.DBService.Model(&models.StockBasic{}).Pluck("ts_code", &stocks)
+	return
+}
+
+func GetStocksByCond(key string, val int) (stocks []string) {
+	impl.DBService.Model(&models.StockBasic{}).Where(key+"=?", val).Pluck("ts_code", &stocks)
 	return
 }