Add PostgresQuery

Postgres compiles Fix tests
vapor · Feb 19, 2022 · 4a2b050 · 4a2b050
1 parent f588870
commit 4a2b050
Show file tree

Hide file tree

Showing 49 changed files with 465 additions and 386 deletions.
diff --git a/Sources/PostgresNIO/Connection/PostgresConnection+Connect.swift b/Sources/PostgresNIO/Connection/PostgresConnection+Connect.swift
@@ -10,16 +10,11 @@ extension PostgresConnection {
         logger: Logger = .init(label: "codes.vapor.postgres"),
         on eventLoop: EventLoop
     ) -> EventLoopFuture<PostgresConnection> {
-
-        let coders = PSQLConnection.Configuration.Coders(
-            jsonEncoder: _defaultJSONEncoder
-        )
-
         let configuration = PSQLConnection.Configuration(
             connection: .resolved(address: socketAddress, serverName: serverHostname),
             authentication: nil,
-            tlsConfiguration: tlsConfiguration,
-            coders: coders)
+            tlsConfiguration: tlsConfiguration
+        )
 
         return PSQLConnection.connect(
             configuration: configuration,

diff --git a/Sources/PostgresNIO/Connection/PostgresConnection+Database.swift b/Sources/PostgresNIO/Connection/PostgresConnection+Database.swift
@@ -65,7 +65,18 @@ extension PostgresConnection: PostgresDatabase {
                 request.prepared = PreparedQuery(underlying: $0, database: self)
             }
         case .executePreparedStatement(let preparedQuery, let binds, let onRow):
-            resultFuture = self.underlying.execute(preparedQuery.underlying, binds, logger: logger).flatMap { rows in
+            var bindings = PostgresBindings()
+            binds.forEach { data in
+                try! bindings._append(data, context: .default)
+            }
+
+            let statement = PSQLExecuteStatement(
+                name: preparedQuery.underlying.name,
+                binds: bindings,
+                rowDescription: preparedQuery.underlying.rowDescription
+            )
+
+            resultFuture = self.underlying.execute(statement, logger: logger).flatMap { rows in
                 guard let lookupTable = preparedQuery.lookupTable else {
                     return self.eventLoop.makeSucceededFuture(())
                 }

diff --git a/Sources/PostgresNIO/Data/PostgresDataType.swift b/Sources/PostgresNIO/Data/PostgresDataType.swift
@@ -26,7 +26,7 @@ public typealias PostgresFormatCode = PostgresFormat
 
 /// The data type's raw object ID.
 /// Use `select * from pg_type where oid = <idhere>;` to lookup more information.
-public struct PostgresDataType: RawRepresentable, Equatable, CustomStringConvertible {
+public struct PostgresDataType: RawRepresentable, Hashable, CustomStringConvertible {
     /// `0`
     public static let null = PostgresDataType(0)
     /// `16`

diff --git a/Sources/PostgresNIO/New/BufferedMessageEncoder.swift b/Sources/PostgresNIO/New/BufferedMessageEncoder.swift
@@ -1,21 +1,21 @@
 import NIOCore
 
-struct BufferedMessageEncoder<Encoder: MessageToByteEncoder> {
+struct BufferedMessageEncoder {
     private enum State {
         case flushed
         case writable
     }
 
     private var buffer: ByteBuffer
     private var state: State = .writable
-    private var encoder: Encoder
+    private var encoder: PSQLFrontendMessageEncoder
 
-    init(buffer: ByteBuffer, encoder: Encoder) {
+    init(buffer: ByteBuffer, encoder: PSQLFrontendMessageEncoder) {
         self.buffer = buffer
         self.encoder = encoder
     }
 
-    mutating func encode(_ message: Encoder.OutboundIn) throws {
+    mutating func encode(_ message: PSQLFrontendMessage) {
         switch self.state {
         case .flushed:
             self.state = .writable
@@ -25,7 +25,7 @@ struct BufferedMessageEncoder<Encoder: MessageToByteEncoder> {
             break
         }
 
-        try self.encoder.encode(data: message, out: &self.buffer)
+        self.encoder.encode(data: message, out: &self.buffer)
     }
 
     mutating func flush() -> ByteBuffer? {

diff --git a/Sources/PostgresNIO/New/Connection State Machine/ConnectionStateMachine.swift b/Sources/PostgresNIO/New/Connection State Machine/ConnectionStateMachine.swift
@@ -84,8 +84,8 @@ struct ConnectionStateMachine {
         // Connection Actions
 
         // --- general actions
-        case sendParseDescribeBindExecuteSync(query: String, binds: [PSQLEncodable])
-        case sendBindExecuteSync(statementName: String, binds: [PSQLEncodable])
+        case sendParseDescribeBindExecuteSync(PostgresQuery)
+        case sendBindExecuteSync(PSQLExecuteStatement)
         case failQuery(ExtendedQueryContext, with: PSQLError, cleanupContext: CleanUpContext?)
         case succeedQuery(ExtendedQueryContext, columns: [RowDescription.Column])
         case succeedQueryNoRowsComming(ExtendedQueryContext, commandTag: String)
@@ -1050,11 +1050,11 @@ extension ConnectionStateMachine {
             }
 
             return false
-        case .decoding(_):
+        case .decoding:
             return true
-        case .unexpectedBackendMessage(_):
+        case .unexpectedBackendMessage:
             return true
-        case .unsupportedAuthMechanism(_):
+        case .unsupportedAuthMechanism:
             return true
         case .authMechanismRequiresPassword:
             return true
@@ -1106,10 +1106,10 @@ extension ConnectionStateMachine {
 extension ConnectionStateMachine {
     mutating func modify(with action: ExtendedQueryStateMachine.Action) -> ConnectionStateMachine.ConnectionAction {
         switch action {
-        case .sendParseDescribeBindExecuteSync(let query, let binds):
-            return .sendParseDescribeBindExecuteSync(query: query, binds: binds)
-        case .sendBindExecuteSync(let statementName, let binds):
-            return .sendBindExecuteSync(statementName: statementName, binds: binds)
+        case .sendParseDescribeBindExecuteSync(let query):
+            return .sendParseDescribeBindExecuteSync(query)
+        case .sendBindExecuteSync(let executeStatement):
+            return .sendBindExecuteSync(executeStatement)
         case .failQuery(let requestContext, with: let error):
             let cleanupContext = self.setErrorAndCreateCleanupContextIfNeeded(error)
             return .failQuery(requestContext, with: error, cleanupContext: cleanupContext)

diff --git a/Sources/PostgresNIO/New/Connection State Machine/ExtendedQueryStateMachine.swift b/Sources/PostgresNIO/New/Connection State Machine/ExtendedQueryStateMachine.swift
@@ -23,8 +23,8 @@ struct ExtendedQueryStateMachine {
     }
 
     enum Action {
-        case sendParseDescribeBindExecuteSync(query: String, binds: [PSQLEncodable])
-        case sendBindExecuteSync(statementName: String, binds: [PSQLEncodable])
+        case sendParseDescribeBindExecuteSync(PostgresQuery)
+        case sendBindExecuteSync(PSQLExecuteStatement)
 
         // --- general actions
         case failQuery(ExtendedQueryContext, with: PSQLError)
@@ -56,18 +56,18 @@ struct ExtendedQueryStateMachine {
         case .unnamed(let query):
             return self.avoidingStateMachineCoW { state -> Action in
                 state = .parseDescribeBindExecuteSyncSent(queryContext)
-                return .sendParseDescribeBindExecuteSync(query: query, binds: queryContext.bind)
+                return .sendParseDescribeBindExecuteSync(query)
             }
 
-        case .preparedStatement(let name, let rowDescription):
+        case .preparedStatement(let prepared):
             return self.avoidingStateMachineCoW { state -> Action in
-                switch rowDescription {
+                switch prepared.rowDescription {
                 case .some(let rowDescription):
                     state = .rowDescriptionReceived(queryContext, rowDescription.columns)
                 case .none:
                     state = .noDataMessageReceived(queryContext)
                 }
-                return .sendBindExecuteSync(statementName: name, binds: queryContext.bind)
+                return .sendBindExecuteSync(prepared)
             }
         }
     }

diff --git a/Sources/PostgresNIO/New/Data/Array+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Array+PSQLCodable.swift
@@ -76,7 +76,10 @@ extension Array: PSQLEncodable where Element: PSQLArrayElement {
         .binary
     }
 
-    func encode(into buffer: inout ByteBuffer, context: PSQLEncodingContext) throws {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into buffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) throws {
         // 0 if empty, 1 if not
         buffer.writeInteger(self.isEmpty ? 0 : 1, as: UInt32.self)
         // b

diff --git a/Sources/PostgresNIO/New/Data/Bool+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Bool+PSQLCodable.swift
@@ -49,7 +49,10 @@ extension Bool: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self ? 1 : 0, as: UInt8.self)
     }
 }
diff --git a/Sources/PostgresNIO/New/Data/Bytes+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Bytes+PSQLCodable.swift
@@ -11,7 +11,10 @@ extension PSQLEncodable where Self: Sequence, Self.Element == UInt8 {
         .binary
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeBytes(self)
     }
 }
@@ -25,7 +28,10 @@ extension ByteBuffer: PSQLCodable {
         .binary
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         var copyOfSelf = self // dirty hack
         byteBuffer.writeBuffer(&copyOfSelf)
     }
@@ -49,7 +55,10 @@ extension Data: PSQLCodable {
         .binary
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeBytes(self)
     }
 

diff --git a/Sources/PostgresNIO/New/Data/Date+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Date+PSQLCodable.swift
@@ -34,9 +34,12 @@ extension Date: PSQLCodable {
         }
     }
 
-    func encode(into buffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         let seconds = self.timeIntervalSince(Self._psqlDateStart) * Double(Self._microsecondsPerSecond)
-        buffer.writeInteger(Int64(seconds))
+        byteBuffer.writeInteger(Int64(seconds))
     }
 
     // MARK: Private Constants

diff --git a/Sources/PostgresNIO/New/Data/Decimal+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Decimal+PSQLCodable.swift
@@ -32,7 +32,10 @@ extension Decimal: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         let numeric = PostgresNumeric(decimal: self)
         byteBuffer.writeInteger(numeric.ndigits)
         byteBuffer.writeInteger(numeric.weight)

diff --git a/Sources/PostgresNIO/New/Data/Float+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Float+PSQLCodable.swift
@@ -36,7 +36,10 @@ extension Float: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.psqlWriteFloat(self)
     }
 }
@@ -77,7 +80,10 @@ extension Double: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.psqlWriteDouble(self)
     }
 }

diff --git a/Sources/PostgresNIO/New/Data/Int+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Int+PSQLCodable.swift
@@ -27,7 +27,10 @@ extension UInt8: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self, as: UInt8.self)
     }
 }
@@ -64,7 +67,10 @@ extension Int16: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self, as: Int16.self)
     }
 }
@@ -105,7 +111,10 @@ extension Int32: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self, as: Int32.self)
     }
 }
@@ -151,7 +160,10 @@ extension Int64: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self, as: Int64.self)
     }
 }
@@ -204,7 +216,10 @@ extension Int: PSQLCodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeInteger(self, as: Int.self)
     }
 }
diff --git a/Sources/PostgresNIO/New/Data/JSON+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/JSON+PSQLCodable.swift
@@ -33,7 +33,10 @@ extension PSQLCodable where Self: Codable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) throws {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) throws {
         byteBuffer.writeInteger(JSONBVersionByte)
         try context.jsonEncoder.encode(self, into: &byteBuffer)
     }

diff --git a/Sources/PostgresNIO/New/Data/Optional+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/Optional+PSQLCodable.swift
@@ -44,11 +44,17 @@ extension Optional: PSQLEncodable where Wrapped: PSQLEncodable {
         }
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) throws {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         preconditionFailure("Should never be hit, since `encodeRaw` is implemented.")
     }
 
-    func encodeRaw(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) throws {
+    func encodeRaw<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) throws {
         switch self {
         case .none:
             byteBuffer.writeInteger(-1, as: Int32.self)

diff --git a/Sources/PostgresNIO/New/Data/RawRepresentable+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/RawRepresentable+PSQLCodable.swift
@@ -23,7 +23,10 @@ extension PSQLCodable where Self: RawRepresentable, RawValue: PSQLCodable {
         return selfValue
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) throws {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) throws {
         try rawValue.encode(into: &byteBuffer, context: context)
     }
 }
diff --git a/Sources/PostgresNIO/New/Data/String+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/String+PSQLCodable.swift
@@ -10,7 +10,10 @@ extension String: PSQLCodable {
         .binary
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         byteBuffer.writeString(self)
     }
 

diff --git a/Sources/PostgresNIO/New/Data/UUID+PSQLCodable.swift b/Sources/PostgresNIO/New/Data/UUID+PSQLCodable.swift
@@ -12,7 +12,10 @@ extension UUID: PSQLCodable {
         .binary
     }
 
-    func encode(into byteBuffer: inout ByteBuffer, context: PSQLEncodingContext) {
+    func encode<JSONEncoder: PostgresJSONEncoder>(
+        into byteBuffer: inout ByteBuffer,
+        context: PSQLEncodingContext<JSONEncoder>
+    ) {
         let uuid = self.uuid
         byteBuffer.writeBytes([
             uuid.0, uuid.1, uuid.2, uuid.3,