Вставить несколько записей в базу данных с Vapor3

Question

Я хочу иметь возможность массового добавления записей в базу данных nosql в Vapor 3.

Это моя структура.

struct Country: Content {

   let countryName: String
   let timezone: String
   let defaultPickupLocation: String

}

Поэтому я пытаюсь передать массив объектов JSON, но я не уверен, как структурировать маршрут и как получить доступ к массиву для декодирования каждого из них.

Я пробовал этот маршрут:

    let countryGroup = router.grouped("api/country")

    countryGroup.post([Country.self], at:"bulk", use: bulkAddCountries)

с этой функцией:

 func bulkAddCountries(req: Request, countries:[Country]) throws ->  Future<String> {
    for country in countries{
    return try req.content.decode(Country.self).map(to: String.self) { countries in



        //creates a JSON encoder to encode the JSON data
        let encoder = JSONEncoder()
        let countryData:Data
        do{
            countryData = try encoder.encode(country) // encode the data
        } catch {
            return "Error. Data in the wrong format."
        }
        // code to save data
    }
    }
}

Итак, как мне структурировать маршрут и функцию, чтобы получить доступ к каждой стране?

Answer 1

Я не уверен, какую базу данных NoSQL вы планируете использовать, но текущие бета-версии MongoKitten 5 и Meow 2.0 делают это довольно легко.

Пожалуйста, обратите внимание, что мы еще не написали документацию для этих двух библиотек, так как сначала обратились к стабильному API. Следующий код примерно то, что вам нужно с MongoKitten 5:

// Register MongoKitten to Vapor's Services
services.register(Future<MongoKitten.Database>.self) { container in
    return try MongoKitten.Database.connect(settings: ConnectionSettings("mongodb://localhost/my-db"), on: container.eventLoop)
}

// Globally, add this so that the above code can register MongoKitten to Vapor's Services
extension Future: Service where T == MongoKitten.Database {}

// An adaptation of your function
func bulkAddCountries(req: Request, countries:[Country]) throws ->  Future<Response> {
    // Get a handle to MongoDB
    let database = req.make(Future<MongoKitten.Database>.self)

    // Make a `Document` for each Country
    let documents = try countries.map { country in
        return try BSONEncoder().encode(country)
    }

    // Insert the countries to the "countries" MongoDB collection
    return database["countries"].insert(documents: documents).map { success in
        return // Return a successful response
    }
}

Answer 2

У меня была аналогичная потребность, и я хочу поделиться своим решением для массовой обработки в Vapor 3. Мне бы хотелось, чтобы другой опытный разработчик помог улучшить мое решение.

Я постараюсь изо всех сил объяснить, что я сделал. И я, наверное, ошибаюсь.

Во-первых, ничего особенного в роутере. Здесь я обрабатываю POST дляitems/batch для массива элементов JSON.

router.post("items", "batch", use: itemsController.handleBatch)

Затем обработчик контроллера.

func createBatch(_ req: Request) throws -> Future<HTTPStatus> {
    // Decode request to [Item]
    return try req.content.decode([Item].self)
        // flatMap will collapse Future<Future<HTTPStatus>> to [Future<HTTPStatus>]
        .flatMap(to: HTTPStatus.self) { items in
        // Handle each item as 'itm'. Transforming itm to Future<HTTPStatus>
        return items.map { itm -> Future<HTTPStatus> in
            // Process itm. Here, I save, producing a Future<Item> called savedItem
            let savedItem = itm.save(on: req)
            // transform the Future<Item> to Future<HTTPStatus>
            return savedItem.transform(to: HTTPStatus.ok)
        }
        // flatten() : “Flattens an array of futures into a future with an array of results”
        // e.g. [Future<HTTPStatus>] -> Future<[HTTPStatus]>
        .flatten(on: req)
        // transform() : Maps the current future to contain the new type. Errors are carried over, successful (expected) results are transformed into the given instance.
        // e.g. Future<[.ok]> -> Future<.ok>
        .transform(to: HTTPStatus.ok)
    }
}