perf: single-statement postings, marshal-once broadcast, client interpolation

Measured, then fixed, the three things that made a crowd impossible.

Ledger: Post issued three round trips per posting, so settlement scaled
in network latency rather than work. It is now two statements regardless
of leg count — settling 1000 winners went 844ms to 220ms. The lock and
the balance read must stay separate statements: a single statement, even
one whose CTE does FOR UPDATE, evaluates against a snapshot taken before
the locks are held, so concurrent transactions read stale balances and
money disappears. The conservation tests caught exactly that.

Broadcast: every connection marshalled its own copy, ~355us each. At any
real crowd that exceeds the tick interval by orders of magnitude. Frames
are now serialised once per broadcast and shared.

Feed: the player list is capped at 24 and carries no public keys, and
running rounds broadcast at 5Hz instead of 60Hz. Clients compute the
multiplier locally from the round start time, which the deterministic
curve makes exact. Frame size fell from 3.6KB to 1.8KB.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
drjones
2026-08-05 22:36:30 +00:00
parent 5c3393b989
commit 038550b6ff
7 changed files with 555 additions and 67 deletions

View File

@@ -15,6 +15,7 @@ import (
"fmt"
"math/big"
"sort"
"strings"
"time"
"github.com/jackc/pgx/v5"
@@ -83,58 +84,126 @@ func (l *Ledger) Post(ctx context.Context, kind string, roundID *int64, postings
return 0, err
}
ordered := append([]Posting(nil), postings...)
sort.Slice(ordered, func(i, j int) bool {
return ordered[i].AccountID < ordered[j].AccountID
})
for _, p := range ordered {
// Lock the account row first, then read its latest balance. Taking the
// lock before the read is what serializes concurrent spends.
var allowNegative bool
if err := tx.QueryRow(ctx,
`SELECT allow_negative FROM accounts WHERE id = $1 FOR UPDATE`,
p.AccountID).Scan(&allowNegative); err != nil {
return 0, fmt.Errorf("locking account %d: %w", p.AccountID, err)
// Merge duplicate accounts before locking. A transaction that touched the
// same account twice would otherwise read a stale balance for the second
// leg and write a posting that contradicts the first.
merged := make(map[int64]int64, len(postings))
order := make([]int64, 0, len(postings))
for _, p := range postings {
if _, seen := merged[p.AccountID]; !seen {
order = append(order, p.AccountID)
}
merged[p.AccountID] += p.AmountMsat
}
sort.Slice(order, func(i, j int) bool { return order[i] < order[j] })
var before int64
if err := tx.QueryRow(ctx,
`SELECT COALESCE(
(SELECT balance_after FROM postings
WHERE account_id = $1 ORDER BY id DESC LIMIT 1), 0)`,
p.AccountID).Scan(&before); err != nil {
return 0, fmt.Errorf("reading balance of account %d: %w", p.AccountID, err)
ids := make([]int64, 0, len(order))
amounts := make([]int64, 0, len(order))
for _, id := range order {
if merged[id] == 0 {
continue // legs cancelled out; nothing to record
}
// Detect wraparound before trusting the result: a credit that
// overflows would otherwise land as a negative balance, and a debit
// that underflows as a positive one.
after := before + p.AmountMsat
if (p.AmountMsat > 0 && after < before) || (p.AmountMsat < 0 && after > before) {
return 0, fmt.Errorf("ledger: amount %d overflows the balance of account %d (%d)",
p.AmountMsat, p.AccountID, before)
}
if after < 0 && !allowNegative {
return 0, fmt.Errorf("%w: account %d holds %d, needs %d",
ErrInsufficientFunds, p.AccountID, before, -p.AmountMsat)
}
if _, err := tx.Exec(ctx,
`INSERT INTO postings
(transaction_id, account_id, amount_msat, balance_before, balance_after)
VALUES ($1, $2, $3, $4, $5)`,
txID, p.AccountID, p.AmountMsat, before, after); err != nil {
ids = append(ids, id)
amounts = append(amounts, merged[id])
}
if len(ids) == 0 {
// Every leg cancelled. The transaction row stands as a record that
// something was attempted, but there is no balance change to write.
if err := tx.Commit(ctx); err != nil {
return 0, err
}
return txID, nil
}
// Lock every account first, in ascending id order so concurrent
// transactions cannot deadlock against each other.
//
// This must be its own statement. A single statement — even one whose CTE
// does FOR UPDATE — evaluates against one snapshot taken before the locks
// are held, so the balance read would see pre-lock data and concurrent
// transactions would silently overwrite each other. Splitting it means the
// second statement takes a fresh snapshot, by which point we hold the
// locks and no other writer can commit against these accounts.
if _, err := tx.Exec(ctx,
`SELECT id FROM accounts WHERE id = ANY($1) ORDER BY id FOR UPDATE`,
ids); err != nil {
return 0, fmt.Errorf("locking accounts: %w", err)
}
// Now write every posting in one statement, however many legs there are.
// Doing this per-posting cost three round trips each, which made
// settlement scale in network latency rather than in real work.
rows, err := tx.Query(ctx, `
WITH input AS (
SELECT unnest($2::bigint[]) AS account_id,
unnest($3::bigint[]) AS amount
),
current AS (
SELECT i.account_id,
i.amount,
COALESCE((SELECT p.balance_after
FROM postings p
WHERE p.account_id = i.account_id
ORDER BY p.id DESC
LIMIT 1), 0) AS balance_before
FROM input i
)
INSERT INTO postings
(transaction_id, account_id, amount_msat, balance_before, balance_after)
SELECT $1, account_id, amount, balance_before, balance_before + amount
FROM current
RETURNING account_id, balance_after`,
txID, ids, amounts)
if err != nil {
// The balance floor is enforced by a database trigger, so an overdraft
// surfaces here. Translate it into the domain error callers expect.
if isBalanceFloorViolation(err) {
return 0, fmt.Errorf("%w: %v", ErrInsufficientFunds, err)
}
return 0, err
}
written := 0
for rows.Next() {
var acct, after int64
if err := rows.Scan(&acct, &after); err != nil {
rows.Close()
return 0, err
}
written++
}
rows.Close()
if err := rows.Err(); err != nil {
if isBalanceFloorViolation(err) {
return 0, fmt.Errorf("%w: %v", ErrInsufficientFunds, err)
}
return 0, err
}
if written != len(ids) {
return 0, fmt.Errorf("ledger: wrote %d postings for %d accounts; "+
"an account id does not exist", written, len(ids))
}
if err := tx.Commit(ctx); err != nil {
if isBalanceFloorViolation(err) {
return 0, fmt.Errorf("%w: %v", ErrInsufficientFunds, err)
}
return 0, err
}
return txID, nil
}
// isBalanceFloorViolation reports whether an error is the database refusing to
// let an account go negative.
func isBalanceFloorViolation(err error) bool {
if err == nil {
return false
}
msg := err.Error()
return strings.Contains(msg, "may not go negative") ||
strings.Contains(msg, "balance_nonnegative")
}
// Transfer moves funds between two accounts. This is the peer-to-peer path.
func (l *Ledger) Transfer(ctx context.Context, from, to int64, amountMsat int64) (int64, error) {
if amountMsat <= 0 {