Android jetpack 上的分页组合 LazyColumn

Question

我正在使用 jetpack compose 开发 Android 聊天应用程序。

聊天消息来自 WebSocket。 （我使用 OkHttp 库）

并且所有消息都应存储在本地数据库中（我使用 Room）

当用户滚动到顶部时，应用程序会尝试从本地数据库获取消息。 如果本地数据库没有数据，它会尝试从远程 API 服务器获取消息。

而这个逻辑是通过Repository模式来实现的。

class ChatMessageRepository (
    private val localMessageDao: LocalMessageDao,
    private val remoteApiServer: RemoteApiServer
) {
    fun getMessages(offset: Int) {
        val cachedMessages = localMessageDao.get(offset)
        if (cachedMessages.isNotEmpty()) {
            return cachedMessages
        }

        return remoteApiServer.getMessages(offset)
            .also {
                localMessageDao.insert(it)
            }
    }
}

我可以将其与 PagingSource 集成。

class ChatPagingSource(
    private val repo: ChatMessageRepository 
): PagingSource<Int, Chat>() {

    override suspend fun load(params: LoadParams<Int>): LoadResult<Int, Chat> {
        val offset = params.key ?: 1
        val messages = repo.getMessages(offset)
        return LoadResult.Page(
            data = messages ,
            prevKey = if (offset == 1) null else offset - messages .size,
            nextKey = null
        )
    }

}

但问题是我不知道如何将WebSocket数据流与PagingSource集成。

class WebSocketService: WebSocketListener() {
    val messageSubject: PublishSubject<WsMsg> = PublishSubject.create()

    override fun onMessage(webSocket: WebSocket, text: String) {
        super.onMessage(webSocket, text)
        messageSubject.onNext(friendStatus)
    }
}

我正在使用 OkHttp 库回调方法接收聊天消息。

我该怎么做？

我是否应该放弃使用 Paging 库，而只使用 LazyColumn 滚动检测进行开发？