Close the r.proceed channel

Signed-off-by: Gabriel Adrian Samfira <[email protected]>
microsoft · Feb 27, 2023 · b2acd4e · b2acd4e
1 parent e79beb3
commit b2acd4e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/internal/wclayer/baselayerreader.go b/internal/wclayer/baselayerreader.go
@@ -207,8 +207,12 @@ func (r *baseLayerReader) Read(b []byte) (int, error) {
 
 func (r *baseLayerReader) Close() (err error) {
 	defer r.s.End()
-	defer func() { oc.SetSpanStatus(r.s, err) }()
+	defer func() {
+		oc.SetSpanStatus(r.s, err)
+		close(r.proceed)
+	}()
 	r.proceed <- false
+	// The r.result channel will be closed once walk() returns
 	<-r.result
 	r.reset()
 	return nil