Swift Vapor 在自定义回复中添加额外信息

Question

我有两个模型，行程和位置。我会 return 一个自定义响应，其中包含一些旅行领域和 tripID 等于 Trip id 的位置数。有我的代码（不工作）。字段 locationCount 始终为空。

func getList(_ request: Request)throws -> Future<Response> {

    let deviceIdReq = request.parameters.values[0].value
    let queryTrips = Trip.query(on: request).filter(\.deviceId == deviceIdReq).all()
    var tripsR = [TripCustomContent]()
    var trips = [Trip]()

    return queryTrips.flatMap { (result) -> (Future<Response>) in
        trips = result

        var count = 0
        for t in trips {
            let tripIdString = String(t.id!)
            let v = Location.query(on: request).filter(\.tripID == tripIdString).count().map({ (res) -> Int in
                return res
            })/*.map{ (result) -> (Int) in
             count = result
             return result
             }*/
            let tripCustomContent = TripCustomContent.init(startTimestamp: t.startTimestamp, endTimestamp: t.endTimestamp, deviceId: t.deviceId, locationCount: v)
            tripsR.append(tripCustomContent)
        }
        let jsonEncoder = JSONEncoder()
        let data = try jsonEncoder.encode(tripsR)
        let response = HTTPResponse.init(status: .ok, version: HTTPVersion.init(major: x, minor: y), headers: HTTPHeaders.init(), body: data)
        let finalResponse = Response.init(http: response, using: request)
        return try g.encode(for: request)
    }
}

这是我的自定义内容结构：

struct TripCustomContent: Encodable {
var startTimestamp: String?
var endTimestamp: String?
var deviceId: String
var locationCount: Future<Int>
}

有什么建议吗？

Answer 1

您正在尝试使用一个尚不可用的值。当您 returning 一个 Future 时，您并没有 returning 里面的值。

所以你希望你的 TripCustomContent 像这样（在 vapor Content 中使用而不是 Codable:

struct TripCustomContent: Content {
    var startTimestamp: String?
    var endTimestamp: String?
    var deviceId: String
    var locationCount: Int
}

您查询 Trip 正确，但 Location 不正确。你也许可以尝试这样的事情：

return queryTrips.flatMap { trips -> Future<[TripCustomContent]> in
    let tripIds = trips.map({ String([=11=].id!) })
    return Location.query(on: request).filter(\.tripID ~~ tripIds).all().map { locations in
        return trips.map { trip in
            let locationCount = locations.filter({ [=11=].tripId == String(trip.id!) }).count
            return TripCustomContent(... locationCount: locationCount)
        }
    }
}

我在这里做了什么？

将行程映射到他们的 tripId 以获得 tripId 数组
获取 tripId 为上述数组中 tripId 之一的所有位置
使用 tripId

TripCustomContent

最后，您不需要自己编码 JSON，只需 return 个符合 Content:

的对象

func getList(_ request: Request) throws -> Future<[TripCustomContent]>

以上可能是您策略的解决方案。但也许你可以看看 relations 它们是否是一种更高效、更简单、更快速的方法。

Swift Vapor 在自定义回复中添加额外信息

Swift Vapor add additional info to custom response

swift

vapor