Weighted: веса и бюджет

Почему штуки не подходят

Обычный семафор считает операции штуками: каждая занимает ровно одно разрешение. Но ресурс, который мы ограничиваем, часто расходуется неравномерно. Одна задача держит соединение, другая — пятьдесят мегабайт памяти, третья — оба сразу. При лимите «5 одновременных операций» пять задач по пятьдесят мегабайт пролезут без проблем, а пять лёгких упрутся в тот же потолок, хотя могли бы работать параллельно. Нужен семафор, который меряет объём ресурса.

Бюджет в единицах ресурса

sem := semaphore.NewWeighted(100) // бюджет: 100 единиц ресурса

if err := sem.Acquire(ctx, 50); err != nil {
    return err            // отменены: вес не занят
}
defer sem.Release(50)     // возвращаем ровно занятое

if err := sem.Acquire(ctx, 1); err != nil { // лёгкая задача на 1 единицу
    return err
}
defer sem.Release(1)

Тяжёлая задача занимает половину бюджета, лёгкая — сотую. Одновременно может работать: две тяжёлые, или одна тяжёлая и пятьдесят лёгких, или сто лёгких. Лимит выражает реальный ресурс — мегабайты, соединения, токены, — а не «число горутин». Ошибку Acquire проверяют до defer Release: при отменённом контексте вес не занят, и возврат незанятого веса тихо раздул бы бюджет.

Acquire(ctx, n)

Acquire(ctx, n) ведёт себя как одиночное разрешение из прошлого урока, только с учётом веса: ждёт, пока освободится n единиц бюджета, отменяется по контексту, ничего не теряя. Release(n) возвращает ровно n — несовпадение весов с Acquire — это ошибка в коде, и про парность заботится тот же defer.