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 {