semaphore.go

package semaphore

import "sync"

type Semaphore struct {
  Channel chan uint
  Mutex sync.Mutex
}

func New(size uint) *Semaphore {
  return &Semaphore{
    Channel: make(chan uint, size),
    Mutex: sync.Mutex{},
  }
}

func (s *Semaphore) Lock(size uint) {
  s.Mutex.Lock()
  for uint(i):=0; i<size; i++ {
    s.Channel <- i
  }
  s.Mutex.Unlock()
}

func (s *Semaphore) Unlock(size uint) {
  for uint(i):=0; i<size; i++ {
    <- s.Channel
  }
}