id_pool.go 633 B

1234567891011121314151617181920212223242526272829303132333435
  1. package gls
  2. // though this could probably be better at keeping ids smaller, the goal of
  3. // this class is to keep a registry of the smallest unique integer ids
  4. // per-process possible
  5. import (
  6. "sync"
  7. )
  8. type idPool struct {
  9. mtx sync.Mutex
  10. released []uint
  11. max_id uint
  12. }
  13. func (p *idPool) Acquire() (id uint) {
  14. p.mtx.Lock()
  15. defer p.mtx.Unlock()
  16. if len(p.released) > 0 {
  17. id = p.released[len(p.released)-1]
  18. p.released = p.released[:len(p.released)-1]
  19. return id
  20. }
  21. id = p.max_id
  22. p.max_id++
  23. return id
  24. }
  25. func (p *idPool) Release(id uint) {
  26. p.mtx.Lock()
  27. defer p.mtx.Unlock()
  28. p.released = append(p.released, id)
  29. }