deving

internal/logic/strategy/data.go

@@ -8,6 +8,8 @@ import (
 	"git.apinb.com/quant/gostock/internal/models"
 )
 
+var Cache = make(map[string]*StockData)
+
 type StockData struct {
 	Basic         models.StockBasic
 	Daily         []models.StockDaily
@@ -15,6 +17,16 @@ type StockData struct {
 	FinaIndicator []models.StockFinaIndicator
 }
 
+func InitCacheByAll() {
+	for _, code := range GetStocks() {
+		Cache[code] = GetFullData(code)
+	}
+}
+
+func InitCacheByCode(code string) {
+	Cache[code] = GetFullData(code)
+}
+
 func GetFullData(code string) *StockData {
 	var data StockData
 	impl.DBService.Where("ts_code = ?", code).First(&data.Basic)