fix: gorouting leak on redis lock

aldor007 · Oct 19, 2022 · 453bc3c · 453bc3c
1 parent 4dc8475
commit 453bc3c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/lock/redis.go b/pkg/lock/redis.go
@@ -120,6 +120,13 @@ func (m *RedisLock) Lock(ctx context.Context, key string) (result LockResult, ok
 					select {
 					case <-ch:
 						m.NotifyAndRelease(ctx, key, nil)
+					case <-result.Cancel:
+						m.memoryLock.Release(ctx, key)
+						pubsub.Close()
+					case <-ctx.Done():
+						m.memoryLock.Release(ctx, key)
+						pubsub.Close()
+						return
 					}
 				}
 			}()