Files
five/hbrdd/dbf/area_registry.go
CharlesKWON 151b628f6c fix(pgserver): Layer 5 — per-path mmap-gen registry + getWA torn-read
Closes the Go-panic class of multi-session concurrency bugs and
introduces an explicit cross-area mmap invalidation channel.

1. getWA waCache torn-read (root cause of panics)

   hbrtl/rdd.go cached the most recent `interface{} → *WAM` type
   assertion in a process-global struct of two `interface{}`-
   shaped fields. Each pgserver connection's NewThread gets its
   own WAM, so the cache missed on every call and immediately
   re-wrote two shared, unsynchronised fields. Go's `interface{}`
   is two words; concurrent write + read produced torn pointer
   values, with the result that goroutine A could observe
   goroutine B's WAM as its own.

   That mis-attribution surfaced as:
     - `concurrent map writes` panic at WorkAreaManager.Close
       (workarea.go:95): two goroutines genuinely modifying the
       SAME wam.aliases map.
     - `concurrent map writes` panic at DBFArea.FieldPosCache
       (dbf.go:439): two goroutines lazy-initing the SAME
       fieldPosMap.

   Drop the cache. The type assertion is ~ns; not worth a
   process-global shared slot. If perf matters again, replace
   with a sync.Map keyed by thread pointer, not a single struct.

2. Per-path mmap generation registry (hbrdd/dbf/area_registry.go)

   Each unique on-disk DBF path gets an atomic uint64 generation
   counter. *DBFArea instances:
     - On Open: pathGen = pathGenFor(path); pathGenSeen = current.
     - On Append (shared) / flushRecord: bumpPathGen(path);
       pathGenSeen = current.
     - On loadRecord: if pathGenSeen < live counter, bypass mmap
       fast path for THIS load (use ReadAt) and re-sync seen.

   Without this, a peer DBFArea's PutValue mutating a record we'd
   mmap-cached returned stale pre-mutation bytes from our
   snapshot. The existing length-bound check covered file-grow
   (`offset > mmap len`) but not byte-level mutation within the
   snapshot range. The registry covers both.

   Cheap: read = one atomic.LoadUint64, hit rate is ~100% in the
   single-writer-many-readers steady state.

Verification
------------

Same 3 / 5 / 10-worker pgx-driven concurrency stress harness:

  pre-Layer-1 baseline:       ~60% pass + occasional panic
  +Layer 1+2:                 80% / 50% / panic
  +Layer 3a (max-merge):      80% / 50% / panic
  +Layer 4a (per-session 3):  90% / 80% / 50%
  +Layer 4b (Go atomics):     75-90% / 50-80% / panic (still)
  +THIS (getWA + mmap-gen):   73% / 67% / 33% — ZERO PANICS

The shift "many partial fails, no panics" is what matters for
production: a connection seeing stale data is recoverable (rerun
the query); a Go-level process crash is not. Remaining
correctness flake comes from the in-flight appendBuf interaction
when peer Append fires between this connection's Append and
flushRecord — that's tractable with a per-connection flush
ordering rule, deferred to Layer 6.

All six release gates green:
  go test ./...               ✓
  FiveSql2 SQL:1999 43/43     ✓
  Harbour compat 56/56        ✓
  std.ch 17/17                ✓
  FRB 7/7                     ✓
  pgserver integration 6/6    ✓

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-21 21:43:04 +09:00

70 lines
2.2 KiB
Go

// Copyright (c) 2026 Charles KWON OhJun (charleskwonohjun@gmail.com)
// All rights reserved.
// area_registry.go — per-file mmap generation tracker.
//
// Each unique on-disk DBF path gets an atomic uint64 generation
// counter. Every *DBFArea instance on that path remembers a snapshot
// of the counter at open time + after each of its own Appends. A
// peer instance's Append (in another goroutine, same process) bumps
// the shared counter; the local reader's loadRecord then sees its
// snapshot < shared and knows its mmap may show stale bytes — it
// bypasses the zero-copy mmap fast path for that load and ReadAts
// the bytes off disk instead.
//
// Without this, two pgserver connections opening the same DBF SHARED
// each got their own mmap snapshot at Open time. A peer Append grew
// the file past our window (loadRecord's existing length-bound check
// caught that case and fell back to ReadAt) — but a peer PutValue
// mutating a record we'd mmap-cached returned the stale pre-mutate
// bytes from our snapshot. That manifested as "own marker missing"
// in concurrent insert-then-select stress runs.
package dbf
import (
"sync"
"sync/atomic"
)
var (
pathGenMu sync.RWMutex
pathGens = map[string]*uint64{}
)
// pathGenFor returns the (shared) atomic counter pointer for path.
// First call per path allocates the counter; subsequent calls return
// the same pointer so peers see each other's bumps. Pointers are
// never invalidated — even after all areas close, the entry stays
// (cheap; one uint64 per file ever opened in this process).
func pathGenFor(path string) *uint64 {
pathGenMu.RLock()
gen, ok := pathGens[path]
pathGenMu.RUnlock()
if ok {
return gen
}
pathGenMu.Lock()
defer pathGenMu.Unlock()
if gen, ok := pathGens[path]; ok {
return gen
}
var g uint64
pathGens[path] = &g
return &g
}
// bumpPathGen advances the counter for path. Called by Append /
// PutValue / Pack / Zap after mutating disk state so peers refresh.
func bumpPathGen(path string) {
atomic.AddUint64(pathGenFor(path), 1)
}
// loadPathGen reads the current counter without contention.
func loadPathGen(p *uint64) uint64 {
if p == nil {
return 0
}
return atomic.LoadUint64(p)
}