Stop a slow host turning a login-guard test into a segfault (closes #186)
All checks were successful
check / check (push) Successful in 2m49s
All checks were successful
check / check (push) Successful in 2m49s
This commit was merged in pull request #188.
This commit is contained in:
@@ -173,10 +173,38 @@ func newLoginGuard(
|
||||
// acquire reserves a verification slot, waiting up to the guard's
|
||||
// wait for one. It reports false when the queue of waiters is
|
||||
// already full, when no slot became available in time, or when the
|
||||
// request was cancelled first; the caller must then answer 503
|
||||
// without verifying anything. The returned function releases the
|
||||
// request was cancelled while waiting; the caller must then answer
|
||||
// 503 without verifying anything. The returned function releases the
|
||||
// slot and must be called exactly once.
|
||||
//
|
||||
// ctx is consulted only once the request has to wait: a slot that is
|
||||
// free on arrival is handed out without looking at it, so an
|
||||
// already-cancelled request can be granted one. That is deliberate
|
||||
// and matches lifecycle.waitDone — the caller abandons the work on
|
||||
// its own ctx and releases the slot immediately, so nothing is spent
|
||||
// on it, and refusing instead would mean shedding a request with
|
||||
// capacity standing free.
|
||||
func (g *loginGuard) acquire(ctx context.Context) (func(), bool) {
|
||||
// A free slot is taken before any timer is armed, and before a
|
||||
// queue place is claimed: a request that never waits is not a
|
||||
// waiter. Without this preamble the bounded select below can find
|
||||
// its slot send and an already-expired timer ready at the same
|
||||
// time, and Go picks among ready cases uniformly at random — so a
|
||||
// process descheduled for longer than the wait sheds a request
|
||||
// with slots standing free, which is precisely when shedding is
|
||||
// least defensible.
|
||||
//
|
||||
// This cannot let a late arrival barge past a queued waiter. A
|
||||
// waiter can only be parked on a FULL buffer, and a release
|
||||
// refills that buffer from the head of the send queue under the
|
||||
// channel lock, so the buffer never appears non-full while anyone
|
||||
// is parked and this send fails whenever there is a waiter.
|
||||
select {
|
||||
case g.slots <- struct{}{}:
|
||||
return func() { <-g.slots }, true
|
||||
default:
|
||||
}
|
||||
|
||||
// Shedding past the queue depth is what keeps waiting memory
|
||||
// bounded; the wait alone only bounds how long one waiter holds
|
||||
// its parsed form, not how many hold one at once.
|
||||
|
||||
Reference in New Issue
Block a user