quant/qsdk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 10 Wiki Activity

review code.

Browse Source
This commit is contained in:
yanweidong
2026-05-01 17:07:20 +08:00
parent b3565ac6d7
commit 5673e5c6ca
18 changed files with 359 additions and 208 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
schema/blocks_index.go
View File

@@ -1,28 +0,0 @@
package schema
import "gorm.io/gorm"
// BlocksIndex 板块索引（采用 dataset/stock 的 code 唯一约束）。
type BlocksIndex struct {
gorm.Model
Code string `gorm:"type:varchar(50);not null;default:'';comment:板块代码;uniqueIndex:uq_blocks_index_code" json:"code"`
Name string `gorm:"type:varchar(50);not null;default:'';comment:板块名称" json:"name"`
IsRecommend bool `gorm:"type:bool;not null;default:false;comment:是否推荐" json:"is_recommend"`
}
func (BlocksIndex) TableName() string {
return "blocks_index"
}
// Key 板块业务主键：code。
func (b *BlocksIndex) Key() string {
return b.Code
}
// DisplayName 展示名称：非空 name，否则回退 code。
func (b *BlocksIndex) DisplayName() string {
if b.Name != "" {
return b.Name
}
return b.Code
}

28
schema/blocks_member.go
View File

@@ -1,28 +0,0 @@
package schema
import "gorm.io/gorm"
// BlocksMember 板块成分股。
type BlocksMember struct {
gorm.Model
TiCode string `gorm:"type:varchar(50);not null;default:'';comment:板块代码;index" json:"ti_code"`
StockCode string `gorm:"type:varchar(50);not null;default:'';comment:股票代码;index" json:"stock_code"`
Weight float64 `gorm:"type:float;not null;default:0;comment:权重" json:"weight"`
}
func (BlocksMember) TableName() string {
return "blocks_member"
}
// Key 成分在板块内的键：ti_code + stock_code。
func (b *BlocksMember) Key() string {
if b.TiCode == "" && b.StockCode == "" {
return ""
}
return b.TiCode + "/" + b.StockCode
}
// HasWeight 是否带权重量化成分。
func (b *BlocksMember) HasWeight() bool {
return b.Weight > 0
}

14
schema/stock_basic.go → schema/dataset_basic.go
View File

@@ -2,8 +2,8 @@ package schema
import "gorm.io/gorm"
// StockBasic 股票基本信息表（合并 dataset/stock 与 gostock 字段；gostock 独有 Level、Desc）。
type StockBasic struct {
// DatasetStockBasic 股票基本信息表（合并 dataset/stock 与 gostock 字段；gostock 独有 Level、Desc）。
type DatasetStockBasic struct {
gorm.Model
TsCode string `gorm:"type:varchar(50);not null;index;comment:TS代码"`
Symbol string `gorm:"type:varchar(50);not null;comment:股票代码"`
@@ -21,17 +21,17 @@ type StockBasic struct {
ActEntType string `gorm:"type:varchar(50);not null;default:'';comment:实控人企业性质"`
}
func (StockBasic) TableName() string {
return "stock_basic"
func (DatasetStockBasic) TableName() string {
return "dataset_stock_basic"
}
// Key 业务主键：TS 代码。
func (s *StockBasic) Key() string {
func (s *DatasetStockBasic) Key() string {
return s.TsCode
}
// DisplaySymbol 展示用代码：有 symbol 用 symbol，否则用 ts_code。
func (s *StockBasic) DisplaySymbol() string {
func (s *DatasetStockBasic) DisplaySymbol() string {
if s.Symbol != "" {
return s.Symbol
}
@@ -39,6 +39,6 @@ func (s *StockBasic) DisplaySymbol() string {
}
// IsNorthbound 是否沪深港通标的（H 沪股通 / S 深股通）。
func (s *StockBasic) IsNorthbound() bool {
func (s *DatasetStockBasic) IsNorthbound() bool {
return s.IsHS == "H" || s.IsHS == "S"
}

28
schema/dataset_blocks_index.go Normal file
View File

@@ -0,0 +1,28 @@
package schema
import "gorm.io/gorm"
// DatasetBlocksIndex 板块索引（采用 dataset/stock 的 code 唯一约束）。
type DatasetBlocksIndex struct {
gorm.Model
TiCode string `gorm:"type:varchar(50);not null;default:'';comment:指数代码;uniqueIndex:uq_blocks_index_code" json:"ti_code"`
Name string `gorm:"type:varchar(50);not null;default:'';comment:板块名称" json:"name"`
IsRecommend bool `gorm:"type:bool;not null;default:false;comment:是否推荐" json:"is_recommend"`
}
func (DatasetBlocksIndex) TableName() string {
return "dataset_blocks_index"
}
// Key 板块业务主键：code。
func (b *DatasetBlocksIndex) Key() string {
return b.TiCode
}
// DisplayName 展示名称：非空 name，否则回退 code。
func (b *DatasetBlocksIndex) DisplayName() string {
if b.Name != "" {
return b.Name
}
return b.TiCode
}

28
schema/dataset_blocks_member.go Normal file
View File

@@ -0,0 +1,28 @@
package schema
import "gorm.io/gorm"
// DatasetBlocksMember 板块成分股。
type DatasetBlocksMember struct {
gorm.Model
TiCode string `gorm:"type:varchar(50);not null;default:'';comment:板块代码;index" json:"ti_code"`
TsCode string `gorm:"type:varchar(50);not null;default:'';comment:股票代码;index" json:"ts_code"`
Weight float64 `gorm:"type:float;not null;default:0;comment:权重" json:"weight"`
}
func (DatasetBlocksMember) TableName() string {
return "dataset_blocks_member"
}
// Key 成分在板块内的键：ti_code + ts_code。
func (b *DatasetBlocksMember) Key() string {
if b.TiCode == "" && b.TsCode == "" {
return ""
}
return b.TiCode + "/" + b.TsCode
}
// HasWeight 是否带权重量化成分。
func (b *DatasetBlocksMember) HasWeight() bool {
return b.Weight > 0
}

18
schema/stock_daily.go → schema/dataset_daily.go
View File

@@ -2,8 +2,8 @@ package schema
import "strconv"
// StockDaily 股票日线数据（两仓库结构一致）。
type StockDaily struct {
// DatasetStockDaily 股票日线数据（两仓库结构一致）。
type DatasetStockDaily struct {
ID uint `gorm:"primarykey;autoIncrement" json:"id"`
TsCode string `gorm:"type:varchar(20);not null;index:idx_ts_code;uniqueIndex:un_code_date;comment:股票代码" json:"ts_code"`
TradeDate int `gorm:"index:idx_trade_date;uniqueIndex:un_code_date;comment:交易日期" json:"trade_date"`
@@ -19,12 +19,12 @@ type StockDaily struct {
Amount float64 `gorm:"type:decimal(20,2);comment:成交额(千元)" json:"amount"`
}
func (StockDaily) TableName() string {
return "stock_daily"
func (DatasetStockDaily) TableName() string {
return "dataset_stock_daily"
}
// Key 业务主键：ts_code + 交易日。
func (d *StockDaily) Key() string {
func (d *DatasetStockDaily) Key() string {
if d.TsCode == "" && d.TradeDate == 0 {
return ""
}
@@ -32,17 +32,17 @@ func (d *StockDaily) Key() string {
}
// IsRising 是否收涨（昨收有效且收盘高于昨收）。
func (d *StockDaily) IsRising() bool {
func (d *DatasetStockDaily) IsRising() bool {
return d.PreClose > 0 && d.Close > d.PreClose
}
// IsFalling 是否收跌。
func (d *StockDaily) IsFalling() bool {
func (d *DatasetStockDaily) IsFalling() bool {
return d.PreClose > 0 && d.Close < d.PreClose
}
// PctChangeFromPre 由昨收计算的涨跌幅（%）；昨收无效时返回 0。
func (d *StockDaily) PctChangeFromPre() float64 {
func (d *DatasetStockDaily) PctChangeFromPre() float64 {
if d.PreClose <= 0 {
return 0
}
@@ -50,7 +50,7 @@ func (d *StockDaily) PctChangeFromPre() float64 {
}
// AmplitudePct 振幅（相对昨收，%）：(最高-最低)/昨收*100。
func (d *StockDaily) AmplitudePct() float64 {
func (d *DatasetStockDaily) AmplitudePct() float64 {
if d.PreClose <= 0 {
return 0
}

12
schema/stock_fina_indicator.go → schema/dataset_fina_indicator.go
View File

@@ -6,8 +6,8 @@ import (
"gorm.io/gorm"
)
// StockFinaIndicator 财务指标模型
type StockFinaIndicator struct {
// DatasetStockFinaIndicator 财务指标模型
type DatasetStockFinaIndicator struct {
gorm.Model
TsCode string `gorm:"type:varchar(20);not null;index:fi_ts_code;uniqueIndex:un_fi_code_date;comment:TS代码"`
Period int `gorm:"index:idx_period;uniqueIndex:un_fi_code_date;comment:报告期数"`
@@ -205,12 +205,12 @@ type StockFinaIndicator struct {
}
// TableName 设置表名
func (StockFinaIndicator) TableName() string {
return "fina_indicator"
func (DatasetStockFinaIndicator) TableName() string {
return "dataset_fina_indicator"
}
// Key 与表 uniqueIndex un_fi_code_date 一致：ts_code + period。
func (f *StockFinaIndicator) Key() string {
func (f *DatasetStockFinaIndicator) Key() string {
if f.TsCode == "" && f.Period == 0 {
return ""
}
@@ -218,7 +218,7 @@ func (f *StockFinaIndicator) Key() string {
}
// RowLabel 便于日志/调试：ts_code + 报告期 end_date + 公告 ann_date。
func (f *StockFinaIndicator) RowLabel() string {
func (f *DatasetStockFinaIndicator) RowLabel() string {
if f.EndDate != "" || f.AnnDate != "" {
return f.TsCode + " end=" + f.EndDate + " ann=" + f.AnnDate
}

14
schema/stock_indicator.go → schema/dataset_indicator.go
View File

@@ -2,8 +2,8 @@ package schema
import "strconv"
// StockIndicator 每日基本面指标（采用 dataset/stock 的 decimal 定义，与采集端迁移一致）。
type StockIndicator struct {
// DatasetStockIndicator 每日基本面指标（采用 dataset/stock 的 decimal 定义，与采集端迁移一致）。
type DatasetStockIndicator struct {
ID uint `gorm:"primarykey;autoIncrement"`
TsCode string `gorm:"type:varchar(20);not null;index:si_ts_code;uniqueIndex:un_si_code_date;comment:股票代码" json:"ts_code"`
TradeDate int `gorm:"index:si_trade_date;uniqueIndex:un_si_code_date;comment:交易日期" json:"trade_date"`
@@ -26,12 +26,12 @@ type StockIndicator struct {
Roe float64 `gorm:"type:decimal(20,4);comment:ROE(%) 净利润/股本"`
}
func (StockIndicator) TableName() string {
return "stock_indicator"
func (DatasetStockIndicator) TableName() string {
return "dataset_stock_indicator"
}
// Key 业务主键：ts_code + 交易日。
func (s *StockIndicator) Key() string {
func (s *DatasetStockIndicator) Key() string {
if s.TsCode == "" && s.TradeDate == 0 {
return ""
}
@@ -39,11 +39,11 @@ func (s *StockIndicator) Key() string {
}
// HasTotalMV 总市值是否已填充（大于 0）。
func (s *StockIndicator) HasTotalMV() bool {
func (s *DatasetStockIndicator) HasTotalMV() bool {
return s.TotalMv > 0
}
// HasCircMV 流通市值是否已填充。
func (s *StockIndicator) HasCircMV() bool {
func (s *DatasetStockIndicator) HasCircMV() bool {
return s.CircMv > 0
}

66
schema/dataset_indicator_pro.go Normal file
View File

@@ -0,0 +1,66 @@
package schema
import "strconv"
// DatasetIndicatorPro 对应 Tushare stk_factor_pro 当前请求字段集（见 tushare/indicator.go 的 fields 列表）。
type DatasetIndicatorPro struct {
ID uint `gorm:"primarykey;autoIncrement"`
TsCode string `gorm:"type:varchar(20);not null;index:dip_ts_code;uniqueIndex:un_dip_code_date;comment:股票代码" json:"ts_code"`
TradeDate int `gorm:"index:dip_trade_date;uniqueIndex:un_dip_code_date;comment:交易日期" json:"trade_date"`
Open float64 `gorm:"type:decimal(20,4);comment:开盘价"`
High float64 `gorm:"type:decimal(20,4);comment:最高价"`
Low float64 `gorm:"type:decimal(20,4);comment:最低价"`
Close float64 `gorm:"type:decimal(20,4);comment:收盘价"`
PreClose float64 `gorm:"type:decimal(20,4);comment:昨收价"`
Change float64 `gorm:"type:decimal(20,4);comment:涨跌额"`
PctChg float64 `gorm:"type:decimal(20,6);comment:涨跌幅%"`
Vol float64 `gorm:"type:decimal(20,2);comment:成交量(手)"`
Amount float64 `gorm:"type:decimal(20,2);comment:成交额(千元)"`
TurnoverRate float64 `gorm:"type:decimal(20,4);comment:换手率(%)"`
TurnoverRateF float64 `gorm:"type:decimal(20,4);comment:换手率(自由流通股)"`
VolumeRatio float64 `gorm:"type:decimal(20,4);comment:量比"`
Pe float64 `gorm:"type:decimal(20,4);comment:市盈率"`
PeTtm float64 `gorm:"type:decimal(20,4);comment:市盈率TTM"`
Pb float64 `gorm:"type:decimal(20,4);comment:市净率"`
Ps float64 `gorm:"type:decimal(20,4);comment:市销率"`
PsTtm float64 `gorm:"type:decimal(20,4);comment:市销率TTM"`
DvRatio float64 `gorm:"type:decimal(20,4);comment:股息率(%)"`
DvTtm float64 `gorm:"type:decimal(20,4);comment:股息率TTM(%)"`
TotalShare float64 `gorm:"type:decimal(20,4);comment:总股本(万股)"`
FloatShare float64 `gorm:"type:decimal(20,4);comment:流通股本(万股)"`
FreeShare float64 `gorm:"type:decimal(20,4);comment:自由流通股本(万股)"`
TotalMv float64 `gorm:"type:decimal(20,4);comment:总市值(万元)"`
CircMv float64 `gorm:"type:decimal(20,4);comment:流通市值(万元)"`
AdjFactor float64 `gorm:"type:decimal(20,6);comment:复权因子"`
MaBfq5 float64 `gorm:"type:decimal(20,6);comment:MA5不复权"`
MaBfq10 float64 `gorm:"type:decimal(20,6);comment:MA10不复权"`
MaBfq20 float64 `gorm:"type:decimal(20,6);comment:MA20不复权"`
MaBfq60 float64 `gorm:"type:decimal(20,6);comment:MA60不复权"`
EmaBfq5 float64 `gorm:"type:decimal(20,6);comment:EMA5不复权"`
EmaBfq10 float64 `gorm:"type:decimal(20,6);comment:EMA10不复权"`
EmaBfq20 float64 `gorm:"type:decimal(20,6);comment:EMA20不复权"`
MacdBfq float64 `gorm:"type:decimal(20,6);comment:MACD不复权"`
MacdDifBfq float64 `gorm:"type:decimal(20,6);comment:MACD DIF不复权"`
MacdDeaBfq float64 `gorm:"type:decimal(20,6);comment:MACD DEA不复权"`
RsiBfq6 float64 `gorm:"type:decimal(20,6);comment:RSI6不复权"`
RsiBfq12 float64 `gorm:"type:decimal(20,6);comment:RSI12不复权"`
RsiBfq24 float64 `gorm:"type:decimal(20,6);comment:RSI24不复权"`
KdjKBfq float64 `gorm:"type:decimal(20,6);comment:KDJ-K不复权"`
KdjDBfq float64 `gorm:"type:decimal(20,6);comment:KDJ-D不复权"`
KdjBfq float64 `gorm:"type:decimal(20,6);comment:KDJ-J不复权"`
BollUpperBfq float64 `gorm:"type:decimal(20,6);comment:BOLL上轨不复权"`
BollMidBfq float64 `gorm:"type:decimal(20,6);comment:BOLL中轨不复权"`
BollLowerBfq float64 `gorm:"type:decimal(20,6);comment:BOLL下轨不复权"`
}
func (DatasetIndicatorPro) TableName() string {
return "dataset_indicator_pro"
}
// Key 业务主键：ts_code + 交易日。
func (s *DatasetIndicatorPro) Key() string {
if s.TsCode == "" && s.TradeDate == 0 {
return ""
}
return s.TsCode + "#" + strconv.Itoa(s.TradeDate)
}

31
schema/dataset_money_total.go Normal file
View File

@@ -0,0 +1,31 @@
package schema
// DatasetMoneyTotal 资金流汇总（采用 dataset/stock 的索引定义）。
type DatasetMoneyTotal struct {
ID uint `gorm:"primarykey"`
TsCode string `gorm:"type:varchar(20);not null;uniqueIndex:uq_money_total_ts_code"`
Last1DayMfAmount float64
Last3DayMfAmount float64
Last1DayTotalAmount float64
Last3DayTotalAmount float64
IsGreaterPervious bool
}
func (DatasetMoneyTotal) TableName() string {
return "dataset_money_total"
}
// Key 与唯一索引 uq_money_total_ts_code 一致。
func (m *DatasetMoneyTotal) Key() string {
return m.TsCode
}
// NetFlow1Day 最近 1 日主力净流入（万元），与字段语义一致。
func (m *DatasetMoneyTotal) NetFlow1Day() float64 {
return m.Last1DayMfAmount
}
// NetFlow3Day 最近 3 日主力净流入（万元）。
func (m *DatasetMoneyTotal) NetFlow3Day() float64 {
return m.Last3DayMfAmount
}

16
schema/pledge_stat.go → schema/dataset_pledge_stat.go
View File

@@ -1,7 +1,7 @@
package schema
// PledgeStat 股权质押统计（表 pledge_stat；dataset/stock 中原为 PledgeStatModel，业务层 Save 请留在各应用内）。
type PledgeStat struct {
// DatasetPledgeStat 股权质押统计（表 dataset_pledge_stat；dataset/stock 中原为 PledgeStatModel，业务层 Save 请留在各应用内）。
type DatasetPledgeStat struct {
ID uint `gorm:"primarykey" json:"id"`
TsCode string `gorm:"type:varchar(50);not null;default:'';comment:股票代码;uniqueIndex:uq_pledge_stat_ts_code" json:"ts_code"`
EndDate int `gorm:"type:int;not null;default:0;comment:截止日期" json:"end_date"`
@@ -12,19 +12,19 @@ type PledgeStat struct {
PledgeRatio float64 `json:"pledge_ratio"`
}
func (PledgeStat) TableName() string {
return "pledge_stat"
func (DatasetPledgeStat) TableName() string {
return "dataset_pledge_stat"
}
// Key 与唯一约束 uq_pledge_stat_ts_code 一致：ts_code。
func (p *PledgeStat) Key() string {
func (p *DatasetPledgeStat) Key() string {
return p.TsCode
}
// HasPledgeFacts 是否具备任一质押统计字段（比例、次数或股本）。
func (p *PledgeStat) HasPledgeFacts() bool {
func (p *DatasetPledgeStat) HasPledgeFacts() bool {
return p.PledgeRatio > 0 || p.PledgeCount > 0 || p.TotalShare > 0
}
// PledgeStatModel 兼容 dataset/stock 中的类型名。
type PledgeStatModel = PledgeStat
// DatasetPledgeStatModel 兼容 dataset/stock 中的类型名。
type DatasetPledgeStatModel = DatasetPledgeStat

31
schema/money_total.go
View File

@@ -1,31 +0,0 @@
package schema
// MoneyTotal 资金流汇总（采用 dataset/stock 的索引定义）。
type MoneyTotal struct {
ID uint `gorm:"primarykey"`
Code string `gorm:"type:varchar(20);not null;uniqueIndex:uq_money_total_code"`
Last1DayMfAmount float64
Last3DayMfAmount float64
Last1DayTotalAmount float64
Last3DayTotalAmount float64
IsGreaterPervious bool
}
func (MoneyTotal) TableName() string {
return "money_total"
}
// Key 与唯一索引 uq_money_total_code 一致。
func (m *MoneyTotal) Key() string {
return m.Code
}
// NetFlow1Day 最近 1 日主力净流入（万元），与字段语义一致。
func (m *MoneyTotal) NetFlow1Day() float64 {
return m.Last1DayMfAmount
}
// NetFlow3Day 最近 3 日主力净流入（万元）。
func (m *MoneyTotal) NetFlow3Day() float64 {
return m.Last3DayMfAmount
}

17
schema/register.go
View File

@@ -5,14 +5,15 @@ import "git.apinb.com/bsm-sdk/core/database"
// RegisterAutoMigrate 将本包内与 stock/gostock 共用的表注册到 bsm-sdk 的迁移列表（可选；也可在各应用 init 中自行 AppendMigrate）。
func RegisterAutoMigrate() {
for _, t := range []any{
&StockBasic{},
&StockDaily{},
&BlocksIndex{},
&BlocksMember{},
&MoneyTotal{},
&PledgeStat{},
&StockIndicator{},
&StockFinaIndicator{},
&DatasetStockBasic{},
&DatasetStockDaily{},
&DatasetBlocksIndex{},
&DatasetBlocksMember{},
&DatasetMoneyTotal{},
&DatasetPledgeStat{},
&DatasetStockIndicator{},
&DatasetIndicatorPro{},
&DatasetStockFinaIndicator{},
} {
database.AppendMigrate(t)
}