fix(refcount): wait for prev to exit in Access

Signed-off-by: Christian Stewart <christian@aperture.us>
aperturerobotics · Jul 23, 2024 · f72f133 · f72f133
1 parent 2e0585b
commit f72f133
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/refcount/refcount.go b/refcount/refcount.go
@@ -358,6 +358,10 @@ func (r *RefCount[T]) Access(ctx context.Context, useCtx bool, cb func(ctx conte
 			if prevCancel != nil {
 				prevCancel()
 			}
+			if prevWait != nil {
+				// wait for previous routine to exit.
+				<-prevWait
+			}
 			return context.Canceled
 		case <-waitCh:
 		}