review code.
This commit is contained in:
29
cmd/main.go
29
cmd/main.go
@@ -71,20 +71,18 @@ func main() {
|
||||
|
||||
// runCollection 执行一次数据采集和存储
|
||||
func runCollection(coll *collector.Collector, store *storage.Storage) {
|
||||
log.Println("开始采集...")
|
||||
|
||||
// 计算Ymd
|
||||
// 检查是否为开市时间
|
||||
now := time.Now()
|
||||
ymd := now.Year()*10000 + int(now.Month())*100 + now.Day()
|
||||
if !collector.IsTradingTime(now) {
|
||||
return
|
||||
}
|
||||
|
||||
log.Println("开始采集...")
|
||||
|
||||
// 采集数据并检查变化
|
||||
status, dataHash, changed, err := coll.CollectAndCheck()
|
||||
if err != nil {
|
||||
log.Printf("采集失败: %v", err)
|
||||
// 记录失败的日志
|
||||
if err := store.SaveCollectionLog("", ymd, false, err.Error()); err != nil {
|
||||
log.Printf("保存采集日志失败: %v", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
@@ -94,27 +92,20 @@ func runCollection(coll *collector.Collector, store *storage.Storage) {
|
||||
// 如果数据没有变化,只记录日志
|
||||
if !changed {
|
||||
log.Println("数据未变化,跳过存储")
|
||||
if err := store.SaveCollectionLog(dataHash, ymd, false, "数据未变化"); err != nil {
|
||||
log.Printf("保存采集日志失败: %v", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 数据有变化,保存到数据库
|
||||
log.Println("数据已变化,开始存储到数据库...")
|
||||
if err := store.SaveStatus(status, dataHash); err != nil {
|
||||
if err := store.SaveData(status); err != nil {
|
||||
log.Printf("保存数据失败: %v", err)
|
||||
// 记录失败的日志
|
||||
if err := store.SaveCollectionLog(dataHash, ymd, true, err.Error()); err != nil {
|
||||
log.Printf("保存采集日志失败: %v", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 记录成功的日志
|
||||
if err := store.SaveCollectionLog(dataHash, ymd, true, "数据保存成功"); err != nil {
|
||||
log.Printf("保存采集日志失败: %v", err)
|
||||
}
|
||||
// if err := store.SaveCollectionLog(dataHash, ymd, true, "数据保存成功"); err != nil {
|
||||
// log.Printf("保存采集日志失败: %v", err)
|
||||
// }
|
||||
|
||||
log.Printf("数据存储成功 - 资产账户: %s, 订单数: %d, 持仓数: %d, 行情数: %d",
|
||||
status.Data.Assets.AccountID,
|
||||
|
||||
Reference in New Issue
Block a user