PostgreSQLDatabase.swift

import Async

/// Creates connections to an identified PostgreSQL database.
public final class PostgreSQLDatabase: Database, LogSupporting {
    /// See `LogSupporting`
    public static func enableLogging(_ logger: DatabaseLogger, on conn: PostgreSQLConnection) {
        conn.logger = logger
    }

    /// This database's configuration.
    public let config: PostgreSQLDatabaseConfig

    /// Creates a new `PostgreSQLDatabase`.
    public init(config: PostgreSQLDatabaseConfig) {
        self.config = config
    }

    /// See `Database.makeConnection()`
    public func newConnection(on worker: Worker) -> Future<PostgreSQLConnection> {
        let config = self.config
        return Future.flatMap(on: worker) {
            return try PostgreSQLConnection.connect(hostname: config.hostname, port: config.port, on: worker) { error in
                print("[PostgreSQL] \(error)")
            }.flatMap(to: PostgreSQLConnection.self) { client in
                return client.authenticate(
                    username: config.username,
                    database: config.database,
                    password: config.password
                ).transform(to: client)
            }
        }
    }
}

extension DatabaseIdentifier {
    /// Default identifier for `PostgreSQLDatabase`.
    public static var psql: DatabaseIdentifier<PostgreSQLDatabase> {
        return .init("psql")
    }
}