GRDB.swift 中处理关联查询与 NULL 值的实践指南

引言

在使用 GRDB.swift 进行数据库操作时，开发者经常会遇到需要从关联表中聚合数据的场景。本文将通过一个实际案例，深入探讨如何在 GRDB.swift 中正确处理关联查询、聚合函数以及可能出现的 NULL 值问题。

问题背景

在医疗类应用中，我们经常需要查询患者信息及其相关的就诊记录。例如，我们需要获取患者的基本信息，同时计算他们在特定日历中的最早创建时间和最晚就诊日期。

初始查询实现

最初的查询实现如下：

func getPatientTimeline() -> [PatientTimelineInfo] {
    var patientTimelineInfo = [PatientTimelineInfo]()
    do {
        patientTimelineInfo = try db.read { db in
            let request = Patient
                .select(
                    Column("id"),
                    Column("patientName")
                )
                .order(Column("patientName"))
                .annotated(with:
                    Patient.visits.min(Column("visitCreated")),
                    Patient.visits.max(Column("visitDate")))
            return try PatientTimelineInfo.fetchAll(db, request)
        }
    } catch {
        fatalError("Unresolved error \(error)")
    }
    return patientTimelineInfo
}

这个查询会返回所有患者及其就诊记录中的最早创建时间和最晚就诊日期。

添加过滤条件

然而，业务需求要求我们只考虑特定日历（如"Marieke"和"Marieke nieuwe"）中的就诊记录。我们尝试修改查询：

func getPatientTimeline1() -> [PatientTimelineInfo] {
    var patientTimeline = [PatientTimelineInfo]()
    do {
        patientTimeline = try db.read { db in
            let filteredVisits = Patient.visits
                .filter(["Marieke", "Marieke nieuwe"].contains(Column("visitCalendar")))
            let request = Patient
                .select(
                    Column("id"),
                    Column("patientName")
                )
                .order(Column("patientName"))
                .annotated(with:
                    filteredVisits.min(Column("visitCreated")),
                    filteredVisits.max(Column("visitDate"))
                )
            return try PatientTimelineInfo.fetchAll(db, request)
        }
    } catch {
        fatalError("Unresolved error \(error)")
    }
    return patientTimeline
}

遇到的错误

执行上述查询时，我们遇到了以下错误：

GRDB/FetchableRecord+Decodable.swift:710: Assertion failed: unexpected NULL value

生成的SQL语句显示，GRDB在LEFT JOIN中重复添加了过滤条件：

SELECT "patient"."id", "patient"."patientName", 
       MIN("visit"."visitCreated") AS "minVisitVisitCreated", 
       MAX("visit"."visitDate") AS "maxVisitVisitDate" 
FROM "patient" 
LEFT JOIN "visit" ON ("visit"."patientId" = "patient"."id") 
                  AND ("visit"."visitCalendar" IN (?, ?)) 
                  AND ("visit"."visitCalendar" IN (?, ?)) 
GROUP BY "patient"."id" 
ORDER BY "patient"."patientName"

问题分析与解决方案

问题根源

1. NULL值问题：当患者没有符合条件的就诊记录时，MIN和MAX聚合函数会返回NULL值。如果我们的PatientTimelineInfo模型将这些字段定义为非可选类型，就会触发断言错误。

2. 重复过滤条件：GRDB生成的SQL中出现了重复的过滤条件，这虽然不影响查询结果，但可能影响性能。

解决方案

正确的做法是：

1. 确保模型中的聚合字段定义为可选类型（如Date?）
2. 添加HAVING子句排除没有符合条件的就诊记录的患者

修改后的查询：

func getPatientTimeline1() -> [PatientTimelineInfo] {
    var patientTimeline = [PatientTimelineInfo]()
    do {
        patientTimeline = try db.read { db in
            let filteredVisits = Patient.visits
                .filter(["Marieke", "Marieke nieuwe"].contains(Column("visitCalendar")))
            let request = Patient
                .select(
                    Column("id"),
                    Column("patientName")
                )
                .order(Column("patientName"))
                .annotated(with:
                    filteredVisits.min(Column("visitCreated")),
                    filteredVisits.max(Column("visitDate"))
                )
                .having(filteredVisits.isEmpty == false)
            return try PatientTimelineInfo.fetchAll(db, request)
        }
    } catch {
        fatalError("Unresolved error \(error)")
    }
    return patientTimeline
}

最佳实践建议

1. 处理NULL值：在设计模型时，考虑聚合字段可能为NULL的情况，使用可选类型。

2. 明确查询意图：使用HAVING子句明确表达"只返回有相关记录的结果"的意图。

3. 性能考虑：对于大型数据集，考虑添加适当的索引，特别是在用作过滤条件的列上。

4. 错误处理：避免使用fatalError，改为提供有意义的错误信息或恢复路径。

总结

通过这个案例，我们学习了如何在GRDB.swift中正确处理关联查询和聚合函数。关键在于理解SQL查询的语义和GRDB的转换规则，特别是当涉及到可能返回NULL值的聚合操作时。正确的模型设计和查询构造可以避免运行时错误，并确保应用稳定运行。