Collecting flows by using an extension function

FlowWithExtension.kt

/** Making this extension function cleans out the code a lot by removing the repetitive
flowWithLifecycle() or repeatOnLifecycle()
**/
fun <T> Flow<T>.safeCollect(
    owner: LifecycleOwner,
    block: (T.() -> Unit)? = null
) = owner.lifecycleScope.launch {
    flowWithLifecycle(owner.lifecycle).collectLatest { block?.invoke(it) }
}

//This Extension function can easily be used like
flow.safeCollect(viewLifecycleOwner){
    //Doing the work
}