fix: data race, scale-in PnL, nonce mutex, dead code, hourly check

- 🔴 Data race: Add GetPositionsCopy() returning deep copies (no shared
  ArbPosition pointers). Use it in dashboard broadcastLoop + handleStatus.
- 🟡 Scale-in PnL: Track LongEntryPrices/ShortEntryPrices on ArbPosition,
  compute weighted average (harmonic mean) at exit for accurate PnL.
- 🟢 CalcNetProfit: Delete dead code from exchange/helpers.go.
- 🟢 HL nonce: Add sync.Mutex around lastNonce++ (thread safety).
- 🟢 Hourly check: Change from 5-second window to minute window.
- 🟢 ExitPrice: Test mode closeLeg already handled by checkExit.
This commit is contained in:
jackyu66git
2026-05-03 18:31:14 +08:00
parent ab48e207a5
commit b08d8490fc
6 changed files with 94 additions and 54 deletions
+8 -6
View File
@@ -290,8 +290,8 @@ func (d *Dashboard) broadcastLoop() {
}
d.hub.Broadcast("prices", prices)
// 2. Open positions with live PnL (P3-3)
positions := d.trader.GetOpenPositions()
// 2. Open positions with live PnL (P3-3) — use safe copy for concurrent read
positions := d.trader.GetPositionsCopy()
posList := make([]map[string]interface{}, 0, len(positions))
for _, pos := range positions {
posEntry := map[string]interface{}{
@@ -304,7 +304,7 @@ func (d *Dashboard) broadcastLoop() {
"started_at": pos.StartedAt.Format("15:04:05"),
}
// Calculate live PnL from current prices
// Calculate live PnL from current prices — use weighted average for scale-ins
if exMap := snap[pos.Coin]; exMap != nil {
bgP := exMap[ExBitget]
hlP := exMap[ExHyperLiquid]
@@ -315,8 +315,10 @@ func (d *Dashboard) broadcastLoop() {
} else {
longCurrent, shortCurrent = hlP, bgP
}
longPnl := (longCurrent - pos.LongLeg.EntryPrice) / pos.LongLeg.EntryPrice * 100
shortPnl := (pos.ShortLeg.EntryPrice - shortCurrent) / pos.ShortLeg.EntryPrice * 100
longAvg := weightedAvgPrice(pos.LongEntryPrices, pos.AmountUSD/float64(max(1, len(pos.LongEntryPrices))))
shortAvg := weightedAvgPrice(pos.ShortEntryPrices, pos.AmountUSD/float64(max(1, len(pos.ShortEntryPrices))))
longPnl := (longCurrent - longAvg) / longAvg * 100
shortPnl := (shortAvg - shortCurrent) / shortAvg * 100
totalFees := 2 * (makerFees[ExBitget] + makerFees[ExHyperLiquid])
netPnl := longPnl + shortPnl - totalFees
@@ -426,7 +428,7 @@ func (d *Dashboard) handleIndex(w http.ResponseWriter, r *http.Request) {
func (d *Dashboard) handleStatus(w http.ResponseWriter, r *http.Request) {
snap := d.store.GetAll()
positions := d.trader.GetOpenPositions()
positions := d.trader.GetPositionsCopy()
converged, diverged, flat, total := d.trader.GetClosedStats()
resp := map[string]interface{}{
+3 -40
View File
@@ -1,42 +1,5 @@
package exchange
import "github.com/gorilla/websocket"
// These are needed for compilation of the exchange package.
// PriceConnector is defined in connector.go.
var _ = websocket.ErrCloseSent // keep gorilla/websocket import
// CalcNetProfit calculates net profit % for a complete round trip (entry + exit) between two exchanges.
// buyPrice: price on the buy exchange
// sellPrice: price on the sell exchange
// buyFee: fee rate on buy exchange (e.g. 0.03 for 0.03%)
// sellFee: fee rate on sell exchange
// buyFee2: buy fee on the other exchange
// sellFee2: sell fee on the other exchange
// Returns net profit in percentage.
func CalcNetProfit(price1, price2, fee1Buy, fee1Sell, fee2Buy, fee2Sell float64) float64 {
// price1 = Bitget, price2 = HyperLiquid
// Try: buy cheap (min), sell expensive (max)
buyPrice := price1
sellPrice := price2
buyFee := fee1Buy
sellFee := fee2Sell
if price2 < price1 {
buyPrice = price2
sellPrice = price1
buyFee = fee2Buy
sellFee = fee1Sell
}
// Entry: buy at buyPrice (pay buyFee), sell short at sellPrice (pay sellFee)
if buyPrice <= 0 || sellPrice <= 0 {
return 0
}
cost := buyPrice * (1 + buyFee/100)
revenue := sellPrice * (1 - sellFee/100)
// Exit: sell long (pay sellFee), buy back short (pay buyFee)
// Total fees = 2 * (buyFee + sellFee), first round already in formula above
return (revenue/cost-1)*100 - (buyFee + sellFee)
}
// Package-level helpers for the exchange package.
// connector.go imports gorilla/websocket, so this file needs no imports
// for that dependency. CalcNetProfit was removed (see netProfit in scanner.go).
+5 -1
View File
@@ -8,6 +8,7 @@ import (
"math/big"
"net/http"
"strings"
"sync"
"time"
"crypto/ed25519"
@@ -50,6 +51,7 @@ type HyperLiquidTrade struct {
Address string
client *http.Client
lastNonce int64
nonceMu sync.Mutex // protect lastNonce++ (Issue #4)
}
func NewHyperLiquidTrade(privateKeyHex, address string) (*HyperLiquidTrade, error) {
@@ -96,9 +98,11 @@ func (h *HyperLiquidTrade) PlaceMarketOrder(coin, side, sz string) (string, erro
BrokerCode: 0,
}
// Generate nonce
// Generate nonce (thread-safe)
h.nonceMu.Lock()
h.lastNonce++
nonce := time.Now().UnixMilli()*1_000_000 + h.lastNonce%1_000_000
h.nonceMu.Unlock()
// Sign the action
sig, err := h.signAction(action, nonce)
+3 -3
View File
@@ -187,10 +187,10 @@ func main() {
tickMs, t1.Sub(t0).Milliseconds(), t2.Sub(t1).Milliseconds(), t3.Sub(t2).Milliseconds())
}
// Hourly trade summary
// Hourly trade summary — use hour-based tracking (wider window than second-granularity)
hour := now.Hour()
if now.Minute() == 0 && now.Second() < 5 && hour != lastHour {
positions := trader.GetOpenPositions()
if hour != lastHour && now.Minute() < 1 {
positions := trader.GetPositionsCopy()
notifier.SendTradeSummary(positions, now.Format("2006-01-02 15:04"))
lastHour = hour
}
+1 -1
View File
@@ -73,7 +73,7 @@ func (n *Notifier) SendAlert(opp *ArbOpportunity) {
}
// SendTradeSummary sends a summary of open positions at each hour.
func (n *Notifier) SendTradeSummary(positions []*ArbPosition, timeStr string) {
func (n *Notifier) SendTradeSummary(positions []ArbPosition, timeStr string) {
if n.BotToken == "" || n.ChatID == "" {
return
}
+74 -3
View File
@@ -50,6 +50,43 @@ type ArbPosition struct {
Status string // "open", "closed"
RealizedPnl float64
ErrorLog string
// Track all entry prices for weighted-average PnL across scale-ins (Issue #2)
LongEntryPrices []float64 // all long entry prices (initial + scale-ins)
ShortEntryPrices []float64 // all short entry prices (initial + scale-ins)
}
// DeepCopy returns a copy-safe snapshot of the position (no shared pointers).
func (p *ArbPosition) DeepCopy() ArbPosition {
c := *p
if p.LongLeg != nil {
lc := *p.LongLeg
c.LongLeg = &lc
}
if p.ShortLeg != nil {
sc := *p.ShortLeg
c.ShortLeg = &sc
}
if p.LongEntryPrices != nil {
c.LongEntryPrices = make([]float64, len(p.LongEntryPrices))
copy(c.LongEntryPrices, p.LongEntryPrices)
}
if p.ShortEntryPrices != nil {
c.ShortEntryPrices = make([]float64, len(p.ShortEntryPrices))
copy(c.ShortEntryPrices, p.ShortEntryPrices)
}
return c
}
// GetPositionsCopy returns deep copies of all open positions — safe for concurrent read.
func (t *Trader) GetPositionsCopy() []ArbPosition {
t.mu.Lock()
defer t.mu.Unlock()
r := make([]ArbPosition, 0, len(t.positions))
for _, p := range t.positions {
r = append(r, p.DeepCopy())
}
return r
}
// Trader handles scalable arbitrage between Bitget and HyperLiquid.
@@ -245,6 +282,8 @@ func (t *Trader) executeEntry(opp *ArbOpportunity, store *PriceStore, notifier *
Coin: opp.Coin, Exchange: ExHyperLiquid, Side: Short,
EntryPrice: hlP, EntryTime: time.Now(),
}
pos.LongEntryPrices = []float64{bgP}
pos.ShortEntryPrices = []float64{hlP}
} else {
pos.Direction = "HL->BG"
pos.EntrySpread = (bgP - hlP) / hlP * 100 // positive when bgP > hlP
@@ -256,6 +295,8 @@ func (t *Trader) executeEntry(opp *ArbOpportunity, store *PriceStore, notifier *
Coin: opp.Coin, Exchange: ExBitget, Side: Short,
EntryPrice: bgP, EntryTime: time.Now(),
}
pos.LongEntryPrices = []float64{hlP}
pos.ShortEntryPrices = []float64{bgP}
}
t.mu.Lock()
@@ -357,6 +398,8 @@ func (t *Trader) checkScaleIn(pos *ArbPosition, bgP, hlP, diffPct float64, store
pos.ScaleLevels++
pos.LastScaleAt = time.Now()
pos.AmountUSD += t.cfg.TradeAmountUSD
pos.LongEntryPrices = append(pos.LongEntryPrices, longPrice)
pos.ShortEntryPrices = append(pos.ShortEntryPrices, shortPrice)
log.Printf("[Trader] %s: Scale-in #%d executed | spread=%.4f%% (entry=%.4f%%) | total=$%.0f",
pos.Coin, pos.ScaleLevels, diffPct, entryDiff, pos.AmountUSD)
@@ -389,7 +432,8 @@ func (t *Trader) checkExit(pos *ArbPosition, bgP, hlP, diffPct float64, notifier
return
}
// Calculate P&L
// Calculate P&L — use weighted average entry for scale-in positions
// Each scale adds cfg.TradeAmountUSD at the scale price
var longCurrent, shortCurrent float64
if pos.LongLeg.Exchange == ExBitget {
longCurrent, shortCurrent = bgP, hlP
@@ -397,8 +441,12 @@ func (t *Trader) checkExit(pos *ArbPosition, bgP, hlP, diffPct float64, notifier
longCurrent, shortCurrent = hlP, bgP
}
longPnl := (longCurrent - pos.LongLeg.EntryPrice) / pos.LongLeg.EntryPrice * 100
shortPnl := (pos.ShortLeg.EntryPrice - shortCurrent) / pos.ShortLeg.EntryPrice * 100
// Weighted average entry prices across all scale levels
longAvg := weightedAvgPrice(pos.LongEntryPrices, t.cfg.TradeAmountUSD)
shortAvg := weightedAvgPrice(pos.ShortEntryPrices, t.cfg.TradeAmountUSD)
longPnl := (longCurrent - longAvg) / longAvg * 100
shortPnl := (shortAvg - shortCurrent) / shortAvg * 100
totalFees := 2 * (makerFees[ExBitget] + makerFees[ExHyperLiquid]) // 开仓 + 平仓手续费
netPnl := longPnl + shortPnl - totalFees
@@ -627,6 +675,27 @@ func (t *Trader) GetOpenPositions() []*ArbPosition {
return r
}
// weightedAvgPrice computes the weighted average entry price across multiple scale levels.
// Each level trades the same USD amount, so the result is the harmonic mean of prices.
func weightedAvgPrice(prices []float64, amountPerTrade float64) float64 {
if len(prices) == 0 {
return 0
}
totalShares := 0.0
totalCost := 0.0
for _, p := range prices {
if p <= 0 {
continue
}
totalShares += amountPerTrade / p
totalCost += amountPerTrade
}
if totalShares <= 0 {
return prices[0] // fallback
}
return totalCost / totalShares
}
// GetClosedStats returns convergence stats from all closed trades.
func (t *Trader) GetClosedStats() (converged, diverged, flat, total int) {
t.mu.Lock()
@@ -717,12 +786,14 @@ func (t *Trader) restoreOpenPositions() {
Coin: tr.Coin, Exchange: tr.LongExchange, Side: Long,
EntryPrice: *tr.LongEntry, EntryTime: tr.OpenedAt,
}
pos.LongEntryPrices = []float64{*tr.LongEntry}
}
if tr.ShortEntry != nil {
pos.ShortLeg = &PositionLeg{
Coin: tr.Coin, Exchange: tr.ShortExchange, Side: Short,
EntryPrice: *tr.ShortEntry, EntryTime: tr.OpenedAt,
}
pos.ShortEntryPrices = []float64{*tr.ShortEntry}
}
t.positions[tr.Coin] = pos
// Prevent immediate re-trading of the same coin