PostgresClient+SQL.swift

import PostgresNIO
import Foundation
import SQLKit

extension PostgresDatabase {
    public func sql(
        encoder: PostgresDataEncoder = PostgresDataEncoder(),
        decoder: PostgresDataDecoder = PostgresDataDecoder()
    ) -> SQLDatabase {
        _PostgresSQLDatabase(database: self, encoder: encoder, decoder: decoder)
    }
}

// MARK: Private

private struct _PostgresSQLDatabase {
    let database: PostgresDatabase
    let encoder: PostgresDataEncoder
    let decoder: PostgresDataDecoder
}

extension _PostgresSQLDatabase: SQLDatabase {
    var logger: Logger {
        self.database.logger
    }
    
    var eventLoop: EventLoop {
        self.database.eventLoop
    }
    
    var dialect: SQLDialect {
        PostgresDialect()
    }
    
    func execute(sql query: SQLExpression, _ onRow: @escaping (SQLRow) -> ()) -> EventLoopFuture<Void> {
        let (sql, binds) = self.serialize(query)
        do {
            return try self.database.query(sql, binds.map { encodable in
                return try self.encoder.encode(encodable)
            }) { row in
                onRow(row.sql(decoder: self.decoder))
            }
        } catch {
            return self.eventLoop.makeFailedFuture(error)
        }
    }
}