Android Arch Component Paging Retryable DataSource

RetryableDataSource.kt

package com.example

import androidx.paging.PositionalDataSource

abstract class RetryableDataSource : PositionalDataSource<String>() {
    private var retryAction: (() -> Any)? = null

    override fun loadRange(params: LoadRangeParams, callback: LoadRangeCallback<String>) {
        try {
            val result = listOf("Api result data")

            callback.onResult(result)
        } catch (exception: Exception) {
            retryAction = { loadRange(params, callback) }

            callback.onResult(emptyList())
        }
    }

    fun retry() {
        if (retryAction != null) {
            retryAction!!.invoke()
            retryAction = null
        } else {
            invalidate()
        }
    }
}