gostock/internal/logic/strategy/data.go

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 GetFullData(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 GetStocks() (stocks []string) {
	impl.DBService.Model(&models.StockBasic{}).Group("ts_code").Pluck("ts_code", &stocks)
	return
}

func GetIndustry() (industry []string) {
	impl.DBService.Model(&models.StockBasic{}).Group("industry").Pluck("industry", &industry)
	return
}

func GetBasic(code string) *models.StockBasic {
	var data StockData
	impl.DBService.Where("ts_code = ?", code).First(&data.Basic)
	return &data.Basic
}