fix bug

internal/logic/a/indicator.go

@@ -2,10 +2,10 @@ package a
 
 import (
 	"log"
-	"os"
 	"strings"
 	"time"
 
+	"git.apinb.com/bsm-sdk/core/utils"
 	"git.apinb.com/dataset/stock/internal/impl"
 	"git.apinb.com/dataset/stock/internal/models"
 )
@@ -17,7 +17,7 @@ func GetStockIndicator() {
 		var last models.StockIndicator
 		err := impl.DBService.Where("ts_code=?", code).Order("trade_date DESC").First(&last).Error
 		if err == nil {
-			start = last.TradeDate.Format("20060102")
+			start = utils.Int2String(last.TradeDate)
 		}
 		params := map[string]string{
 			"ts_code":    code,
@@ -36,16 +36,17 @@ func GetStockIndicator() {
 			return
 		}
 
+		records := make([]*models.StockIndicator, 0)
 		for _, item := range reply.Data.Items {
+			t := utils.String2Int(item[1].(string))
+			if t == 0 {
+				continue
+			}
+
 			var cnt int64
-			impl.DBService.Model(&models.StockIndicator{}).Where("ts_code=? and trade_date=?", item[0].(string), item[1].(string)).Count(&cnt)
-			os.Exit(1)
+			impl.DBService.Model(&models.StockIndicator{}).Where("ts_code=? and trade_date=?", item[0].(string), t).Count(&cnt)
 			if cnt == 0 {
-				t, err := time.Parse("20060102", item[1].(string))
-				if err != nil {
-					continue
-				}
-				impl.DBService.Create(&models.StockIndicator{
+				records = append(records, &models.StockIndicator{
 					TsCode:        item[0].(string),
 					TradeDate:     t,
 					Close:         Any2Float(item[2]),
@@ -67,5 +68,11 @@ func GetStockIndicator() {
 				})
 			}
 		}
+
+		if len(records) > 0 {
+			impl.DBService.CreateInBatches(records, 100)
+		}
+
+		time.Sleep(200 * time.Microsecond)
 	}
 }