swift - 是否可以在独立脚本中使用 Vapor 3 Postgres Fluent？

Question

我正在试验一个独立脚本，该脚本将使用 Vapor 和 Fluent 查询 Postgres 数据库。在普通的 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)

您可以根据需要更改使用的线程数。

现在您可以使用该 worker 创建到数据库的连接:

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

连接在未来，因此您可以映射它并在您的查询中传递连接:

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

确保在使用 shutdownGracefully(queue:_:) 完成后关闭您的 worker