swift - Vapor Swift - 从两个模型获取数据

Question

我在 Vapor 上有下一个功能:

func getPartidosHandler(_ req: Request) throws -> Future<[PartidoWSData]> {
    return Partido.query(on: req).filter(\.estado == nil).all().map(to: [PartidoWSData].self) { partidos in
        var partidosWS: [PartidoWSData] = []
        for partido in partidos {

            // Something here

        }
        return partidosWS
    }
}

下一个结构体 PartidoWSData:

struct PartidoWSData: Content {
    let idPartido: String
    let fecha: String
    let sede1: Future<Sede>
    let sede2: Future<Sede>
}

我的模型 Partido 有两个对 Sede 的引用，“sede1”和“sede2”。

我想要的是该函数给出一个 PartidoWSData 结构数组，其中我可以看到“Partido”、“idPartido”和“fecha”的两个属性，以及与模型相关的两个 Sede。

我该怎么做？

谢谢!

Answer 1

我不确定 Partido 和 Sede 之间到底存在什么类型的关系，因为这里没有包含模型，但假设它是父子关系，你应该能够做这样的事情:

func getPartidosHandler(_ req: Request) throws -> Future<[PartidoWSData]> {
    return Partido.query(on: req).filter(\.estado == nil).all().flatMap { partidos -> Future<[PartidoWSData]> in
        let partidoIDs = try partidos.map { try $0.requireID() }
        return Sede.query(on: req).filter(\.partidoID ~~ partidoIDs).map { sedes -> [PartidoWSData] in
            return partidos.map { partido -> PartidoWSData in
                return PartidoWSData(
                    id: partido.id
                    sedes: sedes.filter { $0.partidoID == partido.id }
                )
            }
        }
    }
}

关键是使用 ~~ 运算符来做一个 x IN (...) 谓词，然后使用 Array.filter以获得适当的结果。