Add check for undefined fields

AplinkosMinisterija · Feb 13, 2024 · 2009b75 · 2009b75
1 parent be8a918
commit 2009b75
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/src/report/report.service.ts b/src/report/report.service.ts
@@ -42,16 +42,20 @@ export class ReportService {
 
   private static docToReportStatistics(e: any): ReportStatisticsDto {
     return new ReportStatisticsDto(
-      e.filter((stat: { _id: string }) => stat._id == 'gautas')[0].count ?? 0,
-      e.filter((stat: { _id: string }) => stat._id == 'tiriamas')[0].count ?? 0,
-      e.filter((stat: { _id: string }) => stat._id == 'ištirtas')[0].count ?? 0,
-      e.filter((stat: { _id: string }) => stat._id == 'sutvarkyta')[0].count ??
-        0,
-      e.filter((stat: { _id: string }) => stat._id == 'nepasitvirtino')[0]
-        .count ?? 0,
+      ReportService.filterStatistics(e, 'gautas'),
+      ReportService.filterStatistics(e, 'tiriamas'),
+      ReportService.filterStatistics(e, 'ištirtas'),
+      ReportService.filterStatistics(e, 'sutvarkyta'),
+      ReportService.filterStatistics(e, 'nepasitvirtino'),
     );
   }
 
+  private static filterStatistics(e: any, status: string): number {
+    return e.filter((stat: { _id: string }) => stat._id == status).length > 0
+      ? e.filter((stat: { _id: string }) => stat._id == status)[0].count ?? 0
+      : 0;
+  }
+
   private static docToPublicReport(e: Report): PublicReportDto {
     return new PublicReportDto(
       e.name,