fix race condition when deleting key lock

abihf · abihf · commit f95494175ed2 · 2022-03-21T19:59:31.000+07:00
diff --git a/key_locker.go b/key_locker.go
@@ -46,6 +46,8 @@ func (l *InMemoryKeyLocker[Key]) Lock(key Key) func() {
 		}
 		item.m.Unlock()
 		if atomic.AddInt32(&item.ref, -1) <= 0 {
+			l.root.Lock()
+			defer l.root.Unlock()
 			delete(l.locks, key)
 		}
 		unlocked = true

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,8 @@ func (l *InMemoryKeyLocker[Key]) Lock(key Key) func() {`
`46`	`46`	`}`
`47`	`47`	`item.m.Unlock()`
`48`	`48`	`if atomic.AddInt32(&item.ref, -1) <= 0 {`
	`49`	`+ l.root.Lock()`
	`50`	`+ defer l.root.Unlock()`
`49`	`51`	`delete(l.locks, key)`
`50`	`52`	`}`
`51`	`53`	`unlocked = true`