From 217eeab70665190493bcc2208e879568b06608d5 Mon Sep 17 00:00:00 2001 From: Sylwester Guzek Date: Thu, 10 May 2018 14:49:30 +0200 Subject: [PATCH] total count fixed --- .../teamtreehouse/flashy/controllers/IndexController.java | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/teamtreehouse/flashy/controllers/IndexController.java b/src/main/java/com/teamtreehouse/flashy/controllers/IndexController.java index f127754..b79a453 100644 --- a/src/main/java/com/teamtreehouse/flashy/controllers/IndexController.java +++ b/src/main/java/com/teamtreehouse/flashy/controllers/IndexController.java @@ -12,6 +12,7 @@ @Controller public class IndexController { + public static final int AMOUNT_TO_SHOW = 3; private FlashCardService flashCardService; @Autowired @@ -22,7 +23,7 @@ public void setFlashCardService(FlashCardService flashCardService) { @RequestMapping("/") public String index(Model model) { StringBuilder ctaBuilder = new StringBuilder(); - List cards = flashCardService.getRandomFlashCards(5); + List cards = flashCardService.getRandomFlashCards(AMOUNT_TO_SHOW); ctaBuilder.append("Refresh your memory about "); for (FlashCard card : cards) { ctaBuilder.append(card.getTerm()); @@ -32,7 +33,7 @@ public String index(Model model) { } ctaBuilder.append(" and "); Long totalCount = flashCardService.getCurrentCount(); - ctaBuilder.append(totalCount); + ctaBuilder.append(totalCount - AMOUNT_TO_SHOW); ctaBuilder.append(" more"); model.addAttribute("cta", ctaBuilder.toString()); model.addAttribute("flashCardCount", totalCount);