fix race condition when deleting key lock

abihf · Mar 21, 2022 · f954941 · f954941
1 parent 92e243b
commit f954941
Showing 1 changed file with 2 additions and 0 deletions.
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