[CWS] close query chan when finishing the btfhub collector (#25500)

DataDog · May 10, 2024 · 3e03e40 · 3e03e40
1 parent 09161cd
commit 3e03e40
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/security/probe/constantfetch/btfhub/main.go b/pkg/security/probe/constantfetch/btfhub/main.go
@@ -113,6 +113,7 @@ func main() {
 		panic(err)
 	}
 
+	twCollector.close()
 	wg.Wait()
 
 	export := twCollector.finish()
@@ -239,6 +240,10 @@ func newTreeWalkCollector(preAllocHint int) *treeWalkCollector {
 	}
 }
 
+func (c *treeWalkCollector) close() {
+	close(c.queryChan)
+}
+
 func (c *treeWalkCollector) finish() constantfetch.BTFHubConstants {
 	slices.SortFunc(c.results, func(a extractionResult, b extractionResult) int {
 		return cmp.Compare(a.counter, b.counter)