Detect recursive use by looking for repeated component connections (#156

)
krzysztofreczek · Sep 6, 2024 · 165d0d1 · 165d0d1
1 parent 7d69da8
commit 165d0d1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/scraper/strategy.go b/pkg/scraper/strategy.go
@@ -161,11 +161,12 @@ func (s *scraper) scrapeStruct(
 	}
 
 	vID := componentID(v)
-	if c, ok := s.typeCounters[vID]; ok && c > maxRecursiveScrapes {
+	vUsageKey := fmt.Sprintf("%s-%s", parentID, vID)
+	if c, ok := s.typeCounters[vUsageKey]; ok && c > maxRecursiveScrapes {
 		s.debug(v, "struct is being used recursively, skipping")
 		return
 	} else {
-		s.typeCounters[vID]++
+		s.typeCounters[vUsageKey]++
 	}
 
 	var c model.Component