108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package schema
|
||
|
||
import "gorm.io/gorm"
|
||
|
||
// ScopeTsCode 按 ts_code 精确过滤；空字符串则不加条件。
|
||
func ScopeTsCode(tsCode string) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if tsCode == "" {
|
||
return db
|
||
}
|
||
return db.Where("ts_code = ?", tsCode)
|
||
}
|
||
}
|
||
|
||
// ScopeTsCodes 按 ts_code IN (...)；nil 或空切片不加条件。
|
||
func ScopeTsCodes(tsCodes []string) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if len(tsCodes) == 0 {
|
||
return db
|
||
}
|
||
return db.Where("ts_code IN ?", tsCodes)
|
||
}
|
||
}
|
||
|
||
// ScopeTradeDateEQ 按 trade_date 等于；0 表示不加条件。
|
||
func ScopeTradeDateEQ(tradeDate int) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if tradeDate == 0 {
|
||
return db
|
||
}
|
||
return db.Where("trade_date = ?", tradeDate)
|
||
}
|
||
}
|
||
|
||
// ScopeTradeDateBetween trade_date 区间 [start,end]；仅传一侧时做单边约束；均为 0 不加条件。
|
||
func ScopeTradeDateBetween(start, end int) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
switch {
|
||
case start > 0 && end > 0:
|
||
return db.Where("trade_date BETWEEN ? AND ?", start, end)
|
||
case start > 0:
|
||
return db.Where("trade_date >= ?", start)
|
||
case end > 0:
|
||
return db.Where("trade_date <= ?", end)
|
||
default:
|
||
return db
|
||
}
|
||
}
|
||
}
|
||
|
||
// ScopeStockDailyTsDate 日线：ts_code + 交易日（任一为空则该项不限制）。
|
||
func ScopeStockDailyTsDate(tsCode string, tradeDate int) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
return db.Scopes(ScopeTsCode(tsCode), ScopeTradeDateEQ(tradeDate))
|
||
}
|
||
}
|
||
|
||
// ScopeStockIndicatorTsDate 指标表：ts_code + trade_date。
|
||
func ScopeStockIndicatorTsDate(tsCode string, tradeDate int) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
return db.Scopes(ScopeTsCode(tsCode), ScopeTradeDateEQ(tradeDate))
|
||
}
|
||
}
|
||
|
||
// ScopeFinaTsPeriod 财务指标：ts_code + period（与 uniqueIndex un_fi_code_date 一致）；period 为 0 时不限制 period。
|
||
func ScopeFinaTsPeriod(tsCode string, period int) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
db = db.Scopes(ScopeTsCode(tsCode))
|
||
if period != 0 {
|
||
db = db.Where("period = ?", period)
|
||
}
|
||
return db
|
||
}
|
||
}
|
||
|
||
// ScopeBlocksIndexCode 板块 code。
|
||
func ScopeBlocksIndexCode(code string) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if code == "" {
|
||
return db
|
||
}
|
||
return db.Where("code = ?", code)
|
||
}
|
||
}
|
||
|
||
// ScopeBlocksMemberPair 板块 ti_code + 成分 stock_code。
|
||
func ScopeBlocksMemberPair(tiCode, stockCode string) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if tiCode != "" {
|
||
db = db.Where("ti_code = ?", tiCode)
|
||
}
|
||
if stockCode != "" {
|
||
db = db.Where("stock_code = ?", stockCode)
|
||
}
|
||
return db
|
||
}
|
||
}
|
||
|
||
// ScopeMoneyTotalCode 资金流 code。
|
||
func ScopeMoneyTotalCode(code string) func(*gorm.DB) *gorm.DB {
|
||
return func(db *gorm.DB) *gorm.DB {
|
||
if code == "" {
|
||
return db
|
||
}
|
||
return db.Where("code = ?", code)
|
||
}
|
||
}
|