fix bug
This commit is contained in:
@@ -1,16 +1,10 @@
|
||||
package a
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"log"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"git.apinb.com/bsm-sdk/core/utils"
|
||||
"git.apinb.com/dataset/stock/internal/impl"
|
||||
"git.apinb.com/dataset/stock/internal/models"
|
||||
tushare "github.com/ShawnRong/tushare-go"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -22,178 +16,10 @@ func NewApiClient() {
|
||||
TushareClient = tushare.New(TushareToken)
|
||||
}
|
||||
|
||||
func GetStockBasic() {
|
||||
// 参数
|
||||
params := map[string]string{
|
||||
"list_status": "L", // 上市状态 L上市 D退市 P暂停上市，默认是L
|
||||
}
|
||||
|
||||
// 字段
|
||||
fields := strings.Split("ts_code,symbol,name,area,industry,fullname,enname,cnspell,market,exchange,list_date,is_hs,act_name,act_ent_type", ",")
|
||||
|
||||
// 根据api 请求对应的接口
|
||||
reply, _ := TushareClient.StockBasic(params, fields)
|
||||
if reply.Code != 0 {
|
||||
log.Println("ERROR", "GetStockBasic", reply.Code, reply.Msg)
|
||||
return
|
||||
}
|
||||
|
||||
for _, item := range reply.Data.Items {
|
||||
var record models.StockBasic
|
||||
data := Anys2Strings(item)
|
||||
|
||||
err := impl.DBService.Where("ts_code=?", data[0]).First(&record).Error
|
||||
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||
// create
|
||||
record = models.StockBasic{
|
||||
TsCode: data[0],
|
||||
Symbol: data[1],
|
||||
Name: data[2],
|
||||
Area: data[3],
|
||||
Industry: data[4],
|
||||
FullName: data[5],
|
||||
EnName: data[6],
|
||||
CnSpell: data[7],
|
||||
Market: data[8],
|
||||
Exchange: data[9],
|
||||
ListDate: data[10],
|
||||
IsHS: data[11],
|
||||
ActName: data[12],
|
||||
ActEntType: data[13],
|
||||
}
|
||||
impl.DBService.Create(&record)
|
||||
} else if err == nil {
|
||||
// check updated.
|
||||
upd := make(map[string]any)
|
||||
if record.Name != data[2] {
|
||||
upd["name"] = data[2]
|
||||
}
|
||||
if record.Area != data[3] {
|
||||
upd["area"] = data[3]
|
||||
}
|
||||
if record.Industry != data[4] {
|
||||
upd["industry"] = data[4]
|
||||
}
|
||||
if record.FullName != data[5] {
|
||||
upd["full_name"] = data[5]
|
||||
}
|
||||
if record.EnName != data[6] {
|
||||
upd["en_name"] = data[6]
|
||||
}
|
||||
if record.CnSpell != data[7] {
|
||||
upd["cn_spell"] = data[7]
|
||||
}
|
||||
if record.Market != data[8] {
|
||||
upd["market"] = data[8]
|
||||
}
|
||||
if record.Exchange != data[9] {
|
||||
upd["exchange"] = data[9]
|
||||
}
|
||||
if record.ListDate != data[10] {
|
||||
upd["list_date"] = data[10]
|
||||
}
|
||||
if record.IsHS != data[11] {
|
||||
upd["is_hs"] = data[11]
|
||||
}
|
||||
if record.ActName != data[12] {
|
||||
upd["act_name"] = data[12]
|
||||
}
|
||||
if record.ActEntType != data[13] {
|
||||
upd["act_ent_type"] = data[13]
|
||||
}
|
||||
if len(upd) > 0 {
|
||||
impl.DBService.Model(record).Updates(upd)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func GetLastDaily() {
|
||||
s, e := ReturnLastDay()
|
||||
params := map[string]string{
|
||||
"start_date": s,
|
||||
"end_date": e,
|
||||
}
|
||||
|
||||
reply, _ := TushareClient.Daily(params, []string{})
|
||||
if reply.Code != 0 {
|
||||
log.Println("ERROR", "GetLastDaily", reply.Code, reply.Msg)
|
||||
return
|
||||
}
|
||||
|
||||
for _, item := range reply.Data.Items {
|
||||
var cnt int64
|
||||
impl.DBService.Model(&models.LastDaily{}).Where("ts_code=? and trade_date=?", item[0].(string), item[1].(string)).Count(&cnt)
|
||||
if cnt == 0 {
|
||||
impl.DBService.Create(&models.LastDaily{
|
||||
TsCode: item[0].(string),
|
||||
TradeDate: item[1].(string),
|
||||
Open: item[2].(float64),
|
||||
High: item[3].(float64),
|
||||
Low: item[4].(float64),
|
||||
Close: item[5].(float64),
|
||||
PreClose: item[6].(float64),
|
||||
Change: item[7].(float64),
|
||||
PctChg: item[8].(float64),
|
||||
Vol: item[9].(float64),
|
||||
Amount: item[10].(float64),
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func GetLastIndicator() {
|
||||
var stocks []models.StockBasic
|
||||
impl.DBService.Find(&stocks)
|
||||
|
||||
s, e := ReturnLastDay()
|
||||
for _, row := range stocks {
|
||||
params := map[string]string{
|
||||
"ts_code": row.TsCode,
|
||||
"start_date": s,
|
||||
"end_date": e,
|
||||
}
|
||||
|
||||
reply, _ := TushareClient.Daily(params, []string{})
|
||||
if reply.Code != 0 {
|
||||
log.Println("ERROR", "GetLastIndicator", reply.Code, reply.Msg)
|
||||
return
|
||||
}
|
||||
|
||||
for _, item := range reply.Data.Items {
|
||||
var cnt int64
|
||||
impl.DBService.Model(&models.LastIndicator{}).Where("ts_code=? and trade_date=?", item[0].(string), item[1].(string)).Count(&cnt)
|
||||
if cnt == 0 {
|
||||
impl.DBService.Create(&models.LastIndicator{
|
||||
TsCode: item[0].(string),
|
||||
TradeDate: item[1].(string),
|
||||
Close: item[2].(float64),
|
||||
TurnoverRate: item[3].(float64),
|
||||
TurnoverRateF: item[4].(float64),
|
||||
VolumeRatio: item[5].(float64),
|
||||
Pe: item[6].(float64),
|
||||
PeTtm: item[7].(float64),
|
||||
Pb: item[8].(float64),
|
||||
Ps: item[9].(float64),
|
||||
PsTtm: item[10].(float64),
|
||||
DvRatio: item[11].(float64),
|
||||
DvTtm: item[12].(float64),
|
||||
TotalShare: item[13].(float64),
|
||||
FloatShare: item[14].(float64),
|
||||
FreeShare: item[15].(float64),
|
||||
TotalMv: item[16].(float64),
|
||||
CircMv: item[17].(float64),
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func ReturnLastDay() (string, string) {
|
||||
now := time.Now()
|
||||
start := now.Format("20060102")
|
||||
end := now.AddDate(0, -1, 0).Format("20060102")
|
||||
end := now.Format("20060102")
|
||||
start := now.AddDate(0, 0, -1).Format("20060102")
|
||||
return start, end
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user