feat(stock):添加美国股票基本信息初始化

- 增加美国股票基本信息的 JSON 文件
- 实现 initStockDataUS 函数用于初始化美国股票数据
- 在主程序中添加美国股票数据初始化的逻辑
This commit is contained in:
ArvinLovegood 2025-02-28 16:35:43 +08:00
parent fdca30ce3a
commit 2c2d689f53
2 changed files with 36516 additions and 0 deletions

36493
build/stock_base_info_us.json Normal file

File diff suppressed because it is too large Load Diff

23
main.go
View File

@ -44,6 +44,9 @@ var stocksBin []byte
//go:embed build/stock_base_info_hk.json
var stocksBinHK []byte
//go:embed build/stock_base_info_us.json
var stocksBinUS []byte
//go:generate cp -R ./data ./build/bin
var Version string
@ -67,6 +70,9 @@ func main() {
if stocksBinHK != nil && len(stocksBinHK) > 0 {
go initStockDataHK()
}
if stocksBinUS != nil && len(stocksBinUS) > 0 {
go initStockDataUS()
}
updateBasicInfo()
@ -182,6 +188,23 @@ func main() {
}
func initStockDataUS() {
var count int64
db.Dao.Model(&models.StockInfoUS{}).Count(&count)
if count > 0 {
return
}
var v []models.StockInfoUS
err := json.Unmarshal(stocksBinUS, &v)
if err != nil {
return
}
for _, item := range v {
db.Dao.Model(&models.StockInfoUS{}).Create(&item)
}
log.Printf("init stock data us %d", len(v))
}
func initStockDataHK() {
var count int64
db.Dao.Model(&models.StockInfoHK{}).Count(&count)