cloudflare/pint
Publicmirrored from https://github.com/cloudflare/pintAvailable
internal/promapi/keylock.go
35lines · modecode
| 1 | package promapi |
| 2 | |
| 3 | import "sync" |
| 4 | |
| 5 | // https://medium.com/@petrlozhkin/kmutex-lock-mutex-by-unique-id-408467659c24 |
| 6 | type partitionLocker struct { |
| 7 | c *sync.Cond |
| 8 | l sync.Locker |
| 9 | s map[string]struct{} |
| 10 | } |
| 11 | |
| 12 | func newPartitionLocker(l sync.Locker) *partitionLocker { |
| 13 | return &partitionLocker{c: sync.NewCond(l), l: l, s: make(map[string]struct{})} |
| 14 | } |
| 15 | |
| 16 | func (p *partitionLocker) locked(id string) (ok bool) { |
| 17 | _, ok = p.s[id] |
| 18 | return ok |
| 19 | } |
| 20 | |
| 21 | func (p *partitionLocker) lock(id string) { |
| 22 | p.l.Lock() |
| 23 | defer p.l.Unlock() |
| 24 | for p.locked(id) { |
| 25 | p.c.Wait() |
| 26 | } |
| 27 | p.s[id] = struct{}{} |
| 28 | } |
| 29 | |
| 30 | func (p *partitionLocker) unlock(id string) { |
| 31 | p.l.Lock() |
| 32 | defer p.l.Unlock() |
| 33 | delete(p.s, id) |
| 34 | p.c.Broadcast() |
| 35 | } |
| 36 | |