Fix multiplied relation rendering

krzysztofreczek · Dec 3, 2020 · 4711f68 · 4711f68
1 parent a753b9b
commit 4711f68
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/view/render.go b/pkg/view/render.go
@@ -50,8 +50,10 @@ func (v view) render(s model.Structure) string {
 		componentsRendered[c.ID] = struct{}{}
 	}
 
+	relationsRendered := map[string]struct{}{}
+
 	for true {
-		numRendered := len(componentsRendered)
+		numRendered := len(relationsRendered)
 
 		for src, to := range s.Relations {
 			for trg := range to {
@@ -79,11 +81,17 @@ func (v view) render(s model.Structure) string {
 					componentsRendered[c.ID] = struct{}{}
 				}
 
+				relationID := src + trg
+				if _, rendered := relationsRendered[relationID]; rendered {
+					continue
+				}
+
 				sb.WriteString(buildComponentConnection(src, trg, v.lineColor))
+				relationsRendered[relationID] = struct{}{}
 			}
 		}
 
-		if numRendered == len(componentsRendered) {
+		if numRendered == len(relationsRendered) {
 			break
 		}
 	}