forked from vapor/postgres-kit
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPostgreSQLProvider.swift
38 lines (33 loc) · 1.22 KB
/
PostgreSQLProvider.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import Async
import Service
/// Provides base `PostgreSQL` services such as database and connection.
public final class PostgreSQLProvider: Provider {
/// Creates a new `PostgreSQLProvider`.
public init() {}
/// See `Provider.register`
public func register(_ services: inout Services) throws {
try services.register(DatabaseKitProvider())
services.register(PostgreSQLDatabaseConfig.self)
services.register(PostgreSQLDatabase.self)
var databases = DatabasesConfig()
databases.add(database: PostgreSQLDatabase.self, as: .psql)
services.register(databases)
}
/// See `Provider.boot`
public func didBoot(_ worker: Container) throws -> Future<Void> {
return .done(on: worker)
}
}
/// MARK: Services
extension PostgreSQLDatabaseConfig: ServiceType {
/// See `ServiceType.makeService(for:)`
public static func makeService(for worker: Container) throws -> PostgreSQLDatabaseConfig {
return try .default()
}
}
extension PostgreSQLDatabase: ServiceType {
/// See `ServiceType.makeService(for:)`
public static func makeService(for worker: Container) throws -> PostgreSQLDatabase {
return try .init(config: worker.make())
}
}