使用具有自定义范围并查看生命周期的Kotlin协程进行轮询

时间:2019-08-02 11:11:14

标签: android kotlin coroutine kotlin-coroutines

我只是从Kotlin协程开始。我正在尝试使用协程轮询服务器,并且想在ActivityFragment暂停时停止轮询,并相应地恢复轮询。因此,我的pollScope的生命周期比ViewModel.viewModelScope提供的生命周期短。我对目前的实施方式不完全满意,有几个问题:

  1. 这是创建pollScope的正确方法。我也想在取消viewModelScope时取消它,所以这就是我指定父级工作的原因。
  2. 如果我使用onResume()取消了pollJobs,为什么协程不能从coroutineContext.cancel()开始?如果我保留一份工作清单并取消它们,它们就可以正常工作。
  3. 这是整体正确的方法吗?有更好的方法吗?
    import androidx.lifecycle.LifecycleOwner
    import androidx.lifecycle.ViewModel
    import androidx.lifecycle.viewModelScope
    import com.spruce.messenger.utils.FullLifecycleObserverAdapter
    import kotlinx.coroutines.*
    import java.io.IOException
    import java.util.concurrent.CopyOnWriteArrayList
    import kotlin.coroutines.CoroutineContext


    suspend fun poll(initialDelay: Long = 5000,
                     maxDelay: Long = 30000,
                     factor: Double = 2.0,
                     block: suspend () -> Unit) {

        var currentDelay = initialDelay
        while (true) {
            try {
                try {
                    block()
                    currentDelay = initialDelay
                } catch (e: IOException) {
                    currentDelay = (currentDelay * factor).toLong().coerceAtMost(maxDelay)
                }
                delay(currentDelay)
                yield()
            } catch (e: CancellationException) {
                break
            }
        }
    }


    class MyDataModel : ViewModel() {
        val pollScope = CloseableCoroutineScope(SupervisorJob(parent = viewModelScope.coroutineContext[Job]) + Dispatchers.Main)

        private val pollJobs = CopyOnWriteArrayList<Job>()

        inner class CloseableCoroutineScope(context: CoroutineContext) : FullLifecycleObserverAdapter(), CoroutineScope {
            override val coroutineContext: CoroutineContext = context

            override fun onPause(owner: LifecycleOwner) {
                super.onPause(owner)
                // coroutineContext.cancel() // this cancels it but then coroutine doesn't start again in onResume()
                pollJobs.forEach { it.cancel() }
            }

            override fun onResume(owner: LifecycleOwner) {
                super.onResume(owner)
                refresh()
            }
        }

        fun refresh() {
            if (pollJobs.count { it.isActive } == 0) {
                startPoll()
            }
        }

        private fun startPoll() = pollScope.launch {
            try {
                poll {
                    //fetch data from server
                }
            } catch (e: Exception) {
                //ignore
            }
        }.also {
            track(it)
        }

        private fun track(job: Job) {
            pollJobs.add(job)
            job.invokeOnCompletion {
                pollJobs.remove(job)
            }
        }
    }

然后在我的片段中,将pollScope添加为生命周期观察者viewLifecycleOwner.lifecycle.addObserver(viewModel.pollScope)

1 个答案:

答案 0 :(得分:2)

  1. 您的pollScope对我来说还不错。

  2. 取消Job时,它会取消该Job的所有协同程序。

  3. 我将ViewModelCoroutineScope一起使用,然后从中轮询。确保在虚拟机死亡时管理我的Job并取消我的协程。

class MyViewModel() : ViewModel(),
CoroutineScope by CoroutineScope(Dispatchers.Main + SupervisorJob()) {
    // ...
    override fun onCleared() {
        cancel()
        super.onCleared()
    }
}