Возможно ли использовать Vapor 3 Postgres Fluent в автономном скрипте?

Question

Я экспериментирую с автономным скриптом, который будет запрашивать базу данных Postgres, используя Vapor и Fluent. В обычном приложении Vapor API это делается просто:

router.get("products") { request in
    return Product.query(on: request).all()
}

Однако в автономном скрипте, так как нет "запроса", я застреваю на том, что заменить "запрос" или DatabaseConnectable с. Вот где я застреваю:

import Fluent
import FluentPostgreSQL

let databaseConfig = PostgreSQLDatabaseConfig(hostname: "localhost",
                                              username: "test",
                                              database: "test",
                                              password: nil)

let database = PostgreSQLDatabase(config: databaseConfig)

let foo = Product.query(on: <??WhatDoIPutHere??>)

Я пытался создать объект, который соответствует DatabaseConnectable, но не мог понять, как правильно заставить этот объект соответствовать.

Answer 1

Вам нужно будет создать группу цикла событий, чтобы иметь возможность делать запросы к базе данных. SwiftNIO-х MultiThreadedEventLoopGroup хорошо для этого:

let worker = MultiThreadedEventLoopGroup(numberOfThreads: 2)

Вы можете изменить количество используемых потоков, как вам нужно.

Теперь вы можете создать соединение с базой данных с этим работником:

let conn = try database.newConnection(on: worker)

Связь в будущем, так что вы можете map это и передать соединение в вашем запросе:

conn.flatMap { connection in
    return Product.query(on: connection)...
}

Убедитесь, что вы выключили своего работника, когда закончили с ним, используя shutdownGracefully(queue:_:)

Answer 2

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

final class StandAloneTest : XCTestCase{
    var expectation : XCTestExpectation?
    func testDbConnection() -> Void {
        expectation = XCTestExpectation(description: "Wating")
        let databaseConfig = PostgreSQLDatabaseConfig(hostname: "your.hostname.here",
                                                      username: "username",
                                                      database: "databasename",
                                                      password: "topsecretpassword")
        let database = PostgreSQLDatabase(config: databaseConfig)
        let worker = MultiThreadedEventLoopGroup(numberOfThreads: 2)
        let conn = database.newConnection(on: worker)

        let sc = SomeClass( a:1, b:2, c:3  ) //etc

        //get all the tupples for this Class type in the base
        let futureTest = conn.flatMap { connection in
              return SomeClass.query(on: connection).all()
        }
        //or save a new tupple by uncommenting the below
        //let futureTest = conn.flatMap { connection in
        //    return someClassInstantiated.save(on: connection)
        //}

        //lets just wait for the future to test it 
        //(PS: this blocks the thread and should not be used in production)
        do{
            let test = try futureTest.wait()
            expectation?.fulfill()
            worker.syncShutdownGracefully()
            print( test )
        }catch{
            expectation?.fulfill()
            print(error)
        }
    }
}

//Declare the class you want to test here using the Fluent stuff in some extension