67 lines
976 B
Go
67 lines
976 B
Go
package locker
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
type Elem struct {
|
|
Pipe chan bool
|
|
Usage int
|
|
}
|
|
|
|
func NewElem() *Elem {
|
|
return &Elem{
|
|
Pipe: make(chan bool, 1),
|
|
}
|
|
}
|
|
|
|
type Locker struct {
|
|
mtx sync.Mutex
|
|
lMap map[string]*Elem
|
|
}
|
|
|
|
func NewLocker() *Locker {
|
|
lock := &Locker{
|
|
lMap: make(map[string]*Elem),
|
|
}
|
|
return lock
|
|
}
|
|
|
|
func (lock *Locker) WaitAndLock(name string) {
|
|
lock.mtx.Lock()
|
|
p, exist := lock.lMap[name]
|
|
if !exist {
|
|
p = NewElem()
|
|
lock.lMap[name] = p
|
|
p.Pipe <- true
|
|
}
|
|
p.Usage += 1
|
|
lock.mtx.Unlock()
|
|
select {
|
|
case <-p.Pipe:
|
|
// NOP
|
|
}
|
|
}
|
|
|
|
func (lock *Locker) Done(name string) {
|
|
lock.mtx.Lock()
|
|
p, exist := lock.lMap[name]
|
|
if exist {
|
|
p.Pipe <- true
|
|
if p.Usage > 0 {
|
|
p.Usage -= 1
|
|
}
|
|
}
|
|
garbageKeys := make([]string, 0)
|
|
for key, _ := range lock.lMap {
|
|
elem := lock.lMap[key]
|
|
if elem.Usage == 0 && key != name {
|
|
garbageKeys = append(garbageKeys, key)
|
|
}
|
|
}
|
|
for _, key := range garbageKeys {
|
|
delete(lock.lMap, key)
|
|
}
|
|
lock.mtx.Unlock()
|
|
}
|