refactor(backend): 移除股票数据 API 中的冗余代码

- 删除了 GetIndexBasic 和 GetStockBaseInfo 方法中的冗余代码
- 移除了不必要的文件写入操作和注释掉的代码
- 优化了代码结构,提高了代码的可读性和维护性
This commit is contained in:
spark 2025-01-03 09:57:19 +08:00
parent 513cd69e3e
commit a35b42f831

View File

@ -168,7 +168,7 @@ func NewStockDataApi() *StockDataApi {
func (receiver StockDataApi) GetIndexBasic() { func (receiver StockDataApi) GetIndexBasic() {
res := &TushareStockBasicResponse{} res := &TushareStockBasicResponse{}
fields := "ts_code,name,market,publisher,category,base_date,base_point,list_date,fullname,index_type,weight_rule,desc" fields := "ts_code,name,market,publisher,category,base_date,base_point,list_date,fullname,index_type,weight_rule,desc"
resp, err := receiver.client.R(). _, err := receiver.client.R().
SetHeader("content-type", "application/json"). SetHeader("content-type", "application/json").
SetBody(&TushareRequest{ SetBody(&TushareRequest{
ApiName: "index_basic", ApiName: "index_basic",
@ -185,12 +185,11 @@ func (receiver StockDataApi) GetIndexBasic() {
logger.SugaredLogger.Error(res.Msg) logger.SugaredLogger.Error(res.Msg)
return return
} }
ioutil.WriteFile("index_basic.json", resp.Body(), 0666) //ioutil.WriteFile("index_basic.json", resp.Body(), 0666)
for _, item := range res.Data.Items { for _, item := range res.Data.Items {
data := map[string]any{} data := map[string]any{}
for _, field := range strings.Split(fields, ",") { for _, field := range strings.Split(fields, ",") {
logger.SugaredLogger.Infof("field: %s", field)
idx := slice.IndexOf(res.Data.Fields, field) idx := slice.IndexOf(res.Data.Fields, field)
if idx == -1 { if idx == -1 {
continue continue
@ -213,7 +212,7 @@ func (receiver StockDataApi) GetIndexBasic() {
func (receiver StockDataApi) GetStockBaseInfo() { func (receiver StockDataApi) GetStockBaseInfo() {
res := &TushareStockBasicResponse{} res := &TushareStockBasicResponse{}
fields := "ts_code,symbol,name,area,industry,cnspell,market,list_date,act_name,act_ent_type,fullname,exchange,list_status,curr_type,enname,delist_date,is_hs" fields := "ts_code,symbol,name,area,industry,cnspell,market,list_date,act_name,act_ent_type,fullname,exchange,list_status,curr_type,enname,delist_date,is_hs"
resp, err := receiver.client.R(). _, err := receiver.client.R().
SetHeader("content-type", "application/json"). SetHeader("content-type", "application/json").
SetBody(&TushareRequest{ SetBody(&TushareRequest{
ApiName: "stock_basic", ApiName: "stock_basic",
@ -225,7 +224,7 @@ func (receiver StockDataApi) GetStockBaseInfo() {
Post(tushare_api_url) Post(tushare_api_url)
//logger.SugaredLogger.Infof("GetStockBaseInfo %s", string(resp.Body())) //logger.SugaredLogger.Infof("GetStockBaseInfo %s", string(resp.Body()))
//resp.Body()写入文件 //resp.Body()写入文件
ioutil.WriteFile("stock_basic.json", resp.Body(), 0666) //ioutil.WriteFile("stock_basic.json", resp.Body(), 0666)
//logger.SugaredLogger.Infof("GetStockBaseInfo %+v", res) //logger.SugaredLogger.Infof("GetStockBaseInfo %+v", res)
if err != nil { if err != nil {
logger.SugaredLogger.Error(err.Error()) logger.SugaredLogger.Error(err.Error())