From 2bab17d0439660b75795bd78814f7737039d1111 Mon Sep 17 00:00:00 2001
From: ZhouYongyou <128128010+zhouyongyou@users.noreply.github.com>
Date: Tue, 4 Nov 2025 21:02:26 +0800
Subject: [PATCH] =?UTF-8?q?fix(trader):=20add=20safety=20checks=20for=20ba?=
 =?UTF-8?q?lance=20sync=20##=20=E4=BF=AE=E5=A4=8D=E5=86=85=E5=AE=B9=20###?=
 =?UTF-8?q?=201.=20=E9=98=B2=E6=AD=A2=E9=99=A4=E4=BB=A5=E9=9B=B6panic=20?=
 =?UTF-8?q?=EF=BC=88=E4=B8=A5=E9=87=8Dbug=E4=BF=AE=E5=A4=8D=EF=BC=89=20-?=
 =?UTF-8?q?=20=E5=9C=A8=E8=AE=A1=E7=AE=97=E5=8F=98=E5=8C=96=E7=99=BE?=
 =?UTF-8?q?=E5=88=86=E6=AF=94=E5=89=8D=E6=A3=80=E6=9F=A5=20oldBalance=20<?=
 =?UTF-8?q?=3D=200=20-=20=E5=A6=82=E6=9E=9C=E5=88=9D=E5=A7=8B=E4=BD=99?=
 =?UTF-8?q?=E9=A2=9D=E6=97=A0=E6=95=88=EF=BC=8C=E7=9B=B4=E6=8E=A5=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0=E4=B8=BA=E5=AE=9E=E9=99=85=E4=BD=99=E9=A2=9D=20-=20?=
 =?UTF-8?q?=E9=81=BF=E5=85=8D=20division=20by=20zero=20panic=20###=202.=20?=
 =?UTF-8?q?=E5=A2=9E=E5=BC=BA=E9=94=99=E8=AF=AF=E5=A4=84=E7=90=86=20-=20?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=95=B0=E6=8D=AE=E5=BA=93=E7=B1=BB=E5=9E=8B?=
 =?UTF-8?q?=E6=96=AD=E8=A8=80=E5=A4=B1=E8=B4=A5=E7=9A=84=E6=97=A5=E5=BF=97?=
 =?UTF-8?q?=20-=20=E6=B7=BB=E5=8A=A0=E6=95=B0=E6=8D=AE=E5=BA=93=E4=B8=BAni?=
 =?UTF-8?q?l=E7=9A=84=E8=AD=A6=E5=91=8A=E6=97=A5=E5=BF=97=20-=20=E6=8F=90?=
 =?UTF-8?q?=E4=BE=9B=E6=9B=B4=E5=AE=8C=E6=95=B4=E7=9A=84=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=20##=20=E6=8A=80=E6=9C=AF=E7=BB=86=E8=8A=82?=
 =?UTF-8?q?=20=E9=97=AE=E9=A2=98=E5=9C=BA=E6=99=AF=EF=BC=9A=E5=A6=82?=
 =?UTF-8?q?=E6=9E=9C=20oldBalance=20=3D=200=EF=BC=8C=E8=AE=A1=E7=AE=97=20c?=
 =?UTF-8?q?hangePercent=20=E4=BC=9A=20panic=20=E4=BF=AE=E5=A4=8D=E5=90=8E?=
 =?UTF-8?q?=EF=BC=9A=E5=9C=A8=E8=AE=A1=E7=AE=97=E5=89=8D=E6=A3=80=E6=9F=A5?=
 =?UTF-8?q?=20oldBalance=20<=3D=200=EF=BC=8C=E7=9B=B4=E6=8E=A5=E6=9B=B4?=
 =?UTF-8?q?=E6=96=B0=E4=BD=99=E9=A2=9D=20##=20=E5=AE=A1=E6=9F=A5=E5=8F=91?=
 =?UTF-8?q?=E7=8E=B0=20-=20P0:=20=E9=99=A4=E4=BB=A5=E9=9B=B6=E9=A3=8E?=
 =?UTF-8?q?=E9=99=A9=EF=BC=88=E5=B7=B2=E4=BF=AE=E5=A4=8D=EF=BC=89=20-=20P1?=
 =?UTF-8?q?:=20=E7=B1=BB=E5=9E=8B=E6=96=AD=E8=A8=80=E5=A4=B1=E8=B4=A5?=
 =?UTF-8?q?=E6=9C=AA=E8=AE=B0=E5=BD=95=EF=BC=88=E5=B7=B2=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=EF=BC=89=20-=20P1:=20=E6=95=B0=E6=8D=AE=E5=BA=93=E4=B8=BAnil?=
 =?UTF-8?q?=E6=9C=AA=E8=AD=A6=E5=91=8A=EF=BC=88=E5=B7=B2=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=EF=BC=89=20=E8=AF=A6=E7=BB=86=E5=AE=A1=E6=9F=A5=E6=8A=A5?=
 =?UTF-8?q?=E5=91=8A=EF=BC=9Acode=5Freview=5Fauto=5Fbalance=5Fsync.md?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 trader/auto_trader.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/trader/auto_trader.go b/trader/auto_trader.go
index de7feda3..6a3fb222 100644
--- a/trader/auto_trader.go
+++ b/trader/auto_trader.go
@@ -292,6 +292,31 @@ func (at *AutoTrader) autoSyncBalanceIfNeeded() {
 	}
 
 	oldBalance := at.initialBalance
+
+	// 防止除以零：如果初始余额无效，直接更新为实际余额
+	if oldBalance <= 0 {
+		log.Printf("⚠️ [%s] 初始余额无效 (%.2f)，直接更新为实际余额 %.2f USDT", at.name, oldBalance, actualBalance)
+		at.initialBalance = actualBalance
+		if at.database != nil {
+			type DatabaseUpdater interface {
+				UpdateTraderInitialBalance(userID, id string, newBalance float64) error
+			}
+			if db, ok := at.database.(DatabaseUpdater); ok {
+				if err := db.UpdateTraderInitialBalance(at.userID, at.id, actualBalance); err != nil {
+					log.Printf("❌ [%s] 更新数据库失败: %v", at.name, err)
+				} else {
+					log.Printf("✅ [%s] 已自动同步余额到数据库", at.name)
+				}
+			} else {
+				log.Printf("⚠️ [%s] 数据库类型不支持UpdateTraderInitialBalance接口", at.name)
+			}
+		} else {
+			log.Printf("⚠️ [%s] 数据库引用为空，余额仅在内存中更新", at.name)
+		}
+		at.lastBalanceSyncTime = time.Now()
+		return
+	}
+
 	changePercent := ((actualBalance - oldBalance) / oldBalance) * 100
 
 	// 变化超过5%才更新
@@ -317,7 +342,11 @@ func (at *AutoTrader) autoSyncBalanceIfNeeded() {
 				} else {
 					log.Printf("✅ [%s] 已自动同步余额到数据库", at.name)
 				}
+			} else {
+				log.Printf("⚠️ [%s] 数据库类型不支持UpdateTraderInitialBalance接口", at.name)
 			}
+		} else {
+			log.Printf("⚠️ [%s] 数据库引用为空，余额仅在内存中更新", at.name)
 		}
 	} else {
 		log.Printf("✓ [%s] 余额变化不大 (%.2f%%)，无需更新", at.name, changePercent)