104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
|
package a
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"errors"
|
|||
|
|
"log"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"git.apinb.com/dataset/stock/internal/impl"
|
|||
|
|
"git.apinb.com/dataset/stock/internal/models"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
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 GetStockCodes() []string {
|
|||
|
|
var col []string
|
|||
|
|
impl.DBService.Model(&models.StockBasic{}).Pluck("ts_code", &col)
|
|||
|
|
return col
|
|||
|
|
}
|