79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package tushare
|
||
|
||
import (
|
||
"time"
|
||
|
||
"git.apinb.com/quant/qsdk/conv"
|
||
)
|
||
|
||
/*
|
||
TradeCal 获取交易日历
|
||
|
||
exchange: 交易所代码，SSE 上交所，SZSE 深交所，BSE 北交所，空字符串代表所有
|
||
start_date: 开始日期，格式:YYYYMMDD
|
||
end_date: 结束日期，格式:YYYYMMDD
|
||
is_open: 是否开盘，'0'休市，'1'交易，空字符串代表所有
|
||
*/
|
||
func (cli *TushareClient) TradeCal(exchange, start_date, end_date, is_open string) (*TushareRespData, error) {
|
||
params := map[string]any{}
|
||
|
||
if exchange != "" {
|
||
params["exchange"] = exchange
|
||
}
|
||
if start_date != "" {
|
||
params["start_date"] = start_date
|
||
}
|
||
if end_date != "" {
|
||
params["end_date"] = end_date
|
||
}
|
||
if is_open != "" {
|
||
params["is_open"] = is_open
|
||
}
|
||
|
||
req := TushareReq{
|
||
APIName: "trade_cal",
|
||
Params: params,
|
||
}
|
||
|
||
fields := []map[string]string{
|
||
{"exchange": "交易所代码"},
|
||
{"cal_date": "日历日期"},
|
||
{"is_open": "是否开盘"},
|
||
{"pretrade_date": "上一交易日"},
|
||
}
|
||
|
||
return cli.Do(req, fields)
|
||
}
|
||
|
||
func (cli *TushareClient) LastTradeDay() string {
|
||
result, err := cli.TradeCal("SSE", time.Now().AddDate(0, 0, -15).Format("20060102"), time.Now().Format("20060102"), "1")
|
||
if err != nil {
|
||
return ""
|
||
}
|
||
cal := result.Map()
|
||
if len(cal) == 0 {
|
||
return ""
|
||
}
|
||
return conv.AnyToString(cal[0]["cal_date"])
|
||
}
|
||
|
||
func (cli *TushareClient) LimitTradeDay(limit int) []string {
|
||
if limit <= 0 {
|
||
return []string{}
|
||
}
|
||
if limit > 100 {
|
||
limit = 100
|
||
}
|
||
result, err := cli.TradeCal("SSE", time.Now().AddDate(0, 0, -100).Format("20060102"), time.Now().Format("20060102"), "1")
|
||
if err != nil {
|
||
return []string{}
|
||
}
|
||
cal := result.Map()
|
||
days := []string{}
|
||
for i := 0; i < limit; i++ {
|
||
item := cal[i]
|
||
days = append(days, conv.AnyToString(item["cal_date"]))
|
||
}
|
||
return days
|
||
}
|