fix bug

internal/logic/a/basic.go

@@ -0,0 +1,103 @@
+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
+}