如何通过扩展方法简化Fragment中连续作用域函数的重复代码？

问题背景 #

在Fragment中大量重复使用以下协程代码来收集Flow状态：

lifecycleScope.launch {
    viewLifecycleOwner.repeatOnLifecycle(Lifecycle.State.STARTED) {
        viewModel.StateOfSomething.collectLatest {
            withContext(Dispatchers.Main) {
                // 处理逻辑
            }
        }
    }
}

希望将其简化为：

myScopeLaunch {
    viewModel.StateOfSomething.collectLatest {
        withContext(Dispatchers.Main) {
            // 处理逻辑
        }
    }
}

若能进一步简化为以下形式且确保运行在主线程则更佳：

myScopeLaunch {
   viewModel.StateOfSomething.collectLatest {
       // 处理逻辑（自动在主线程执行）
   }
}

解决方案 #

1. 基础版扩展：实现myScopeLaunch

通过给Fragment添加扩展函数，封装重复的lifecycleScope.launch和repeatOnLifecycle逻辑：

import androidx.fragment.app.Fragment
import androidx.lifecycle.Lifecycle
import androidx.lifecycle.repeatOnLifecycle
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.launch

fun Fragment.myScopeLaunch(block: suspend CoroutineScope.() -> Unit) {
    lifecycleScope.launch {
        viewLifecycleOwner.repeatOnLifecycle(Lifecycle.State.STARTED, block)
    }
}

使用时就可以直接调用myScopeLaunch，内部逻辑和原代码完全一致：

myScopeLaunch {
    viewModel.StateOfSomething.collectLatest {
        withContext(Dispatchers.Main) {
            // 处理逻辑
        }
    }
}

2. 优化版：自动确保主线程执行

实际上，lifecycleScope的默认调度器是Dispatchers.Main.immediate，且repeatOnLifecycle在Lifecycle.State.STARTED状态下启动的协程本身就运行在主线程。如果你的Flow没有通过flowOn切换到后台调度器，collectLatest的回调会直接在主线程执行，完全可以去掉withContext(Dispatchers.Main)。

如果需要强制确保回调在主线程（比如某些Flow可能在后台发射），可以在扩展函数中指定上下文：

fun Fragment.myScopeLaunch(block: suspend CoroutineScope.() -> Unit) {
    lifecycleScope.launch(Dispatchers.Main) {
        viewLifecycleOwner.repeatOnLifecycle(Lifecycle.State.STARTED) {
            block()
        }
    }
}

这样就能直接使用最简化的形式：

myScopeLaunch {
   viewModel.StateOfSomething.collectLatest {
       // 处理逻辑（自动运行在主线程）
   }
}

内容的提问来源于stack exchange，提问作者Metropol_Tilkisi