39 lines
1.1 KiB
Go
39 lines
1.1 KiB
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
|
|
}
|