mirror of
https://github.com/ArvinLovegood/go-stock.git
synced 2025-07-19 00:00:09 +08:00
c81b1a730d
- 新增 TushareApi 结构体和 GetDaily 方法，用于获取 A 股日线行情数据 - 在 openai_api.go 中添加获取股票日 K线数据的协程 - 在 utils.go 中添加股票代码与 tushare 代码相互转换的函数 - 更新相关测试文件以支持新功能
66 lines
1.8 KiB
Go
66 lines
1.8 KiB
Go
package data
|
|
|
|
import (
|
|
"github.com/duke-git/lancet/v2/convertor"
|
|
"github.com/duke-git/lancet/v2/slice"
|
|
"github.com/go-resty/resty/v2"
|
|
"go-stock/backend/logger"
|
|
)
|
|
|
|
// @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) 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{}
|
|
_, err := receiver.client.R().
|
|
SetHeader("content-type", "application/json").
|
|
SetBody(&TushareRequest{
|
|
ApiName: "daily",
|
|
Token: receiver.config.TushareToken,
|
|
Params: map[string]any{
|
|
"ts_code": tsCode,
|
|
"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
|
|
}
|