with replicas

mcanlas · Mar 27, 2024 · 64b898e · 64b898e
1 parent 60b2879
commit 64b898e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/temporal-diagrams-demo/src/main/scala/com/htmlism/temporaldiagrams/demo/DemoDsl.scala b/temporal-diagrams-demo/src/main/scala/com/htmlism/temporaldiagrams/demo/DemoDsl.scala
@@ -69,11 +69,16 @@ object DemoDsl:
             if isBright then skinPlantUmlBlue("component", "Service") else skinPlantUmlWhite("component")
           )
 
-        case Database(name, replicas) =>
+        case Database(name, numReplicas) =>
+          val replicas =
+            (1 to numReplicas)
+              .map(n => PlantUml.Database(s"replica-$n", None, Option.when(isBright)("Database"), xs = Set.empty))
+              .toSet[PlantUml.Entity]
+
           PlantUml.ComponentDiagram(
             PlantUml.Package(
               "Persistence",
-              PlantUml.Database(name, None, Option.when(isBright)("Database"), xs = Set.empty)
+              replicas + PlantUml.Database(name, None, Option.when(isBright)("Database"), xs = Set.empty)
             ),
             if isBright then skinPlantUmlYellow("database", "Database") else skinPlantUmlWhite("database")
           )