fix: #1092 StoryBook chapter number displayed incorrectly

- Correct wrong sort order of storybook chapters
elimu-ai · Dec 3, 2020 · 452b9c4 · 452b9c4
1 parent 278619f
commit 452b9c4
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/main/java/ai/elimu/web/content/storybook/StoryBookListController.java b/src/main/java/ai/elimu/web/content/storybook/StoryBookListController.java
@@ -1,7 +1,10 @@
 package ai.elimu.web.content.storybook;
 
+import ai.elimu.dao.StoryBookChapterDao;
 import org.apache.logging.log4j.Logger;
 import ai.elimu.dao.StoryBookDao;
+import ai.elimu.model.content.StoryBook;
+import ai.elimu.model.content.StoryBookChapter;
 import ai.elimu.model.enums.ReadingLevel;
 import org.apache.logging.log4j.LogManager;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -18,6 +21,9 @@ public class StoryBookListController {
 
     @Autowired
     private StoryBookDao storyBookDao;
+
+    @Autowired
+    private StoryBookChapterDao storyBookChapterDao;
 
     @RequestMapping(method = RequestMethod.GET)
     public String handleRequest(Model model) {
@@ -28,6 +34,20 @@ public String handleRequest(Model model) {
         model.addAttribute("storyBooksLevel3", storyBookDao.readAllOrdered(ReadingLevel.LEVEL3));
         model.addAttribute("storyBooksLevel4", storyBookDao.readAllOrdered(ReadingLevel.LEVEL4));
         model.addAttribute("storyBooksUnleveled", storyBookDao.readAllUnleveled());
+
+        // Correct wrong sort order of storybook chapters
+        // TODO: remove this once executed
+        for (StoryBook storyBook : storyBookDao.readAll()) {
+            logger.info("Fixing chapter sortOrder for StoryBook " + storyBook.getId());
+            Integer sortOrder = 0;
+            for (StoryBookChapter storyBookChapter : storyBookChapterDao.readAll(storyBook)) {
+                logger.info("storyBookChapter.getId(): " + storyBookChapter.getId());
+                logger.info("sortOrder: " + sortOrder);
+                storyBookChapter.setSortOrder(sortOrder);
+                storyBookChapterDao.update(storyBookChapter);
+                sortOrder++;
+            }
+        }
 
         return "content/storybook/list";
     }