avoiding multiple list iterations via mapNotNull

2021-10-21 12:03:41 +01:00 · 2021-10-21 12:03:41 +01:00 · b146501f29
parent be67836a3e
commit b146501f29
1 changed files with 3 additions and 1 deletions
--- a/vector/src/main/java/im/vector/app/features/notifications/ProcessedEvent.kt
+++ b/vector/src/main/java/im/vector/app/features/notifications/ProcessedEvent.kt
@ -27,4 +27,6 @@ data class ProcessedEvent<T>(
    }
 }

-fun <T> List<ProcessedEvent<T>>.onlyKeptEvents() = filter { it.type == ProcessedEvent.Type.KEEP }.map { it.event }
+fun <T> List<ProcessedEvent<T>>.onlyKeptEvents() = mapNotNull { processedEvent ->
+    processedEvent.event.takeIf { processedEvent.type == ProcessedEvent.Type.KEEP }
+}