Android开发：如何在ViewModel中判断PagingData<T>是否为空？

要在ViewModel中判断PagingData<ShoppingListItem>是否为空，最可靠的方式是结合Room数据库直接查询条目总数——因为PagingData是流式分页数据，无法直接判断整体是否为空，而查询总数能直接、高效地获取数据集状态。以下是具体实现步骤：

1. 添加Room查询总数的Dao方法 #

在你的ShoppingListDao中新增查询条目总数的方法，返回Flow<Int>以监听数据变化：

@Dao
interface ShoppingListDao {
    // 原有分页查询方法
    @Query("SELECT * FROM shopping_list_items")
    fun getAllItems(): PagingSource<Int, ShoppingListItem>

    // 新增：查询条目总数
    @Query("SELECT COUNT(*) FROM shopping_list_items")
    fun getItemCount(): Flow<Int>
}

2. 新增获取总数的UseCase #

创建一个UseCase封装总数查询逻辑：

class GetShoppingListItemCountUseCase @Inject constructor(
    private val shoppingListDao: ShoppingListDao
) {
    operator fun invoke(): Flow<Int> {
        return shoppingListDao.getItemCount()
    }
}

3. 调整ViewModel逻辑 #

修改ViewModel，通过监听总数变化来切换Empty/Success状态：

@HiltViewModel
class ShoppingListScreenViewModel @Inject constructor(
    private val getAllShoppingListItemsUseCase: GetAllShoppingListItemsUseCase,
    private val getShoppingListItemCountUseCase: GetShoppingListItemCountUseCase // 注入新增的UseCase
) {
    private val _listItemsLoadingState = MutableStateFlow<ListItemsState>(ListItemsState.Loading)
    val listItemsLoadingState = _listItemsLoadingState.asStateFlow()

    init {
        getAllShoppingListItemsFromDb()
    }

    private fun getAllShoppingListItemsFromDb() {
        viewModelScope.launch {
            // 获取分页数据Flow
            val pagingFlow = getAllShoppingListItemsUseCase().distinctUntilChanged()
            
            // 监听总数变化，实时更新状态
            getShoppingListItemCountUseCase().collect { itemCount ->
                _listItemsLoadingState.value = when (itemCount) {
                    0 -> ListItemsState.Empty
                    else -> ListItemsState.Success(pagingFlow)
                }
            }
        }
    }
}

4. 调整UI层显示空状态 #

在ShoppingListScreen的状态判断分支中添加Empty状态的UI：

@Composable
fun ShoppingListScreen(
    navController: NavHostController,
    shoppingListScreenViewModel: ShoppingListScreenViewModel,
    sharedViewModel: SharedViewModel
) {
    val screenHeight = LocalConfiguration.current.screenHeightDp.dp
    val allItemsState = shoppingListScreenViewModel.listItemsLoadingState.collectAsState().value

    Scaffold(/* 原有Scaffold配置 */) {
        Box {
            when (allItemsState) {
                is ListItemsState.Loading -> ConditionalCircularProgressBar(isDisplayed = true)
                is ListItemsState.Error -> Text("Error!")
                is ListItemsState.Empty -> {
                    // 显示空状态提示
                    Column(
                        modifier = Modifier.fillMaxSize(),
                        verticalArrangement = Arrangement.Center,
                        horizontalAlignment = Alignment.CenterHorizontally
                    ) {
                        Text(text = "购物列表为空，点击添加按钮开始吧！")
                    }
                }
                is ListItemsState.Success -> {
                    // 原有列表显示逻辑
                    ConditionalCircularProgressBar(isDisplayed = false)
                    val successItems = allItemsState.allItems?.collectAsLazyPagingItems()
                    LazyColumn(/* 原有LazyColumn配置 */) {
                        // 原有item渲染逻辑
                    }
                }
                else -> {}
            }
        }
    }
}

方案说明 #

• 直接通过Room查询总数能准确、高效判断数据集是否为空，避免了PagingData分页加载带来的判断误差
• 监听总数的Flow可以实时响应数据变化（比如添加/删除条目），自动切换状态
• 保持了ViewModel对状态的统一管理，UI层只需根据状态渲染对应内容

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