From a8072b8c1af0aee40795832341d94bbb0efdc15b Mon Sep 17 00:00:00 2001
From: David Watkins <davidwatkins73@gmail.com>
Date: Mon, 29 Jan 2024 13:34:48 +0000
Subject: [PATCH] Sample data loader

#6948
---
 .../finos/waltz/jobs/example/DemoDataLoader.java   | 14 ++++++++++----
 1 file changed, 10 insertions(+), 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
index 4c172e8ba5..20f5758026 100644
--- 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) {