fix(traffic): re-enable clients and serialize the write in Reset All Client Traffic

ClientService.ResetAllTraffics zeroed up/down but, unlike every sibling reset
path, never restored enable=true, so clients that had been auto-disabled for
exceeding their quota stayed cut with zero usage after a reset. It also wrote
client_traffics directly on the shared DB handle instead of through the serial
traffic writer, reintroducing the cross-transaction lock-order deadlock the
writer exists to prevent. Restore enable and run the reset inside
submitTrafficWrite within one transaction.
This commit is contained in:
MHSanaei
2026-07-15 02:08:06 +02:00
parent b6928f4939
commit 3eb214d022
2 changed files with 39 additions and 9 deletions
@@ -52,3 +52,27 @@ func TestAddTrafficCommitsDespiteDisableHelperError(t *testing.T) {
t.Fatalf("traffic tick was rolled back by a best-effort disable-helper error: up=%d down=%d, want 500/700", reloaded.Up, reloaded.Down)
}
}
// Reset All Client Traffic must re-enable clients that were auto-disabled for
// exceeding their quota, matching every other reset path; otherwise a reset
// leaves depleted clients cut with zero usage.
func TestResetAllTrafficsReenablesDepletedClients(t *testing.T) {
db := initTrafficTestDB(t)
svc := &ClientService{}
if err := db.Create(&xray.ClientTraffic{InboundId: 1, Email: "spent@x", Enable: false, Up: 60, Down: 60, Total: 100}).Error; err != nil {
t.Fatalf("seed: %v", err)
}
if _, err := svc.ResetAllTraffics(); err != nil {
t.Fatalf("ResetAllTraffics: %v", err)
}
row := readTraffic(t, db, "spent@x")
if row.Up != 0 || row.Down != 0 {
t.Fatalf("usage not reset: up=%d down=%d", row.Up, row.Down)
}
if !row.Enable {
t.Fatal("a depleted client must be re-enabled after Reset All Client Traffic, matching every other reset path")
}
}