From 4fb91a4f378ba5f3f641547918d435d81cd8bc96 Mon Sep 17 00:00:00 2001
From: wreulicke <12907474+wreulicke@users.noreply.github.com>
Date: Sat, 12 Sep 2020 14:22:07 +0900
Subject: [PATCH] Lock properly in AIMDLimit

---
 limit/aimd.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/limit/aimd.go b/limit/aimd.go
index 975df4b..d390d53 100644
--- a/limit/aimd.go
+++ b/limit/aimd.go
@@ -83,7 +83,7 @@ func (l *AIMDLimit) notifyListeners(newLimit int) {
 // OnSample the concurrency limit using a new rtt sample.
 func (l *AIMDLimit) OnSample(startTime int64, rtt int64, inFlight int, didDrop bool) {
 	l.mu.Lock()
-	l.mu.Unlock()
+	defer l.mu.Unlock()
 
 	l.commonSampler.Sample(rtt, inFlight, didDrop)