如何按传入列表的顺序对Room查询结果排序？

可以实现按传入的:ids列表顺序排序的逻辑，下面提供两种常用的实现方式：

一、SQL层面直接排序（适用于Room+SQLite场景） #

利用SQL的CASE WHEN或INSTR函数，在查询时直接指定排序规则，确保数据库返回的结果顺序与传入的ID列表一致。

方式1：CASE WHEN动态映射排序权重 #

通过动态生成CASE WHEN语句，为每个ID分配对应的排序索引，适合ID数量不多的场景：

// DAO中定义RawQuery方法
@RawQuery(observedEntities = [ItemEntity::class])
fun getPagingSource(rawQuery: SupportSQLiteQuery): PagingSource<Int, ItemEntity>

// 调用时构建排序SQL
fun getOrderedItems(type: EntityType, ids: List<String>): PagingSource<Int, ItemEntity> {
    // 生成CASE WHEN子句
    val caseClauses = ids.mapIndexed { index, id -> "WHEN '$id' THEN $index" }.joinToString(" ")
    val sql = """
        SELECT * FROM item_table 
        WHERE type = ? AND itemId IN (${ids.joinToString(",") { "'$it'" }})
        ORDER BY CASE itemId $caseClauses END
    """.trimIndent()
    // 构建查询对象，传入类型参数
    val query = SimpleSQLiteQuery(sql, arrayOf(type.name))
    return getPagingSource(query)
}

方式2：INSTR函数匹配位置 #

通过将ID列表拼接为逗号分隔的字符串，利用INSTR函数查找每个ID在字符串中的位置来排序，写法更简洁：

// DAO中定义带排序的查询
@Query("""
    SELECT * FROM item_table 
    WHERE type = :type AND itemId IN (:ids)
    ORDER BY INSTR(',' || :idsString || ',', ',' || itemId || ',')
""")
fun getPagingSource(type: EntityType, ids: List<String>, idsString: String): PagingSource<Int, ItemEntity>

// 调用时传入拼接后的ID字符串
val idsString = ids.joinToString(",")
val orderedPagingSource = getPagingSource(type, ids, idsString)

二、内存层面排序（适配分页场景） #

如果SQL层面的实现受限，也可以在查询到数据后，在内存中按照传入的ID列表顺序重新排序。需要注意适配PagingSource的分页逻辑：

// 自定义PagingSource处理排序
class OrderedItemPagingSource(
    private val dao: ItemDao,
    private val type: EntityType,
    private val ids: List<String>
) : PagingSource<Int, ItemEntity>() {

    override suspend fun load(params: LoadParams<Int>): LoadResult<Int, ItemEntity> {
        val page = params.key ?: 0
        val pageSize = params.loadSize

        // 分页查询原始数据
        val rawItems = dao.getItemsByTypeAndIds(type, ids, page * pageSize, pageSize)
        // 按传入的ids列表顺序排序
        val orderedItems = rawItems.sortedBy { ids.indexOf(it.itemId) }

        // 计算分页键
        val prevKey = if (page > 0) page - 1 else null
        val nextKey = if (orderedItems.size == pageSize) page + 1 else null

        return LoadResult.Page(
            data = orderedItems,
            prevKey = prevKey,
            nextKey = nextKey
        )
    }

    override fun getRefreshKey(state: PagingState<Int, ItemEntity>): Int? {
        return state.anchorPosition?.let { anchorPosition ->
            state.closestPageToPosition(anchorPosition)?.prevKey?.plus(1)
                ?: state.closestPageToPosition(anchorPosition)?.nextKey?.minus(1)
        }
    }
}

// DAO中添加分页查询方法
@Query("""
    SELECT * FROM item_table 
    WHERE type = :type AND itemId IN (:ids)
    LIMIT :limit OFFSET :offset
""")
suspend fun getItemsByTypeAndIds(type: EntityType, ids: List<String>, offset: Int, limit: Int): List<ItemEntity>

两种方式对比 #

• SQL层面排序：数据库直接返回有序结果，分页逻辑更准确，但ID列表过长时可能生成复杂SQL，影响性能；
• 内存层面排序：SQL写法简单，无需动态拼接，但分页后排序可能在边界页出现顺序断层，适合ID列表顺序变化不频繁的场景。

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