如何在SwiftData中对关联数据执行类似SELECT COUNT的查询？

问题背景 #

现有两个SwiftData模型：AccountTransaction和TransactionFile，二者为一对多关联。当前通过AccountTransaction的fileCount计算属性获取关联文件数量，但该属性需要加载完整的files数组（包含二进制data字段），造成冗余数据加载。需要在SwiftData原生框架内实现无需拉取冗余关联数据的计数功能，同时保留SwiftData的云同步能力以支持多设备兼容。

当前实现问题 #

当前fileCount依赖self.files?.count，访问files会触发SwiftData加载整个关联对象数组，包括TransactionFile中的data二进制数据，导致不必要的性能开销：

var fileCount: Int {
    self.files?.count ?? 0
}

解决方案 #

1. 使用聚合查询单独获取计数 #

针对每个AccountTransaction，通过独立的聚合查询获取关联文件数量，无需加载完整的TransactionFile对象。

实现步骤：

在AccountTransaction中添加方法，通过ModelContext执行计数查询：

func getFileCount(in context: ModelContext) -> Int {
    let predicate = #Predicate<TransactionFile> { $0.transaction?.id == self.id }
    let descriptor = FetchDescriptor<TransactionFile>(predicate: predicate, fetchLimit: 0)
    do {
        return try context.count(for: descriptor)
    } catch {
        print("Failed to count files: \(error)")
        return 0
    }
}

使用时直接调用该方法并传入当前ModelContext，即可获取计数，不会加载任何TransactionFile的具体数据（包括二进制data）。

2. 预加载计数到AccountTransaction模型（缓存方案） #

如果需要频繁访问计数，可在AccountTransaction中添加持久化的fileCount字段，通过关联对象的增删操作维护该值：

实现步骤：

1. 修改AccountTransaction模型，添加持久化计数字段：

@Model
final class AccountTransaction {
    // ... 原有字段 ...
    var fileCount: Int = 0 // 改为持久化字段，而非计算属性
    // ... 其他内容 ...
}

2. 在TransactionFile的增删操作中更新关联AccountTransaction的fileCount：

// 添加文件时
func addFile(to transaction: AccountTransaction, context: ModelContext) {
    let file = TransactionFile(transaction: transaction)
    transaction.files?.append(file)
    transaction.fileCount += 1
    try? context.save()
}

// 删除文件时
func removeFile(_ file: TransactionFile, context: ModelContext) {
    if let transaction = file.transaction {
        transaction.fileCount -= 1
    }
    context.delete(file)
    try? context.save()
}

此方案优势是计数访问无需额外查询，性能最优；缺点是需要手动维护计数一致性，适合计数频繁访问的场景。

3. 使用FetchDescriptor的propertiesToFetch优化关联加载 #

如果仍希望通过files关联获取计数，但不想加载二进制数据，可通过FetchDescriptor指定仅加载必要属性，避免拉取data字段：

修改查询代码：

在performAccountTransactionFetch中，为关联的files指定仅加载id字段：

var fetchDescriptor = FetchDescriptor<AccountTransaction>(predicate: predicate)
// 指定加载关联files时仅获取id，不加载data等冗余字段
fetchDescriptor.propertiesToFetch = [
    \AccountTransaction.id,
    \AccountTransaction.name,
    // ... 其他需要的AccountTransaction字段 ...
    .relationship(\AccountTransaction.files, properties: [\TransactionFile.id])
]
fetchDescriptor.fetchLimit = transactionsPerPage
fetchDescriptor.fetchOffset = self.currentAccountTransactionPage * transactionsPerPage
fetchDescriptor.sortBy = [.init(\.createdOn, order: .reverse)]

此方法会加载files的轻量对象（仅包含id），避免了二进制data的开销，fileCount仍可通过files?.count获取。

方案选择建议 #

• 若计数访问频率较低，优先选择方案1（聚合查询），无需修改模型结构，实现简单。
• 若计数需要频繁访问，优先选择方案2（持久化计数字段），性能最优，但需要维护数据一致性。
• 若需要同时访问部分关联对象属性（非二进制），可选择方案3，优化关联加载的字段范围。

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