Sample data loader

finos#6948
davidwatkins73 · Apr 14, 2024 · b609117 · b609117
1 parent a40efb6
commit b609117
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/waltz-jobs/src/main/java/org/finos/waltz/jobs/example/DemoDataLoader.java b/waltz-jobs/src/main/java/org/finos/waltz/jobs/example/DemoDataLoader.java
@@ -112,10 +112,7 @@ private void go() throws IOException {
             processDataTypes(waltz, workbook.getSheet("data-types"), now);
             processDataFlows(waltz, workbook.getSheet("flows"), now);
 
-            LOG.debug("Rebuilding categories");
-            hierarchyService.buildFor(EntityKind.ORG_UNIT);
-            hierarchyService.buildFor(EntityKind.MEASURABLE);
-            hierarchyService.buildFor(EntityKind.PERSON);
+            rebuildHierarchies();
 
 
             throw new RuntimeException("rolling back, comment this line if you really want to do this!");
@@ -125,6 +122,15 @@ private void go() throws IOException {
     }
 
 
+    private void rebuildHierarchies() {
+        LOG.debug("Rebuilding hierarchies");
+        hierarchyService.buildFor(EntityKind.ORG_UNIT);
+        hierarchyService.buildFor(EntityKind.MEASURABLE);
+        hierarchyService.buildFor(EntityKind.PERSON);
+        hierarchyService.buildFor(EntityKind.DATA_TYPE);
+    }
+
+
     private void processDataFlows(DSLContext waltz,
                                   Sheet flowSheet,
                                   Timestamp now) {