SamplePagingSource.kt

class SamplePagingSource(
    private val sampleRepository: SampleRepository
) : PagingSource<Int, SampleEntity>() {

    override suspend fun load(params: LoadParams<Int>): LoadResult<Int, SampleEntity> {
        return try {
            val data = sampleRepository.getNextPage(lastSeenId = params.key ?: 1)
                .map { SampleEntity(it, "Page number: $it") }
            LoadResult.Page(
                data = data,
                prevKey = null,
                nextKey = if (data.isNotEmpty()) {
                    data.last().id + 1
                } else {
                    // return null when page is empty to denote the end of the pagination
                    null
                }
            )
        } catch (e: Exception) {
            LoadResult.Error(e)
        }
    }
}