如何在Vapor 3中从JSON响应中保存父子关系

时间:2019-10-09 19:31:41

标签: swift vapor vapor-fluent

我正在使用 Vapor 3 开发REST API。此API使用另一个API来创建内容,以后将由应用程序使用。 因此,我创建了一个从此API(联盟和季节)获取内容的函数,并将其存储在我的MySQL数据库中。 API的响应还包含我也想存储的嵌套对象(如果可能的话)都在同一请求中。这是API响应:

{
"data": [
         {
             "id": 271,
             "name": "Superliga",
             "current_season_id": 16020,
             "season": {
                 "data": {
                     "id": 16020,
                     "name": "2019/2020",
                     "league_id": 271,
                 }
             }
        }
    ]
}

这是模型:

final class League: MySQLModel {

    var id: League.ID?

    var name: String

    var current_season_id: Season.ID

    var currentSeason: Parent<League, Season> {
        return parent(\League.current_season_id)
    }
}

final class Season: MySQLModel {

    var id: Season.ID?

    var name: String

    var league_id: League.ID

    var league: Parent<Season, League> {
        return parent(\.league_id)
    }
}

这是执行请求并将其存储在DB中的功能。

func getLeagues(using context: CommandContext) throws -> EventLoopFuture<Void> {
    guard let url = URL(string: "SOME_API_URL") else { return .done(on: context.container) }

    let client = try context.container.client()
    return client.get(url).flatMap({ (response) -> EventLoopFuture<Void> in // do the request
        let leagues = response.content.get([League].self, at: "data") // get the array of EventLoopFuture<[League]>

        return context.container.requestPooledConnection(to: .mysql).flatMap({ (connection) -> EventLoopFuture<Void> in // connecto to DB

            let savedLeagues = leagues.flatMap(to: [League].self, { (flattenLeagues) -> EventLoopFuture<[League]> in
                return flattenLeagues.map { (league) -> EventLoopFuture<League> in

                    return league.create(orUpdate: true, on: connection) // save on the DB

                }.flatten(on: context.container)
            })

            return savedLeagues.flatMap { (_) -> EventLoopFuture<Void> in
                return .done(on: context.container)
            }
        })
    })
}

问题将是:是否可以保存父子关系?我是否必须使用解码/编码功能手动进行? 我确实实现了编码/解码并创建了League,但不知道如何创建Season,以及在进行league.create(orUpdate: true, on: connection)

时如何保存所有内容

将寻求任何帮助。

1 个答案:

答案 0 :(得分:1)

如我所见,您可以先解码APIModel,然后像这样在扁平化循环中保存两个对象

struct APILeague: Content {
    let id: League.ID
    let name: String
    let current_season_id: Season.ID
    struct _Season: Codable {
        let data: Season
    }
    let season: _Season
}
final class League: MySQLModel {
    var id: League.ID?
    var name: String
    var current_season_id: Season.ID

    var currentSeason: Parent<League, Season> {
        return parent(\League.current_season_id)
    }

    init (_ data: APILeague) {
        self.id = data.id
        self.name = data.name
        self.current_season_id = data.current_season_id
    }
}

func getLeagues(using context: CommandContext) throws -> Future<Void> {
    guard let url = URL(string: "SOME_API_URL") else { return .done(on: context.container) }

    let client = try context.container.client()
    return client.get(url).flatMap { response in // do the request
        return response.content.get([APILeague].self, at: "data").flatMap { leagues in
            return context.container.requestPooledConnection(to: .mysql).flatMap { connection in // connecto to DB
                let operations = leagues.map { league in
                    return League(league).create(orUpdate: true, on: connection).flatMap { _ in
                        return league.season.data.create(orUpdate: true, on: connection).transform(to: ()) // transforming to Void
                    } 
                }
                return operations.flatten(on: context.container)flatMap {
                    return .done(on: context.container)
                }
            }
        }
    }
}