deving

internal/logic/strategy/data.go

@@ -0,0 +1,27 @@
+package strategy
+
+import (
+	"git.apinb.com/quant/gostock/internal/impl"
+	"git.apinb.com/quant/gostock/internal/models"
+)
+
+type StockData struct {
+	Basic         models.StockBasic
+	Daily         []models.StockDaily
+	Indicator     []models.StockIndicator
+	FinaIndicator []models.StockFinaIndicator
+}
+
+func GetData(code string) *StockData {
+	var data StockData
+	impl.DBService.Where("ts_code = ?", code).First(&data.Basic)
+	impl.DBService.Where("ts_code = ?", code).Order("trade_date desc").Find(&data.Daily)
+	impl.DBService.Where("ts_code = ?", code).Order("trade_date desc").Find(&data.Indicator)
+	impl.DBService.Where("ts_code = ?", code).Order("period desc").Find(&data.FinaIndicator)
+	return &data
+}
+
+func GetIndustry() (industry []string) {
+	impl.DBService.Model(&models.StockBasic{}).Group("industry").Pluck("industry", &industry)
+	return
+}