chore: update the event handler with medical technology data

SmartOperatingBlock · May 9, 2023 · ab454ae · ab454ae
1 parent 7ac9b68
commit ab454ae
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/src/main/kotlin/application/handler/ProcessEventHandlers.kt b/src/main/kotlin/application/handler/ProcessEventHandlers.kt
@@ -274,9 +274,22 @@ object ProcessEventHandlers {
                             MedicalDeviceServices.GetMedicalDeviceUsageByProcessId(
                                 surgicalProcess.id,
                                 medicalDeviceRepository
-                            ).execute().toList().mapNotNull {
+                            ).execute().mapNotNull {
                                 MedicalDeviceServices.GetMedicalDeviceById(it, medicalDeviceRepository).execute()
-                            }
+                            },
+                            MedicalDeviceServices.GetMedicalTechnologyUsageByProcessId(
+                                surgicalProcess.id,
+                                medicalDeviceRepository
+                            ).execute().map {
+                                Pair(
+                                    it.first,
+                                    MedicalDeviceServices.GetMedicalTechnologyById(
+                                        it.second,
+                                        medicalDeviceRepository,
+                                        it.third
+                                    ).execute()
+                                )
+                            }.filterPairs()
                         )
                     )
                 )
@@ -291,6 +304,12 @@ object ProcessEventHandlers {
                 ).execute()
             }
             PatientDataServices.DeletePatient(surgicalProcess.patientId, patientRepository).execute()
+            MedicalDeviceServices.GetMedicalDeviceUsageByProcessId(
+                surgicalProcess.id,
+                medicalDeviceRepository
+            ).execute().forEach {
+                MedicalDeviceServices.DeleteImplantableMedicalDevice(it, medicalDeviceRepository).execute()
+            }
         }
     }
 
@@ -386,6 +405,14 @@ object ProcessEventHandlers {
         operation()
     } else false
 
+    /**
+     * Utility function to filter the null values of a list of pair.
+     */
+    fun <T, U> List<Pair<T?, U?>>.filterPairs(): List<Pair<T, U>> =
+        mapNotNull { (t, u) ->
+            if (t == null || u == null) null else t to u
+        }
+
     /**
      * Utility function to check if a [SurgicalProcess] is over.
      */