Главные идеи, тренды и события в технологиях, науке и AI от лучших авторов.
6 days ago
6 days ago
6 days ago
6 days ago
6 days ago
Invalidation events happen in industry (e.g. move from ZIRP to post-ZIRP management environment) but it’s difficult for non-operators to understand implications with conviction
type Sem struct {
limit int32
cur *atomic.Int32
}
func NewSem(init int) *Sem {
return &Sem{
limit: int32(init),
cur: &atomic.Int32{},
}
}
func (s *Sem) TryAcquire() bool {
for {
cur := s.cur.Load()
if cur >= s.limit {
return false
}
if s.cur.CompareAndSwap(cur, cur+1) {
return true
}
}
}
func (s *Sem) Release() {
cur := s.cur.Add(-1)
if cur < 0 {
panic("semaphore release without acquire")
}
}
Базовый бенч с limit=50 при 200 горутинах показал, что AtomicSem быстрее ChannelSem примерно в 2 раза:
goos: darwin
goarch: arm64
pkg: go-helloworld/semaphore
cpu: Apple M3 Pro
BenchmarkAtomicSem
BenchmarkAtomicSem-11 42 28481842 ns/op
BenchmarkChannelSem
BenchmarkChannelSem-11 19 59474664 ns/op
PASS
Но данное решение плохо работает с большим количеством потоков, так как будет больше конкуренция на CAS loop операции между потоками -> получаем деградацию CAS
Решение:
- Добавить fast check load - завершаем функцию если лимит уже превышен
- Потом быстрый спин-цикл (напр 302 итераций) с CAS + добавить runtime.Gosched(), чтобы горутина переодически снимала себя с потока
- Если спины не дали результата - делать exponential backoff (time.Sleep) до х50
Пример:
func (s *Sem) OptimizedTryAcquire() bool {
// Быстрая проверка если уже заполнено
if s.cur.Load() >= s.limit {
return false
}
// Короткий спин с CAS
const spinCount = 302
for i := 0; i < spinCount; i++ {
cur := s.cur.Load()
if cur >= s.limit {
return false
}
if s.cur.CompareAndSwap(cur, cur+1) {
return true
}
// периодически снимаем горутину через goshed
if (i & 7) == 0 {
runtime.Gosched()
}
}
// Экспоненциальный backoff
backoff := 1 * time.Microsecond
const maxBackoff = 50 * time.Millisecond
for {
// до сна проверяем — возможно место освободилось
cur := s.cur.Load()
if cur >= s.limit {
return false
}
if s.cur.CompareAndSwap(cur, cur+1) {
return true
}
time.Sleep(backoff)
backoff *= 2
if backoff > maxBackoff {
backoff = maxBackoff
}
}
}{}
