mirror of
https://github.com/ArvinLovegood/go-stock.git
synced 2025-07-19 00:00:09 +08:00
b2b0300aa1
- 新增 getUSStockPriceInfo 函数用于获取美股实时行情信息 - 修改 SearchStockPriceInfo 函数，支持美股代码查询 - 更新 Tushare 数据接口，增加对美股每日数据的支持 - 优化股票代码处理逻辑，兼容不同市场代码格式
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package data
|
|
|
|
import (
|
|
"github.com/duke-git/lancet/v2/convertor"
|
|
"github.com/duke-git/lancet/v2/slice"
|
|
"github.com/duke-git/lancet/v2/strutil"
|
|
"github.com/go-resty/resty/v2"
|
|
"go-stock/backend/logger"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// @Author spark
|
|
// @Date 2025/2/17 12:33
|
|
// @Desc
|
|
//-----------------------------------------------------------------------------------
|
|
|
|
type TushareApi struct {
|
|
client *resty.Client
|
|
config *Settings
|
|
}
|
|
|
|
func NewTushareApi(config *Settings) *TushareApi {
|
|
return &TushareApi{
|
|
client: resty.New(),
|
|
config: config,
|
|
}
|
|
}
|
|
|
|
// GetDaily tushare A股日线行情
|
|
func (receiver TushareApi) GetDaily(tsCode, startDate, endDate string, crawlTimeOut int64) string {
|
|
logger.SugaredLogger.Debugf("tushare daily request: ts_code=%s, start_date=%s, end_date=%s", tsCode, startDate, endDate)
|
|
fields := "ts_code,trade_date,open,high,low,close,pre_close,change,pct_chg,vol,amount"
|
|
resp := &TushareStockBasicResponse{}
|
|
stockType := getStockType(tsCode)
|
|
tsCodeNEW := getTsCode(tsCode)
|
|
logger.SugaredLogger.Debugf("tushare daily request: %s,tsCode:%s,tsCodeNEW:%s", stockType, tsCode, tsCodeNEW)
|
|
_, err := receiver.client.SetTimeout(time.Duration(crawlTimeOut)*time.Second).R().
|
|
SetHeader("content-type", "application/json").
|
|
SetBody(&TushareRequest{
|
|
ApiName: stockType,
|
|
Token: receiver.config.TushareToken,
|
|
Params: map[string]any{
|
|
"ts_code": tsCodeNEW,
|
|
"start_date": startDate,
|
|
"end_date": endDate,
|
|
},
|
|
Fields: fields}).
|
|
SetResult(resp).
|
|
Post(tushareApiUrl)
|
|
if err != nil {
|
|
logger.SugaredLogger.Error(err)
|
|
return ""
|
|
}
|
|
res := ""
|
|
if resp.Data.Items != nil && len(resp.Data.Items) > 0 {
|
|
fieldsStr := slice.JoinFunc(resp.Data.Fields, ",", func(s string) string {
|
|
return "\"" + convertor.ToString(s) + "\""
|
|
})
|
|
res += fieldsStr + "\n"
|
|
for _, item := range resp.Data.Items {
|
|
//logger.SugaredLogger.Debugf("%s", slice.Join(item, ","))
|
|
t := slice.JoinFunc(item, ",", func(s any) any {
|
|
return "\"" + convertor.ToString(s) + "\""
|
|
})
|
|
res += t + "\n"
|
|
}
|
|
}
|
|
logger.SugaredLogger.Debugf("tushare response: %s", res)
|
|
return res
|
|
}
|
|
|
|
func getTsCode(code string) any {
|
|
if strutil.HasPrefixAny(code, []string{"US", "us", "gb_"}) {
|
|
code = strings.Replace(code, "gb_", "", 1)
|
|
code = strings.Replace(code, "us", "", 1)
|
|
return code
|
|
}
|
|
return code
|
|
}
|
|
|
|
func getStockType(code string) string {
|
|
if strutil.HasSuffixAny(code, []string{"SZ", "SH", "sh", "sz"}) {
|
|
return "daily"
|
|
}
|
|
if strutil.HasSuffixAny(code, []string{"HK", "hk"}) {
|
|
return "hk_daily"
|
|
}
|
|
if strutil.HasPrefixAny(code, []string{"US", "us", "gb_"}) {
|
|
return "us_daily"
|
|
}
|
|
return ""
|
|
}
|