From b3aa0d8e32ea6d8b441765f11eff2402c2b1bbd3 Mon Sep 17 00:00:00 2001 From: Trey t Date: Thu, 3 Feb 2022 15:32:58 -0600 Subject: [PATCH] update code to be mas fun --- Shared/Models/ContentModeViewModel.swift | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/Shared/Models/ContentModeViewModel.swift b/Shared/Models/ContentModeViewModel.swift index 96daee0..a9f8197 100644 --- a/Shared/Models/ContentModeViewModel.swift +++ b/Shared/Models/ContentModeViewModel.swift @@ -18,15 +18,11 @@ class ContentModeViewModel: ObservableObject { } private var numberOfEntries: Int { - var num = 0 - grouped.keys.forEach({ - let year = grouped[$0] - let monthKeys = year?.keys - monthKeys?.forEach({ - num += year![$0]!.count - }) - }) - return num + grouped.keys.map{ + grouped[$0]!.values.reduce(0) { sum, array in + sum + array.count + } + }.reduce(0, +) } init() { @@ -40,6 +36,7 @@ class ContentModeViewModel: ObservableObject { private func getGroupedData() { grouped = PersistenceController.shared.splitIntoYearMonth() numberOfItems = numberOfEntries + print(numberOfItems) } public func shouldShowVotingHeader() -> Bool {